An Improved Yolov5 for Multi-Rotor UAV Detection

Abstract

Multi-rotor drones have a wide range of applications in practical scenarios; however, the use of multi-rotor drones for illegal acts is also on the rise, in order to improve the recognition accuracy of multi-rotor drones. A new multi-rotor drone detection algorithm is proposed. Firstly, the Yolov5 backbone is replaced with Efficientlite, thus reducing the number of parameters in the model. Secondly, adaptively spatial feature fusion is injected into the head of the baseline model to facilitate the fusion of feature maps with different spatial resolutions, in order to balance the accuracy loss caused by the lightweight of the model backbone. Finally, a constraint of angle is introduced into the original regression loss function to avoid the mismatch between the prediction frame and the real frame orientation during the training process in order to improve the speed of network convergence. Experiments show that the improved Yolov5s exhibits better detection performance, which provides a superior method for detecting multi-rotor UAVs in real-world scenarios.