基于改进的YOLOv5遥感影像飞机目标检测  

Aircraft target detection based on improved YOLOv5 in remote sensing imagery

作　　者：黄子恒 芮杰[1] 林雨准 王淑香[1] 刘相云 HUANG Ziheng;RUI Jie;LIN Yuzhun;WANG Shuxiang;LIU Xiangyun(Information Engineering University,Zhengzhou 450001,China)

机构地区：[1]信息工程大学,河南郑州450001

出　　处：《测绘通报》2024年第8期73-78,89,共7页Bulletin of Surveying and Mapping

摘　　要：针对现有目标检测算法在遥感影像飞机目标检测上存在检测速度慢、精度低、影像背景复杂、不易区分背景和目标,以及预测过程的收敛速度较慢、效率低等问题,本文在YOLOv5算法模型的基础上,采用引入注意力机制和更换损失函数两种优化策略,提升了算法对飞机目标的检测性能。在DOTA数据集上的训练结果显示,通过在YOLOv5网络结构主干部分的C3模块中引入CBAM注意力机制,算法检测性能得到显著提升。其中,模型训练结果的精确度(P)提升6%,召回率(R)提升2%,平均精度(mAP)提升2.8%;在YOLOv5预测过程中分别采用Focal EIoU和SIoU损失函数对原有CIoU损失函数进行替换,试验结果表明,改进后的算法模型回归精度显著提高,其中采用SIoU损失函数的模型优化效果最好,模型训练结果的精确度(P)提升4.3%,召回率(R)提升2%,平均精度值(mAP)提升2.7%。改进后的YOLOv5算法为实现对飞机目标的高精度实时检测提供参考。In response to the problems of slow detection speed,low accuracy,complex image background,difficulty in distinguishing background and target,slow convergence speed and low efficiency in the prediction process of existing object detection algorithms in remote sensing image aircraft target detection,this paper adopts two optimization strategies,namely the introduction of attention mechanism and replacement loss function.Based on the YOLOv5 algorithm model,to improve the algorithm s detection performance for aircraft targets,the training results on the DOTA dataset show that by introducing the CBAM attention mechanism in the C3 module of the YOLOv5 network architecture backbone,the algorithm s detection performance has been significantly improved.Among them,the accuracy of the model training results P has been improved by 6%,the recall rate R has been improved by 2%,and the average precision(mAP)value has been improved by 2.8%.In the YOLOv5 prediction process,Focal EIoU and SIoU loss functions are used to replace the original CIoU loss functions.The experimental results show that the improved algorithm model significantly improve the regression accuracy,among which the model optimized using SIoU loss function have the best effect.The precision(P)of the model training results increased by 4.3%,the recall rate R increased by 2%,and the average precision(mAP)value increased by 2.7%.The improved YOLOv5 algorithm provides a reference for achieving high-precision real-time detection of aircraft targets.

关 键 词：目标检测 飞机检测 YOLOv5 注意力机制 损失函数 

分 类 号：P237[天文地球—摄影测量与遥感]