UAV Small Target Detection Based on XMB-YOLOv5s

doi:10.3969/j.issn.1006-2475.2025.04.005

Abstract

Abstract: From the drone viewpoint， the detection of dense， small targets faces various shortcomings， such as low accuracy， false detection of certain targets， and missed detections. To address these issues， this paper proposes a drone-based small target detection technique using XMB-YOLOv5s. Firstly， a small target detection layer is adopted for more effective extraction and utilization of detail information within the image. Secondly， the structured embedding of BottleneckCSP and C3TR modules is used to update the C3 module to reduce computational complexity and improve training efficiency. Subsequently， the integration of the CBAM attention mechanism enables the network to better recognize and process features， thus enhancing image recognition accuracy. Finally， the Focal-EIoU Loss is employed to resolve the insensitivity of the CIoU Loss to small target detection. The experimental results indicate that， compared with traditional YOLOv5s algorithm， the XMB-YOLOv5s algorithm has increased P by 4.6 percentage points， R by 4.4 percentage points， mAP50 by 4.9 percentage points， mAP75 by 5.1 percentage points， mAP50-95 by 4 percentage points on the VisDrone2019 data set， providing a novel approach for small target detection in drone applications.

Key words: unmanned aerial vehicles （UAV）, deep learning, object detection, machine vision, XMB-YOLOv5s

CLC Number:

TP391
TP399

ZHUANG Yu, FU Xiaojin, LI Sha, WU Zheng. UAV Small Target Detection Based on XMB-YOLOv5s[J]. Computer and Modernization, 2025, 0(04): 29-35.

References

［1］ WANG Q， FENG W Q， YAO L F， et al. TPH-YOLOv5-Air： Airport confusing object detection via adaptively spatial feature fusion［J］. Remote Sensing， 2023，15（15）. DOI: 10.3390/rs15153883.
［2］ ZHAO Q， LIU B H， LYU S C， et al. TPH-YOLOv5++： Boosting object detection on drone-captured scenarios with cross-layer asymmetric transformer［J］. Remote Sensing， 2023，15（6）. DOI: 10.3390/rs15061687.
［3］赵浩权. 面向无人机图像的小目标检测方法研究［D］. 长春：吉林大学， 2023.
［4］ YANG R J， LI W F， SHANG X N， et al. KPE-YOLOv5： An improved small target detection algorithm based on YOLOv5［J］. Electronics， 2023，12（4）. DOI: 10.3390/electronics12040817.
［5］ ZHU Z， LIANG D， ZHANG S H， et al. Traffic-sign detection and classification in the wild［C］// Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition. IEEE， 2016：2110-2118.
［6］王淑青，张鹏飞，要若天，等. 基于改进YOLOv5的太阳能电池片表面缺陷检测［J］. 仪表技术与传感器， 2022（5）：111-116.
［7］张锦，屈佩琪，孙程，等. 基于改进YOLOv5的安全帽佩戴检测算法［J］. 计算机应用， 2022，42（4）：1292-1300.
［8］ LIU Z， GAO X H， WAN Y， et al. An improved YOLOv5 method for small object detection in UAV capture scenes［J］. IEEE Access， 2023，11：14365-14374.
［9］彭自然，张颖清，肖伸平. 基于YOLOv5的太阳电池表面缺陷检测［J/OL］. 太阳能学报：1-8（2023-12-20）［2024-03-10］. http：//kns.cnki.net/kcms/detail/11.2082.TK.20231219.
1650.002.html.
［10］ REDMON J， DIVVALA S， GIRSHICK R， et al. You only look once： Unified， real-time object detection［C］// Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition. IEEE， 2016：779-788.
［11］ WANG H， JIN Y， KE H C， et al. DDH-YOLOv5： Improved YOLOv5 based on double IoU-aware decoupled head for object detection［J］. Journal of Real-time Image Processing， 2022，19（6）：1023-1033.
［12］王文胜，李继旺，吴波，等. 基于YOLOv5交通标志识别的智能车设计［J］. 国外电子测量技术， 2021，40（10）：158-164.
［13］张宏群，班勇苗，郭玲玲，等. 基于YOLOv5的遥感图像舰船的检测方法［J］. 电子测量技术， 2021，44（8）：87-92.
［14］吕禾丰，陆华才. 基于YOLOv5算法的交通标志识别技术研究［J］. 电子测量与仪器学报， 2021，35（10）：137-144.
［15］ ARIFANDO R， ETO S， WADA C， et al. Improved YOLOv5-based light weight object detection algorithm for people with visual impairment to detect buses［J］. Applied Sciences， 2023，13（9）. DOI: 10.3390/app13095802.
［16］罗伟，刘思远，徐健祥，等. 基于改进YOLOv5s的太阳能电池缺陷检测算法［J］. 计算机与现代化， 2023（7）：119-126.
［17］ LIN T Y， MAIRE M， BELONGIE S， et al. Microsoft COCO： Common objects in context［C］// Proceedings of the 13th European Conference on Computer Vision. Springer， 2014：740-755.
［18］傅强，卢志伟，潘磊. CC-YOLOv5：改进的YOLOv5用于无人机航拍图像目标检测［J］. 现代计算机， 2023，29（7）：16-22.
［19］任欢，王旭光. 注意力机制综述［J］. 计算机应用， 2021，41（增1）：1-6.
［20］祁宣豪，智敏. 图像处理中注意力机制综述［J］. 计算机科学与探索， 2024，18（2）：345-362.
［21］任书玉，汪晓丁，林晖. 目标检测中注意力机制综述［J］. 计算机工程， 2024，50（12）：16-32.
［22］ WOO S， PARK J， LEE J Y， et al. CBAM： Convolutional block attention module［J］. arXiv preprint arXiv：1807.06521，
2018.
［23］张德春，李海涛，李勋，等. 基于CBAM和BiFPN改进YoloV5的渔船目标检测［J］. 渔业现代化， 2022，49（3）：71-80.
［24］ ZHENG Z H， WANG P， LIU W， et al. Distance-IoU loss： Faster and better learning for bounding box regression［C］// Proceedings of the 34th AAAI Conference on Artificial Intelligence / the 32nd Innovative Applications of Artificial Intelligence Conference / the 10th AAAI Symposium on Educational Advances in Artificial Intelligence. AAAI， 2020：12993-13000.
［25］ GAO J F， CHEN Y， WEI Y M， et al. Detection of specific building in remote sensing images using a novel YOLO-S-CIOU model case： Gas station identification［J］. Sensors， 2021，21（4）. DOI: 113390/s21041375.
［26］张乾闯，郭晨霞，杨瑞峰，等. 基于改进YOLO算法的光纤环绕制缺陷检测［J］. 电子测量技术， 2023，46（10）：32-39.
［27］ ZHANG Y F， REN W Q， ZHANG Z， et al. Focal and efficient IOU loss for accurate bounding box regression［J］. Neurocomputing， 2022，506：146-157.
［28］ DU D W， ZHU P F， WEN L Y， et al. VisDrone-DET2019： The vision meets drone object detection in image challenge results［C］// Proceedings of the 2019 IEEE/CVF International Conference on Computer Vision Workshop. IEEE， 2019：213-226.

