基于改进YOLOv7-tiny的高效轻量遥感图像目标检测方法
DOI:
CSTR:
作者:
作者单位:

1.江西理工大学电气工程与自动化学院赣州341000;2.江西理工大学多维智能感知与控制江西省重点实验室 赣州341000;3.江西理工大学机电工程学院赣州341000

作者简介:

通讯作者:

中图分类号:

TP751.1;TN919.5

基金项目:

国家自然科学基金(51268017, 72061016)项目资助


Efficient and lightweight target detection method for remote sensing images based on improved YOLOv7-tiny
Author:
Affiliation:

1.School of Electrical Engineering and Automation, Jiangxi University of Science and Technology, Ganzhou 341000, China; 2.Jiangxi Provincial Key Laboratory of Multidimensional Intelligent Perception and Control, Jiangxi University of Science and Technology, Ganzhou 341000, China; 3.School of Mechanical and Electrical Engineering, Jiangxi University of Science and Technology, Ganzhou 341000, China

Fund Project:

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

    针对现有遥感图像目标检测方法在受资源限制的小型设备中检测精度不足问题,提出了一种基于改进YOLOv7tiny 算法的高效轻量遥感图像目标检测方法。首先,针对遥感图像小目标分布密集问题,设计了一种低跨度的上下文解耦检测头模块,通过融合深层和浅层特征分别实现目标检测的分类和回归任务,有效解决了遥感图像小目标漏检和误检的问题。同时,针对遥感图像目标多尺度问题,设计了一种并行级联注意力机制,通过并行三分支网络与空间注意力模块相结合,增强了网络对多尺度目标特征的提取能力。此外,引入Focal-EIoU损失函数,提高模型泛化能力。对模型进行了对比实验、消融实验、部署实验和泛化实验,结果表明,在DIOR-5s和NWPU VHR-10数据集上的检测精度分别达到了85.4%、90.6%,相较原模型分别提高了2.6%、1.7%。且模型大小仅为19.1 MB,检测速度为64.1 fps,验证了算法的有效性。

    Abstract:

    To address the issue of low detection accuracy in remote sensing image target detection methods for small devices with limited resources, an efficient and lightweight method based on the improved YOLOv7-tiny algorithm is proposed. To address the issue of dense distribution of small targets in remote sensing images, a low-span context decoupling detection head module is designed. This module fuses deep and shallow features to perform classification and regression tasks for target detection. It effectively solves the problems of leakage and misdetection of small targets in remote sensing images. Meanwhile, a parallel series attention mechanism is proposed to enhance the network’s ability to extract multi-scale target features for remote sensing image targets. This is achieved by combining the parallel three-branch network with the spatial attention module. Additionally, the model’s generalization ability is improved through the introduction of the Focal-EIoU loss function. The comparison experiments, ablation experiments, deployment experiments and generalization experiments were conducted on the model. Experimental results indicate that the detection accuracy on DIOR-5s and NWPU VHR-10 datasets improved by 2.6% and 1.7%, respectively, compared to the original model. The model size is only 19.1 MB, and the detection speed is 64.1 fps, verifying the algorithm’s effectiveness.

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

邝先验,程福军,吴翠琴,雷卉.基于改进YOLOv7-tiny的高效轻量遥感图像目标检测方法[J].电子测量与仪器学报,2024,38(7):22-33

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-10-18
  • 出版日期:
文章二维码