• • 下一篇
王磊,胡节,彭博
摘要: 针对火灾图像标签过少、背景复杂多样导致的半监督目标检测困难的问题,提出一种用于半监督火灾检测的分布自适应和动态课程伪标签框架(DADCPL-SFD)。该框架主要由师生互学框架、软标签、分布自适应和动态课程伪标签这4个部分组成。首先,采用师生互学框架的半监督学习范式替换YOLOv5-l的全监督学习范式,以应对数据标签少的场景;其次,采用软标签以获取更多有效的伪标签正例,优化半监督学习过程;再次,引入分布自适应损失,减少源域和目标域的数据分布差异,使模型在不同域上表现一致;最后,设计一种基于课程思想的动态课程伪标签策略,根据伪标签生成的情况在不同训练时期调整阈值,以筛选更合理的伪标签。在火焰和烟雾数据集(DFS)多个监督比例上(1%、2%、5%和10%)的实验结果表明,所提框架的平均精度均值(mAP),相较于全监督学习平均提升了5.32个百分点,在交并比(IoU)阈值为0.5下的平均精度(AP50)平均提升了11.87个百分点,充分验证了DADCPL-SFD的高效性和准确性。
中图分类号: