节点文献

改进YOLOv8的无人机小目标识别算法

Algorithm of Improved YOLOv8 for Small Target Recognition of Drones

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 向长春; 李兴山; 张海锋; 何成利; 王磊; 郝涌汀;

【Author】 Xiang Changchun;Li Xingshan;Zhang Haifeng;He Chengli;Wang Lei;Hao Yongting;

【通讯作者】 李兴山;

【机构】 沈阳理工大学机械工程学院; 北方华安工业集团有限公司; 齐齐哈尔建华机械有限公司; 四川华川工业股份有限公司;

【摘要】 针对深度学习算法难以部署到无人机硬件和小目标易漏检误检问题,提出改进YOLOv8的轻量化小目标识别算法DLS-YOLO。采用深度可分离卷积替代普通卷积轻量化网络,设计多分支空洞卷积结构IRFB增大感受野,在主干网络中加入ECA注意力机制增强特征表达能力,增设160×160尺寸检测头,提高小目标检测能力。在VisDrone2019数据集上,DLS-YOLO比YOLOv8s在mAP@0.5上提高了2.3%,计算量降低11.2GFLOPs,模型尺寸缩减至5.7 M,性能优于其他主流检测方法。在Jetson Xavier NX平台验证算法的有效性,平均单帧处理时间34.4 ms,实现近实时检测。

【Abstract】 In view of the problems that deep learning algorithm is difficult to be deployed to drone hardware and small targets are easy to be missed and mis-detected, a lightweight small target recognition algorithm, DLS-YOLO, is proposed to improve YOLOv8. This paper adopts the deep separable convolution to replace the ordinary convolution lightweight network, designs the multi-branch cavity convolution structure, IRFB, to increase the receptive field, introduces the ECA attention mechanism to the main network to enhance the ability of feature expression and add the 160×160 detection head to improve the ability of detection of the small targets. On VisDrone2019 data set, DLS-YOLO is 2.3% higher than YOLOv8s in mAP@0.5, the computational load is reduced by 11.2GFLOPs and the model size is reduced to 5.7 M, which is superior to other mainstream detection methods. The effectiveness of the algorithm is verified on the Jetson Xavier NX platform with an average processing time for a single frame of 34.4ms, thus can realize near real-time detection.

【关键词】 无人机; YOLOv8; 目标识别; 轻量化; 小目标;
【Key words】 Drone; YOLOv8; target recognition; lightweight; small target;
  • 【分类号】V279;TP391.41
  • 【下载频次】124
节点文献中: 

本文链接的文献网络图示:

本文的引文网络