基于AMD-YOLOv8的无人机图像小目标检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金(62372100)


Small Target Detection in UAV Images Based on AMD-YOLOv8
Author:
Affiliation:

Fund Project:

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

    无人机航拍图像目标检测在实时监控、异常检测等领域具有重要应用, 但航拍图像中复杂的背景、目标尺度变化大和小目标比例高等问题增加了检测难度. 为此, 本文提出了一种改进的目标检测算法AMD-YOLOv8. 首先, 该算法采用微小目标检测头替代大目标检测头, 有效保留小目标特征信息; 引入动态目标检测头Dyhead, 通过对水平维度、空间维度以及通道维度进行细粒度的注意力调整, 显著提高了模型对目标细节特征的提取能力. 其次, 设计出多尺度全局注意力模块MSGA, 利用不同卷积核运算和全局上下文布局, 增强对远距离小目标的检测能力. 然后, 提出了双向密集扩展特征金字塔网络BDEFPN, 该网络通过扩展尺度和密集连接, 实现了高效的多尺度信息融合. 最后, 采用LAMP剪枝策略对模型进行轻量化处理, 通过自适应评估层间冗余连接并进行删除, 有效降低计算量并加块推理速度. 实验结果表明, 改进后的AMD-YOLOv8在VisDrone2019数据集上, 参数量比YOLOv8n减少46.0%, mAP50提升了8.6%, FPS达到98.3 f/s; 在UAVDT数据集的测试结果进一步验证了该算法优越的泛化能力, 证明了其在无人机航拍图像检测中的有效性.

    Abstract:

    Target detection for UAV aerial images is significant in the applications of real-time monitoring and anomaly detection. However, the detection process is complicated due to challenges such as complex backgrounds, large variations in target scales, and a high proportion of small targets. To address these issues, this study proposes an improved algorithm for small target detection, AMD-YOLOv8. First, the algorithm replaces the large-target detection head with a micro-target detection head to effectively preserve small target features. A dynamic target detection head (Dyhead) is introduced, which applies fine grained attention adjustments across the horizontal, spatial, and channel dimensions, significantly enhancing the model’s ability to extract detailed features of the target. Second, a multi-scale global attention (MSGA) module is designed, utilizing various convolutional kernel operations and global context layouts to enhance the detection capability for distant small targets. Third, a bidirectional dense extended feature pyramid network (BDEFPN) is proposed to efficiently integrate multi-scale information through scale expansion and dense connections. Finally, the LAMP pruning strategy is applied for model lightweight by adaptively assessing and removing redundant connections between layers, effectively reducing the computational load and accelerating inference speed. Experimental results demonstrate that the improved AMD-YOLOv8 reduces parameters by 46.0% compared to YOLOv8n on the VisDrone2019 dataset while achieving an 8.6% increase in mAP50 and an FPS of 98.3 f/s. Additionally, test results of the UAVDT dataset further confirm the algorithm’s superior generalization ability, validating its effectiveness in the detection of UAV aerial images.

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

杨树莹,葛华勇.基于AMD-YOLOv8的无人机图像小目标检测.计算机系统应用,2025,34(10):173-183

复制
分享
相关视频

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

京公网安备 11040202500063号