YOLOdrive: A Lightweight Autonomous Driving Single-Stage Target Detection Approach
Liya Wang, Shaona Hua, Chunying Zhang, Guanghui Yang, Jing Ren, Jie Li · IEEE Internet of Things Journal · 2024
With the continuous development of autonomous driving, real-time target detection has become increasingly critical in autonomous driving systems. However, traditional target detection algorithms usually require huge computational resources, limiting their application on embedded autonomous driving platforms. To address this challenge, a lightweight single-stage target detection algorithm is proposed YOLOdrive. The inverted residual structure, linear bottleneck layer, and depth-separable convolution in MobileNetv2 are utilized to improve the YOLOv8 backbone network, while the spatial channel reconstructed convolution is used to improve the C2f module of YOLOv8, and the convolution of YOLOv8 neck and detection head is replaced by the depth-separable convolution. Experimental results verify that the average accuracy of YOLOdrive algorithm on MS COCO2017 data set and VOC2007 data set is improved compared with the baseline model YOLOv8-Nano. The amount of model parameters has been reduced by more than 50%, and the amount of computation in the model has been reduced by more than 70%. The algorithm drastically reduces the amount of parameters and computational complexity of the network, improves the operational efficiency, saves the storage space of the network, and maintains a high-detection performance.