一种高效的无人机航拍小目标检测算法
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391.41

基金项目:

国家自然科学基金项目(62276202, 62106186, 72071209).


An efficient algorithm for small object detection in unmanned aerial vehicle images
Author:
Affiliation:

Fund Project:

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

    无人机航拍图像具有尺度差异大、背景干扰和目标模糊等特点, 给小目标检测带来诸多挑战. 针对这些问题, 提出一种高效的无人机航拍小目标检测算法. 首先利用空洞卷积增大感受野、保持细节分辨率的特点, 设计并行空洞卷积模块; 其次设计注意力上下采样分支模块, 利用闸门机制对提取到的特征进行选择, 强化特征表达; 最后结合小目标检测头设计并行空洞卷积注意力金字塔网络, 对多尺度特征进行特征融合. 在VisDrone2023数据集和DOTA数据集上, 所提出算法在小目标检测的平均准确率均值均优于其他主流算法, 相较于基线方法在平均准确率均值上提升7.3 %, 参数量减少0.58 M, FPS提升11.2, 达到43.5, 验证了所提算法的高效性. 在复杂场景ExDark数据集上, 所提出算法在平均准确率均值上优于其他低光增强模型和暗检测器, 相较于PE-YOLO在平均准确率均值上提升2.4 %, 验证了所提算法的鲁棒性和实用性.

    Abstract:

    Unmanned aerial vehicle(UAV) images have distinctive characteristics such as significant scale variance, background interference, and fuzzy objects, which brings many challenges for object detection algorithms. To address these issues, an efficient algorithm for small object detection in UAV images is proposed. Firstly, parallel atrous convolution(PAC) is designed to increase the receptive field and maintain detail resolution by utilizing atrous convolution. Then, an attentional upsampling and downsampling module is designed to selectively enhance the extracted features by using gate mechanisms. Finally, in combination with tiny object prediction head (TOPH), a parallel atrous convolution attention pyramid network(PACAPN) is designed to fuse multi-scale features. On the VisDrone2023 dataset and the DOTA dataset, the proposed algorithm achieves higher mean average precision(mAP) in both small object detection in comparison other algorithms. Compared with the baseline method, the mAP is increased by 7.3 %, the number of parameters is reduced by 0.58 M, and the FPS(frame per second) is increased by 11.2 to 43.5, which verifies the effectiveness of the algorithm. On the ExDark dataset, the proposed algorithm achieves higher mAP in comparison low-light enhancement models and dark detectors. Compared with the PE-YOLO, the mAP is increased by 2.4 %, which verifies the robustness and practicality of the proposed algorithm.

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

高卫峰,易宇轩,黄玲玲,等.一种高效的无人机航拍小目标检测算法[J].控制与决策,2025,40(8):2525-2533

复制
相关视频

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