关键词:
目标检测
半监督学习
特征对齐
特征金字塔
ASFF
摘要:
目的针对半监督目标检测导致数据特征表示不充分,数据样本类不均衡等问题,提出一种基于特征对齐和特征融合的半监督目标检测方法。方法在常见的半监督目标检测框架中,伪标签是完全根据分类分数生成的,然而,高置信度预测并不总是保证准确的bbox定位。为了解决定位不准确问题和特征表示不充分问题,受Consistent Teacher中的FAM-3D算法启发,考虑分类和定位的最优特征可能在不同尺度上,引入T-head特征对齐头算法,在Unbiased Teacher V2中成功地将分类和定位分支进行对齐,并且引入ASFF,通过空间过滤冲突信息的方法来抑制不一致性,从而提高了特征的尺度不变性,实现特征在空间上的融合;通过学习不同特征图之间的联系来解决特征金字塔内部的不一致性问题。结果根据实验结果,改进的算法在COCO数据集、VOC数据集上都有一定的比例提升。结论改进的算法可以有效减轻数据表示不充分和数据样本类不均衡问题,同时也提高了算法的精度。