UAV-YOLO: 基于YOLOv7的红外小目标检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:


UAV-YOLO: Infrared Small Target Detection Based on YOLOv7
Author:
Affiliation:

Fund Project:

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

    在无人机红外小目标检测领域, 复杂地面背景与目标尺寸微小的特性常使检测模型出现漏检或误检问题. 针对这一情况, 本研究以YOLOv7框架为基础, 设计出一种轻量级的高精度红外小目标检测算法, 命名为UAV-YOLO (unmanned aerial vehicle-you only look once). 首先, 考虑到检测目标多为小目标, 对YOLOv7基础网络进行轻量化改造, 将ELAN、ELAN-W和CARAFE模块中的1×1卷积以及颈部网络的1×1卷积替换为GSConv, 同时去除检测效率不高的P5检测头, 并新增专门用于检测小目标的P2检测头. 不仅提升了检测效率, 还大幅减少了参数量, 实现了模型的轻量化. 其次, 在骨干网络中融入了改进后的金字塔池化模块SPPFCSPC. 该模块的加入, 能够有效扩展模型的感知范围, 从而提高对红外小目标的检测精度. 然后, 把内容感知特征重组(CARAFE)架构集成到YOLOv7中. 这一架构可以更好地保留和优化小目标的特征表示. 同时, 在检测头前引入CA (coordinate attention)注意力机制模块. 该模块能够精准定位小目标, 让检测头更容易聚焦于关键区域. 最后, 采用归一化高斯Wasserstein距离(NWD)损失来替代CIoU. 这种替代降低了模型对位置偏差的敏感性, 进一步提升了检测效率. 实验数据显示, 与原始模型相比, 该模型的mAP达到了95.7%, 比YOLOv7基础模型提升了5.2%; 而参数量仅为 12.0M, 下降了67.7%. 这些优化改进在保证高精度的同时, 大幅减少了参数量, 充分验证了基于YOLOv7的红外小目标检测模型的实用性, 使检测性能得到了显著提升.

    Abstract:

    In the field of infrared small target detection for unmanned aerial vehicle (UAV), complex ground backgrounds and the inherently small size of targets often result in missed detections or false alarms in detection models. To address this problem, this study proposes a lightweight and high-precision infrared small target detection algorithm based on the YOLOv7 framework, termed unmanned aerial vehicle-you only look once (UAV-YOLO). First, given that the targets of interest are predominantly small, the 1×1 convolutions in the ELAN, ELAN-W, and CARAFE modules of the YOLOv7 backbone network, as well as those in the neck network, are replaced with GSConv. At the same time, the P5 detection head, which provides limited efficiency in infrared small target detection, is removed, and a newly added P2 detection head is designed specifically for small targets. These modifications not only enhance detection efficiency but also substantially reduce the number of parameters, thus achieving a lightweight model. Second, an improved SPPFCSPC pyramid pooling module is integrated into the backbone network. The inclusion of this module effectively expands the receptive field of the model, thus improving the detection accuracy of infrared small targets. Subsequently, the content-aware reassembly of feature (CARAFE) module is incorporated into YOLOv7. This module enhances the preservation and optimization of feature representations for small targets. Furthermore, a coordinate attention (CA) mechanism is introduced before the detection head, enabling more precise localization of small targets and allowing the detection head to focus on key regions. Finally, the normalized Wasserstein distance (NWD) loss is adopted to replace the CIoU loss. This replacement reduces the sensitivity of the model to positional deviations and further improves detection performance. Experimental results demonstrate that compared with the baseline YOLOv7 model, the proposed UAV-YOLO achieves an mAP of 95.7%, representing a 5.2% improvement, while the number of parameters is reduced to 12.0M, a decrease of 67.7%. These improvements ensure high precision while significantly reducing the number of parameters, thus verifying the effectiveness and practicality of the proposed infrared small target detection model for UAV applications.

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

苏健,张莉莉. UAV-YOLO: 基于YOLOv7的红外小目标检测.计算机系统应用,2026,35(2):209-225

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2025-07-16
  • 最后修改日期:2025-08-13
  • 录用日期:
  • 在线发布日期: 2025-12-19
  • 出版日期:
文章二维码
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京市海淀区中关村南四街4号,邮政编码:100190
电话:010-62661041 传真: Email:csa@iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号