基于目标检测的标签分配方法及装置
知识产权简介
本发明提供了一种针对一阶段目标检测算法的标签分配方法及系统,包括:基于训练时损失函数值将训练样本分为正样本和负样本;根据预测框的置信度以及预测框与真值框之间的重叠度将假阳性预测框分为三类;针对第一类假阳性样本,设置其分类标签的数值与回归精度成正比;针对第二类假阳性样本,基于对真阳性样本的估计,计算每个样本属于第二类假阳性样本的风险,设置训练样本的分类标签的数值与此风险成反比并为所有正样本分配回归损失权重。这套针对一阶段算法的标签分配方法有效地抑制了假阳性预测框的产生,相比于传统的标签分配方法有效地提升了检测算法的准确率,在检测目标分布密集的场景下有显著优势。