Subtle-YOLOv8: a detection algorithm for tiny and complex targets in UAV aerial imagery

被引:1
|
作者
Zhao, Sicheng [1 ,2 ]
Chen, Jinguang [1 ,2 ]
Ma, Lili [1 ,2 ]
机构
[1] Xian Polytech Univ, Sch Comp Sci, Shaanxi Key Lab Clothing Intelligence, Xian 710048, Peoples R China
[2] Hubei Engn Res Ctr Intelligent Detect & Identifica, Wuhan 430205, Peoples R China
关键词
UAV; Small-object detection; YOLOv8; Deformable convolution; Attention mechanism; WIoU;
D O I
10.1007/s11760-024-03520-7
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Unmanned Aerial Vehicle (UAV) imagery for small target detection plays a crucial role in traffic safety, military defense, and agricultural production. Despite rapid advancements in target detection algorithms, tiny targets like pedestrians, people, and bicycles still encounter significant challenges in practical applications, including occlusions, low resolution, and difficulties in capture and segmentation. These challenges require detectors to be highly adaptive and capable of precisely distinguishing between targets and dynamic backgrounds. To address these issues, we use YOLOv8 as the baseline model and proposes a new detection network named Subtle-YOLOv8. Initially, dynamic snake convolution (DSConv) is incorporated into the backbone network to enhance the perception of subtle information and feature extraction efficiency. Secondly, an attention mechanism called Efficient Multi-scale Attention Module (EMA) is introduced to optimize the neck network to improve the transfer of key features. Finally, we designed a tiny object detection head and replace the original loss function with Wise-IoU, focusing the model more on samples of ordinary quality and further enhancing the detection capabilities for tiny targets. Experimental results show that our model achieves a 6.2% improvement in average detection precision over the baseline with a slight increase in parameters. It particularly excels in handling complex tiny targets such as pedestrians and people, with detection precision improvements of 14% and 12%, respectively. The code will be soon released at https://github.com/WilliamXSS/SubtleYOLO
引用
收藏
页码:8949 / 8964
页数:16
相关论文
共 50 条
  • [21] DCM-YOLOv8: An Improved YOLOv8-Based Small Target Detection Model for UAV Images
    Xing, Zhecong
    Zhu, Yuan
    Liu, Rui
    Wang, Weiqi
    Zhang, Zhiguo
    ADVANCED INTELLIGENT COMPUTING TECHNOLOGY AND APPLICATIONS, PT VI, ICIC 2024, 2024, 14867 : 367 - 379
  • [22] A lightweight enhanced YOLOv8 algorithm for detecting small objects in UAV aerial photography
    Pan, Wei
    Yang, Zhe
    VISUAL COMPUTER, 2025,
  • [23] Image target detection algorithm based on YOLOv7-tiny in complex background
    Xue S.
    An H.
    Lv Q.
    Cao G.
    Hongwai yu Jiguang Gongcheng/Infrared and Laser Engineering, 2024, 53 (01):
  • [24] A Novel Variant of Yolov7-Tiny for Object Detection on Aerial Vehicle Images
    Huy Hoang Nguyen
    Van Quang Nghiem
    Minh Son Hoang
    Tuan Khoi Nghiem
    Ngoc Minh Dang
    COMMUNICATION AND INTELLIGENT SYSTEMS, VOL 1, ICCIS 2023, 2024, 967 : 253 - 265
  • [25] MS-YOLOv7:YOLOv7 Based on Multi-Scale for Object Detection on UAV Aerial Photography
    Zhao, LiangLiang
    Zhu, MinLing
    DRONES, 2023, 7 (03)
  • [26] Improved YOLOv5s Algorithm for Small Target Detection in UAV Aerial Photography
    Li, Shixin
    Liu, Chen
    Tang, Kaiwen
    Meng, Fanrun
    Zhu, Zhiren
    Zhou, Liming
    Chen, Fankai
    IEEE ACCESS, 2024, 12 : 9784 - 9791
  • [27] A UAV Aerial Image Target Detection Algorithm Based on YOLOv7 Improved Model
    Qin, Jie
    Yu, Weihua
    Feng, Xiaoxi
    Meng, Zuqiang
    Tan, Chaohong
    ELECTRONICS, 2024, 13 (16)
  • [28] YOLOD: A Target Detection Method for UAV Aerial Imagery
    Luo, Xudong
    Wu, Yiquan
    Zhao, Langyue
    REMOTE SENSING, 2022, 14 (14)
  • [29] YOLOv8-MPEB small target detection algorithm based on UAV images
    Xu, Wenyuan
    Cui, Chuang
    Ji, Yongcheng
    Li, Xiang
    Li, Shuai
    HELIYON, 2024, 10 (08)
  • [30] YOLOv7-P: a lighter and more effective UAV aerial photography object detection algorithm
    Sun, Fengxi
    He, Ning
    Wang, Xin
    Liu, Hongfei
    Zou, Yuxiang
    SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (11) : 8327 - 8335