基于改进 YOLOv5 算法的复杂场景交通目标检测.

Real-time target detection in traffic scenarios is the prerequisite of electronic monitoring, automatic driving, and other functions. In view of the low detection efficiency of existing target detection algorithms and the low accuracy of most light target detection algorithms, which are easy to obtain wrong or insufficient target detection, this paper adopts the improved YOLOv5 target detection algorithm for model training, and the pseudo-label strategy for training process optimization. Then, the KITTI traffic target dataset tags are merged into three categories for model training and testing. Through the experimental comparison, the improved YOLOv5 model in this paper achieves 92.5% mAP in all categories, which is 3% higher than the original YOLOv5 training model. Finally, the three categories of the trained models are deployed on the Jetson Nano embedded platform for inference testing, and TensorRT is used to accelerate inference. The average inference time per frame of image is 77 ms, which meets the goal of real-time detection. [ABSTRACT FROM AUTHOR]