关键词:
小目标检测
深度学习
部分卷积
注意力机制
无人机
摘要:
随着无人机的广泛应用,提升无人机航拍图像检测效果的重要性日益凸显。针对目前主流目标检测算法在检测航拍图像中小目标时效果不理想的问题,本文提出一种基于部分卷积与注意力融合检测头的小目标检测算法。引入更高效的FasterNet主干网络,使用部分卷积(PConv)减少了深度卷积时的内存访问和冗余计算,增大网络规模提升空间特征提取的效果的同时,控制了网络计算时间。优化特征提取网络,提升对微小尺寸目标的检测效果。在检测头中引入Dynamic Head,有效地应用注意力机制,提升了检测性能。最后优化边界框损失函数为Inner-ShapeIoU,聚焦边框自身形状与尺度,提升边框回归计算精度,使用辅助边框,提升收敛速度。在公开数据集VisDrone2019上进行实验,本文算法与原YOLOv8n算法相比,准确率P提高了11.9%,mAP50提升13.4%,表明小目标检测精度显著提升。