YOLO-LIRTU: a lightweight infrared real-time UAV detection framework
摘要
The rapid proliferation of unmanned aerial vehicles (UAVs) has brought many valuable applications, but has also introduced significant security concerns, which require robust detection frameworks. Infrared imaging provides a viable solution for UAV detection under low-visibility conditions, yet existing approaches suffered from high computational costs and inefficiencies in detecting small targets. This work presents YOLO-LIRTU, which is a lightweight framework designed to address hardware constraints, small-target feature degradation, and parameter redundancy in infrared UAV detection. By integrating an optimized ShuffleNetV2 backbone with depthwise separable convolutions and channel shuffle mechanisms, a dynamic multi-scale feature enhancement module, and a decoupled feature generation technique using GhostConv operations, the framework achieves notable computational efficiency while maintaining detection accuracy. The redesigned backbone reduces parameter count through spatial feature reorganization, while the dynamic downsampling strategy preserves critical details for distant targets. Evaluated on the Anti-UAV dataset containing 211,527 infrared images, YOLO-LIRTU requires only 1.73M parameters and 7 GFLOPs, delivering 76.04%