Computer Science ›› 2023, Vol. 50 ›› Issue (11A): 220900050-8.doi: 10.11896/jsjkx.220900050

• Image Processing & Multimedia Technology • Previous Articles     Next Articles

Improved YOLOv5 Small Drones Target Detection Algorithm

LU Qi1, YU Yuanqiang1, XU Daoming1, ZHANG Qi 2   

  1. 1 School of Radar NCO,Air Force Early Warning Academy,Wuhan 430345,China
    2 Xi'an Satellite Control Center,Xi'an 710000,China
  • Published:2023-11-09
  • About author:LU Qi,born in 1994,master.His main research interests include intelligent detection and automatic target recognition.
  • Supported by:
    Consolidate Foundation Project of Airforce Early Warning Academy(HJGC-2022-001).

Abstract: The detection of low-altitude slow-speed small targets has always been the focus and difficulty in the field of early warning detection.At present,the mainstream target detection algorithms based on neural networks are mainly designed to be applied to VOC dataset or COCO dataset,and the detection accuracy is not ideal in specific scenarios.Aiming at the specific detection scene of small drones target detection in complex background,a small drones target detection algorithm based on improved YOLOv5 is proposed.First,a small target detection layer is added to obtain a large-sized shallow feature map,thereby improving the detection ability of the algorithm for small targets.Secondly,for the problem of different sizes of small drones,K-means++clustering algorithm is used to detect the prior frame.The size of the inspection frame is optimized and matched with each feature layer.Finally,the Mosaic-SOD methods of data augmentation and improved loss function are used to enhance the algorithm’s ability to perceive small targets and improve the efficiency of network training.The improved algorithm is applied to the target detection of small drones in complex background.Experimental results show that compared with the original YOLOv5 algorithm,the proposed algorithm has higher detection accuracy and characteristics in target detection of small rotor UAV.The extraction capability,although the detection speed has decreased to a certain extent,can still meet the real-time requirements by detecting the visible light video stream.

Key words: Low-Slow-Small target, Anti-UAV system, Deep learning, Small object detection, Small drones detection, Data augmentation, Feature fusion, YOLOv5(You Only Look Only version 5)

CLC Number: 

  • TP391.4
