YOLO-Fast: a lightweight object detection model for edge devices
摘要
Lightweight convolutional neural networks have created new opportunities for object recognition, enabling high-performance algorithms to operate on resource-constrained devices while preserving strong representational and generalization capabilities. This paper introduces a lightweight YOLO-Fast object detection model. By replacing the backbone feature extraction network, redundant computations are minimized, thereby optimizing inference speed. An attention module is incorporated to enhance detection accuracy and feature fusion capability. The experimental findings based on the publicly accessible VisDrone dataset indicate that the proposed model has a 1.03% improvement in mAP over the baseline model. Through sparse training and channel pruning, the model’s inference speed is further optimized, while the number of network parameters is reduced. Moreover, knowledge distillation restores the network’s accuracy to pre-pruning levels. The model’s computational load is reduced by 35.5%, with a processing time of 12 ms per image on the embedded Atlas 200I development board. The code is available at: https://github.com/ZJ-Song-Lab/YOLO-Fast.