[1]	LI Kai, JIN Yunpeng, LI Haiyang, KONG Shasha, YANG Peng, FANG Chengwu, HUANG Xiangjie, HAN Yaosheng, LI Chunmei. AGP Calculation Methods in UAV Imagery Based on Image Segmentation [J]. Computer and Modernization, 2025, 0(04): 83-88.
[2]	WANG Jiale, SONG Wenai, FU Lizhen. Occupational Pneumoconiosis Screening Based on HA-Net Model [J]. Computer and Modernization, 2025, 0(04): 103-110.
[3]	TANG Rui1, WU Jianchao1, CHEN Jianbo1, CHAI Jiang1, WANG Qian1, HE Yuchen2. Improved YOLOv8s Algorithm Based on GiraffeDet for Transmission Line Icing Detection [J]. Computer and Modernization, 2025, 0(03): 6-11.
[4]	SHAO Yeqin1, WANG Haiquan2, ZHOU Kunyang3, GUO Yudi2, SHI Quan1. One-stage Semi-supervised Object Detection by Reusing Unreliable Pseudo-labels [J]. Computer and Modernization, 2025, 0(03): 52-59.
[5]	LI Haoran1, HE Wenxue1, XU Jiazhen1, YANG Banghua2. Classification Method of EEG Signals for Depression Based on Multi-Scale Dynamic Convolution and Attention Mechanism [J]. Computer and Modernization, 2025, 0(03): 60-65.
[6]	PU Yaya, WANG Yanbo, SU Yongdong, XU Zhongcheng. Multi-scale Feature Image Defogging Algorithm Based on Content-guided Attention Fusion [J]. Computer and Modernization, 2025, 0(03): 78-85.
[7]	LUO Hao, LI Xianfeng. Remote Sensing Image Classification Based on Multi-scale Feature Extraction [J]. Computer and Modernization, 2025, 0(03): 86-92.
[8]	LIU Yaokai, REN Dejun, LIU Chongyi, LU Yudong. AMDFF-Net： Adaptive Multi-dimensional Feature Fusion Network for Tiny Object Detection [J]. Computer and Modernization, 2025, 0(03): 106-112.
[9]	ZHANG Yue, GUO Zixin, HUANG Yibin, YAN Tao. Network Intrusion Detection Method Based on Convolutional Neural Networks with convLSTM [J]. Computer and Modernization, 2025, 0(03): 119-126.
[10]	LIU Chongyi, LI Hua, REN Dejun, LIU Yaokai, WANG Yulong. Anomaly Detection Algorithm Based on Bidirectional Multi-scale Knowledge Distillation [J]. Computer and Modernization, 2025, 0(02): 58-63.
[11]	HE Guotao1, ZHAO Chunhui2, LIU Zhenyu1, WANG Long1. Optimization for Camera Self-calibration Based on Horizon Detection in Road Scenes [J]. Computer and Modernization, 2025, 0(02): 77-85.
[12]	ZHAO Yin, YIN Siqing, ZHANG Yonglai. Improved Traffic Sign Detection Algorithm of YOLOv7 [J]. Computer and Modernization, 2025, 0(02): 94-99.
[13]	HE Feixiong1, XIE Haiwei1, PU Chao2, ZOU Chuanming2, JIA Yixuan1. A New Method of Pavement Disease Detection Based on Improved YOLOv8 [J]. Computer and Modernization, 2025, 0(02): 108-113.
[14]	WANG Peng1, NI Bin1, GUO Zhuangzhuang1, ZHANG Shusheng1, WANG Zhi1, CAI Runkai2. Identification of Typical Defects in Key Components of Overhead Lines Based on Improved YOLOv5 [J]. Computer and Modernization, 2025, 0(02): 114-120.
[15]	YAN Xiaoqi, PENG Yiqing, REN Xiaoling. Point Cloud Data Classification Method of PointNet++ with Position Adaptive Convolution [J]. Computer and Modernization, 2025, 0(01): 44-49.