您现在的位置是:首页 >学无止境 >OpenPCDet系列 | 6.PointPillars模型分类、回归、角度损失的构建网站首页学无止境

OpenPCDet系列 | 6.PointPillars模型分类、回归、角度损失的构建

Clichong 2024-06-14 17:19:07
简介OpenPCDet系列 | 6.PointPillars模型分类、回归、角度损失的构建

模型损失计算

在进行anchor的正负样本分配后,具体来说就是对iou最大以及满足阈值的anchor进行相应的类别分配以及gt编码信息分配,以及正样本的权重分配,最后的所有预测信息以及anchor分配信息保存在self.forward_ret_dict这个字典中,现在就是利用这个字典来进行损失的计算。

在这里插入图片描述

在调用PointPillars算法的get_training_loss函数时,

风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。