关键词:
小目标检测
重参数化
特征融合
损失函数
摘要:
针对无人机航拍图像目标尺寸小、分布密集且被遮挡造成误检漏检等问题,提出一种结合重参数化思想与多层次特征融合的航拍图像小目标检测算法.首先,利用重参数化思想设计了重参数化卷积模块(reparameterized convolution module,RCM),与C2f模块结合设计了C2f-RCM模块,通过扩大感受野有效绘制上下文信息,更好地提取图像中的细微特征.其次,为解决颈部网络在特征融合部分造成的信息丢失问题,提出一种多层次特征融合模块(multi-level feature fusion module,MFFM),该模块利用跨层次间的信息融合,有效减少了在遮挡情况下的漏检现象,使得网络在检测大、中、小目标时能够显著提升准确度.最后,提出一种Inner-Shape IoU边界框回归损失函数,通过构建辅助边框和关注边界框的自身形状,以增强模型的收敛速度.实验结果表明,与基线模型相比,本文方法在VisDrone2019中,mAP@0.5、Precision、Recall分别提高了5.7%、5.7%、2.4%,在AI-TOD中,mAP@0.5、Precision、Recall提升了3.7%、3.9%、5.3%,验证了本文方法在航拍图像小目标检测方面的有效性.