Abstract:Lightweight object detection algorithms play a paramount role in unmanned aerial vehicles (UAVs) remote sensing. However, UAV remote sensing requires target detection algorithms to have higher inference speeds and greater accuracy in detection. At present, most lightweight object detection algorithms have achieved fast inference speed, but their detection precision is not satisfactory. Consequently, this paper presents a refined iteration of the lightweight object detection algorithm to address the above issues. The MobileNetV3 based on the efficient channel attention (ECA) module is used as the backbone network of the model. In addition, the focal and efficient intersection over union (FocalEIoU) is used to improve the regression performance of the algorithm and reduce the false-negative rate. Furthermore, the entire model is pruned using the convolution kernel pruning method. After pruning, model parameters and floating-point operations (FLOPs) on VisDrone and DIOR datasets are reduced to 1.2 M and 1.5 M and 6.2 G and 6.5 G, respectively. The pruned model achieves 49 frames per second (FPS) and 44 FPS inference speeds on Jetson AGX Xavier for VisDrone and DIOR datasets, respectively. To fully exploit the performance of the pruned model, a plug-and-play structural re-parameterization fine-tuning method is proposed. The experimental results show that this fine-tuned method improves mAP@0.5 and mAP@0.5:0.95 by 0.4% on the VisDrone dataset and increases mAP@0.5:0.95 by 0.5% on the DIOR dataset. The proposed algorithm outperforms other mainstream lightweight object detection algorithms (except for FLOPs higher than SSDLite and mAP@0.5 Below YOLOv7 Tiny) in terms of parameters, FLOPs, mAP@0.5, and mAP@0.5:0.95. Furthermore, practical validation tests have also demonstrated that the proposed algorithm significantly reduces instances of missed detection and duplicate detection.

R-TOSS: A Framework for Real-Time Object Detection using Semi-Structured Pruning

A Compression Pipeline for One-Stage Object Detection Model

TOD: Transprecise Object Detection to Maximise Real-Time Accuracy on the Edge

Achieving Real-Time Object Detection on MobileDevices with Neural Pruning Search

YOLOv4-5D: An Effective and Efficient Object Detector for Autonomous Driving

A Real-Time Object Detector for Autonomous Vehicles Based on YOLOv4

Pruning DETR: efficient end-to-end object detection with sparse structured pruning

YOLOv4-dense: A smaller and faster YOLOv4 for real-time edge-device based object detection in traffic scene

DETRs Beat YOLOs on Real-time Object Detection

YOLObile: Real-Time Object Detection on Mobile Devices via Compression-Compilation Co-Design

Real-Time Small Drones Detection Based on Pruned YOLOv4

Vehicle and Pedestrian Detection Based on Improved YOLOv7-Tiny

YOLO-RDP: Lightweight Steel Defect Detection through Improved YOLOv7-Tiny and Model Pruning

An Efficient Real-Time Object Detection Framework on Resource-Constricted Hardware Devices via Software and Hardware Co-design

ABCP: Automatic Block-wise and Channel-wise Network Pruning via Joint Search

Optimization of the Pedestrian and Vehicle Detection Model based on Cloud-Edge Collaboration

Re-Parameterization After Pruning: Lightweight Algorithm Based on UAV Remote Sensing Target Detection

Real-Time Object Detection Based on YOLO-v2 for Tiny Vehicle Object

Developing a Compressed Object Detection Model based on YOLOv4 for Deployment on Embedded GPU Platform of Autonomous System

L1RR: Model Pruning Using Dynamic and Self-Adaptive Sparsity for Remote-Sensing Target Detection to Prevent Target Feature Loss

Embedded YOLO: A Real-Time Object Detector for Small Intelligent Trajectory Cars