Lightweight Detection Algorithm for Air-to-Ground Threat Object

Abstract In air-to-ground threat object detection, the typically strong concealment, small size, and high movement speed of threat objects often lead to challenges such as missed detections, low accuracy, and an excessive number of network parameters. An efficient lightweight detection algorithm for air-to-ground threat objects is proposed, built upon the YOLOv7-tiny framework. To address the issue of large network parameters, lightweight modules ELAN-GS and Slim-Neck are employed. Additionally, to refine the detection precision for small objects, the NWD metric and NWD loss function are introduced. Results from experiments on the DOTA dataset highlight that, compared with the original network, the proposed algorithm achieves a 33.2% reduction in parameters and a 26.3% reduction in computational cost. The small object accuracy, [email protected]:0.95, is 23.7%, which is a 4.4% enhancement relative to the original network. The significant reduction in parameters and computational cost, along with the substantial improvement in small object detection accuracy, demonstrate the effectiveness of the proposed algorithm.