基于改进YOLOv5的遥感图像飞机目标检测
作者:
作者单位:

1.安徽商贸职业技术学院信智学院,安徽 芜湖 241000;2.芜湖市物联网智慧交通工程技术研究中心, 安徽 芜湖 241000

作者简介:

通讯作者:

基金项目:

安徽省高校自然科学研究重点项目(2022AH052740);安徽省职成教2022教育科研规划课题(Azcj2022128);安徽商贸职业技术学院自然科学重点项目(2022KZZ05)。


Aircraft Target Detection Based on Improved YOLOv5 for Remote Sensing Images
Author:
Affiliation:

1.Xinzhi College, Anhui Vocational & Technical College of Business & Trade,Wuhu, Anhui 241000, China;2.Wuhu Internet of Things Smart Transportation Engineering Technology Research Center , Wuhu, Anhui 241000, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
    摘要:

    针对遥感图像中飞机检测尺寸大小不一、背景复杂导致的难以识别问题,提出一种基于YOLOv5网络模型的改进方法。首先,在YOLOv5网络模型中融入Swin-Transformer模块,使网络全局建模并使全维度信息交互,以提升网络的特征提取能力;其次,对损失函数进行优化,引入SIOU损失函数以考虑真实框和预测框之间的向量角度问题。对比实验表明,优化前后网络模型检测精度均为95.3%。在检测精度相同的情况下,改进后的网络模型召回率为91.2%,比改进前提升0.6个百分点;改进后平均检测精度mAP0.5为95.7%,比改进前提升0.2个百分点。结果表明,改进后的YOLOv5网络模型能在一定程度上提升遥感图像中飞机目标检测性能。

    Abstract:

    Aiming at the difficult recognition problem caused by the different sizes of aircraft detection and complex backgrounds in remote sensing images, an improving method based on the YOLOv5 network model is proposed. First, the Swin-Transformer module is incorporated into the YOLOv5 network model to model the network globally and make the full-dimensional information interactive to improve the feature extraction capability of the network; second, the loss function is optimized and the SIOU loss function is introduced to consider the vector angle problem between the real frame and the predicted frame. The comparison experiments show that the detection accuracy of the network model before and after optimization is both 95.3%. With the same detection accuracy, the recall rate of the improved network model is 91.2%, which is 0.6 percentage points higher than that before the improvement; the average detection accuracy mAP0.5 is 95.7% after the improvement, which is 0.2 percentage points higher than that before the improvement. The results show that the improved YOLOv5 network model can improve the performance of aircraft target detection in remote sensing images to a certain extent.

    参考文献
    相似文献
    引证文献
引用本文

孙廨尧,侯秀丽,罗青青.基于改进YOLOv5的遥感图像飞机目标检测[J].西昌学院学报(自然科学版),2023,37(2):67-72.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
历史
  • 收稿日期:2023-02-06
  • 最后修改日期:2023-03-01
  • 录用日期:2023-04-12
  • 在线发布日期: 2023-07-25