Real-time Lightweight Target Detection Network under Autonomous Driving

Abstract

Target detection in traffic scenes has been a focal point concerning computer vision, especially in the context of autonomous driving, including the recognition of vehicles, pedestrians, and traffic signs. However, the computational power of devices operating in traffic scenarios is limited, placing stringent demands on computational resources and latency. To address these challenges, this study proposed a lightweight detection algorithm, GV2-YOLO, which prioritizes both speed and accuracy. First, we apply the GhostNetv2 architecture and incorporate the Ghost module to decrease the parameters of the backbone feature extraction network. Second, the algorithm also integrates SPPF and Slim-neck by GSConv techniques, which effectively reduces the computational complexity while ensuring the resulting significant reduction in detection accuracy. Our proposed algorithm achieves a mAP of 84.7% on the CCTSDB dataset. The algorithm has a total parameter count of 9.1 M, making it ideal for deployment in embedded autonomous driving platforms.