[1]GRANT R.RPAs FOR ALL[J].Air Force Magazine,2012,95(8):54-57
[2]XIE X.US Army releases anti-drone technology manual (Part 1)[J].Modern Military,2017(7):91-100.
[3]XIE X.US Army releases anti-drone technology manual (Part 2)[J].Modern Military,2017(8):85-89.
[4]GUELFI E A,BUDDHIKA J,ROBISON T.The Imperative for the U.S.Military to Develop a Counter-UAS Strategy[J].JFQ:Joint Force Quarterly,2020(97):4-12.
[5]MA W,CHIGAN X X.Research on Development of Anti-UAV Technology[J].Aero Weaponry,2020,27(6):19-24.
[6]JIE C,MIAO Z,YE T T.Research on the development of anti-UAV systems in U.S.military service[J].Airborne Missile,2020(12):36-42.
[7]CHEN W.A Survey on Low-Slow-Small UAV Countermeasure Equipment andTheoretical[J].Henan Science and Technology,2022,41(2):10-13.
[8]ZHU M Z,CHEN X,LIU X,et al.Situation and key technology of tactical laser anti-UAV[J].Infrared and Laser Engineering,2021,50:188-200.
[9]WU Y F.Research on LSS-Target(the Low altitude,Slowspeed andSmall target) in complex background [D].Changchun:Changchun university of Chinese Academy of Sciences,China(Chinese Academy of Sciences Institute of Optical Precision Machinery and Physical),2020.
[10]XU D M,ZHANG H w.Overview of Radar LSS Target Detection Technology[J].Modern Defence Technology,2018,46(1):148-155.
[11]MAO T.Research on Small Object Detection Algorithm Based onYOLOv5[D].Anhui University of Science and Technology,2021.
[12]REN J,WANG Z J,ZHANG Y F,et al.YOLOv5-R:lightweight real-time detection based on improved YOLOv5[J].Journal of Electronic Imaging,2022,31(3):033033.
[13]XIANG X Z,WANG Z Y,QIAO Y L.An Improved YOLOv5Crack Detection Method Combined with Transformer[J].IEEE Sensors Journal,2022,22(14):1
[14]JIANG X K,HU H C,LIU X,et al.A smoking behavior detection method based on the YOLOv5 network[J].Journal of Physics:Conference Series,2022,2232:012001.
[15]DONG X D,YAN S,DUAN C Q.A lightweight vehicles detection network model based on YOLOv5[J].Engineering Applications of Artificial Intelligence,2022,113:104914.
[1] ZHAO Mingmin, YANG Qiuhui, HONG Mei, CAI Chuang. Smart Contract Fuzzing Based on Deep Learning and Information Feedback [J]. Computer Science, 2023, 50(9): 117-122.
[2] XU Jie, WANG Lisong. Contrastive Clustering with Consistent Structural Relations [J]. Computer Science, 2023, 50(9): 123-129.
[3] LI Haiming, ZHU Zhiheng, LIU Lei, GUO Chenkai. Multi-task Graph-embedding Deep Prediction Model for Mobile App Rating Recommendation [J]. Computer Science, 2023, 50(9): 160-167.
[4] HUANG Hanqiang, XING Yunbing, SHEN Jianfei, FAN Feiyi. Sign Language Animation Splicing Model Based on LpTransformer Network [J]. Computer Science, 2023, 50(9): 184-191.
[5] CHEN Guojun, YUE Xueyan, ZHU Yanning, FU Yunpeng. Study on Building Extraction Algorithm of Remote Sensing Image Based on Multi-scale Feature Fusion [J]. Computer Science, 2023, 50(9): 202-209.
[6] ZHU Ye, HAO Yingguang, WANG Hongyu. Deep Learning Based Salient Object Detection in Infrared Video [J]. Computer Science, 2023, 50(9): 227-234.
[7] ZHANG Yian, YANG Ying, REN Gang, WANG Gang. Study on Multimodal Online Reviews Helpfulness Prediction Based on Attention Mechanism [J]. Computer Science, 2023, 50(8): 37-44.
[8] SONG Xinyang, YAN Zhiyuan, SUN Muyi, DAI Linlin, LI Qi, SUN Zhenan. Review of Talking Face Generation [J]. Computer Science, 2023, 50(8): 68-78.
[9] WANG Xu, WU Yanxia, ZHANG Xue, HONG Ruize, LI Guangsheng. Survey of Rotating Object Detection Research in Computer Vision [J]. Computer Science, 2023, 50(8): 79-92.
[10] ZHOU Ziyi, XIONG Hailing. Image Captioning Optimization Strategy Based on Deep Learning [J]. Computer Science, 2023, 50(8): 99-110.
[11] ZHANG Xiao, DONG Hongbin. Lightweight Multi-view Stereo Integrating Coarse Cost Volume and Bilateral Grid [J]. Computer Science, 2023, 50(8): 125-132.
[12] LIANG Jiayin, XIE Zhipeng. Text Paraphrase Generation Based on Pre-trained Language Model and Tag Guidance [J]. Computer Science, 2023, 50(8): 150-156.
[13] WANG Yu, WANG Zuchao, PAN Rui. Survey of DGA Domain Name Detection Based on Character Feature [J]. Computer Science, 2023, 50(8): 251-259.
[14] ZHOU Fengfan, LING Hefei, ZHANG Jinyuan, XIA Ziwei, SHI Yuxuan, LI Ping. Facial Physical Adversarial Example Performance Prediction Algorithm Based on Multi-modal Feature Fusion [J]. Computer Science, 2023, 50(8): 280-285.
[15] LI Kun, GUO Wei, ZHANG Fan, DU Jiayu, YANG Meiyue. Adversarial Malware Generation Method Based on Genetic Algorithm [J]. Computer Science, 2023, 50(7): 325-331.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!