YOLO-edge: real-time vehicle detection for edge devices
摘要
To achieve real-time detection on resource-constrained edge devices, a lightweight vehicle detection algorithm named YOLO-edge was developed based on the YOLOv5s framework. In YOLO-edge, a slim neck was designed to reduce the computational cost in the feature channel fusion process. A modified fast spatial pyramid pooling technique enhances feature extraction efficiency. A multi-scale feature fusion architecture with a broad receptive field integrates shallow and deep features, preserving details for comprehensive feature representation. The loss function was optimized to improve the accuracy of bounding box predictions. Experimental results demonstrate that YOLO-edge achieves a 5.1% increase in mean average precision (mAP) compared to YOLOv5s. Real-time detection rates of 34 and 47 frames per second (FPS) are achieved on edge devices Jetson TX2 and Jetson Orin NX, respectively. YOLO-edge exhibited superior speed and accuracy on edge devices, outperforming state-of-the-art YOLO methods in vehicle detection.