UAV Detection Based on Improved YOLOv4 Object Detection Model

Abstract

With the development of UAV technology, the improper use of UAVs has seriously endangered public safety. In order to meet the needs of UAV intrusion detection, this paper proposes a method for UAV image detection using deep learning technology. This method performs lightweight processing in the YOLOv4 object detection model to obtain a faster detection speed model, and uses the CA attention mechanism module to replace the original SE attention mechanism module to build a new model (called CA-YOLOv4-L) with stronger object detection capabilities. Use the improved model to test the UAV pictures and the results show that: The improved model CA-YOLOv4-L achieves an average accuracy of 94.63% on UAV images, a detection speed of 39.66FPS, and a 34.8% reduction in parameters compared to the original model. The proposed method can effectively identify the UAV in the picture, can identify multiple UAV targets, and estimate the position of the UAV in the image.