基于YOLOv5的遥感图像目标检测
DOI:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家重点研发计划基金资助项目(2018AAA0100400);湖南省自然科学基金资助项目(2020JJ6089, 2020JJ6088);湖南省教育厅科研基金资助项目(21A0350,21C0439,19A133)


Research on a YOLOv5-Baed Remote Sensing Image Target Detection
Author:
Affiliation:

Fund Project:

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

    为了解决在遥感图像目标检测任务中目标背景繁杂难以识别且目标尺寸复杂的问题,提出一种基于YOLOv5的遥感图像检测优化模型。首先,对输入数据进行马赛克增强,增加样本多样性,同时采用自适应锚框计算,寻求最优初值锚框;然后,把通过主干网络提取到的特征层进行特征融合得到最优特征层,再对定位损失进行优化,采用CIoU loss作为定位损失函数,Focal loss作为分类损失函数;最后,在测试时对输入图片采用自适应图片缩放,以减少信息冗余,加快模型检测速率。该模型能有效捕捉图像特征,实现快速精准的目标定位。对公开10类地理空间物体检测数据集(NWPU-VHR 10)和RSOD数据集进行了训练测试,对比试验表明,优化模型mAP达到0.989 6,比优化前的模型mAP提升了2.31%,与使用相同数据集的其他模型的最优值进行比较,其mAP提升了8.19%,该方法能有效提高遥感图像检测精度。

    Abstract:

    In order to solve the problem of a complex target background which is hard to identify, as well as a complex target size in remote sensing image target detection task, a remote sensing image detection optimization model based on YOLOv5 has thus been proposed. Firstly, a mosaic enhancement is carried out on the input data to increase the diversity of samples; meanwhile, an adaptive anchor frame calculation is used to obtain the optimal initial value of the anchor frame. Then the feature layer extracted through the backbone network is to be fused, thus obtaining the optimal feature layer, to be followed by an optimization of the positioning loss, with CIoU loss used as the positioning loss function and Focal loss as the classification loss. Finally, adaptive image scaling is used for a reduction in information redundancy and speeding up of model detection. The optimized model can capture image features effectively and achieve a fast and accurate target location. Training tests are carried out on 10 types of geospatial object detection data sets (NWPU-VHR 10) as well as RSOD data sets. Comparative experiments show that, the mAP of the optimized model reaches 0.989 6, which is 2.31% higher than the model prior to the optimization. Compared with the optimization of other models using the same data sets, the mAP is 8.19% higher, verifying the fact that the optimized model can effectively improve the detection accuracy of remote sensing images.

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

董丽君,曾志高,易胜秋,文志强,孟 辰.基于YOLOv5的遥感图像目标检测[J].湖南工业大学学报,2022,36(3):44-50.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2021-09-26
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2022-05-10
  • 出版日期: 2022-05-01