西安电子科技大学学报 ›› 2025, Vol. 52 ›› Issue (2): 47-56.doi: 10.19665/j.issn1001-2400.20250304

• • 上一篇    下一篇

一种轻量级小目标无人机检测YOLO模型

阳小兵(), 李钊(), 许艳红()   

  1. 西安电子科技大学 网络与信息安全学院,陕西 西安 710126
  • 收稿日期:2024-10-15 出版日期:2025-04-20 发布日期:2025-03-19
  • 通讯作者: 李 钊(1981—),男,副教授,E-mail:zli@xidian.edu.cn
  • 作者简介:阳小兵(2000—),男,西安电子科技大学硕士研究生,E-mail:xbyang@stu.xidian.edu.cn;
    许艳红(1999—),女,西安电子科技大学硕士研究生,E-mail:yhxu_1@stu.xidian.edu.cn
  • 基金资助:
    国家自然科学基金(62072351);国家自然科学基金(62202359);国家自然科学基金(U23A20300);陕西省重点研发计划(2023JCZD35);高等学校学科创新引智计划(B16037);河南省科技攻关项目(252102211120)

Lightweight YOLO model for small UAV object detection

YANG Xiaobing(), LI Zhao(), XU Yanhong()   

  1. School of Cyber Engineering,Xidian University,Xi’an 710126,China
  • Received:2024-10-15 Online:2025-04-20 Published:2025-03-19

摘要:

由于无人机体积小、空域背景复杂且容易与鸟类等天空目标混淆,已有的目标检测模型精度不足。虽然增加模型规模可以在一定程度上提升检测精度,但也会降低模型推理速度、增大参数量与计算量。此外,目前可用于小目标无人机检测的数据集缺乏,难以有效支持无人机检测方法设计。针对以上问题,首先根据现有的开源无人机与鸟类检测数据集,采用基于目标面积压缩的小目标样本增强方法,构建一个可用于小目标无人机与鸟类分类任务的数据集。然后,对YOLOv8模型进行改造,通过使用轻量级自适应下采样卷积结构(LADC),建立了轻量级模型YOLO-LADC,可以在降低模型参数量和计算量的同时提升检测精度。更进一步,在YOLO-LADC的颈部网络增加分支结构得到YOLO-LADCS模型,以更好地适应小目标无人机检测任务。对比实验表明,YOLO-LADCS能够在参数量相比YOLOv8的轻量级衍生版本YOLOv8n减少14%的情况下,将小目标检测的平均准确率提升约1.1%。

关键词: 目标检测, 神经网络, 无人机检测, 小目标, 轻量化

Abstract:

Due to the small size of Unmanned Aerial Vehicles(UAVs),complex airspace background,and easy confusion with sky objects such as birds,the existing object detection models lack sufficient accuracy.Although increasing the model size can improve the detection accuracy to a certain extent,it also reduces the inferring speed and significantly increases the number of parameters and computational complexity of the model.In addition,the lack of datasets which are suitable for small UAV object detection makes it challenging to provide adequate support for designing effective models.To address the aforementioned deficiencies,this paper first constructs a dataset from existing open-source datasets using a target-area-compression based small object sample enhancement method,which can be utilized in small UAV object detection tasks.Then,we design a lightweight and high-accuracy network model called YOLO-LADC,based on the YOLOv8.This model incorporates a novel downsampling convolution structure that reduces the number of model parameters and computations while enhancing the detection accuracy.Moreover,we add a small object detection branch to the neck network of the YOLO-LADC to achieve the YOLO-LADCS,which is better suited for small UAV object detection tasks.Comparative experiments show that the YOLO-LADCS is able to improve the average accuracy of a small object by 1.1% with a 14% reduction in the number of parameters compared to the YOLOv8n(a lightweight version of the YOLOv8).

Key words: object detection, neural networks, uav detection, small object, lightweight

中图分类号: 

  • TP183