Pedestrian detection based on one-stage YOLO algorithm

Abstract

Pedestrian detection is mostly used in autonomous driving scenarios, which require high real-time detection. Most of the existing algorithms are based on Two-Stage detection, with poor real-time performance. This paper proposes a pedestrian detection system based on One-Stage, and uses One-Stage-based YOLO-Tiny, YOLO and YOLO-SPP algorithms to test and analyze the system in the scene of detecting pedestrians of different sizes. The results show that YOLO and YOLO-SPP have high average confidence, and YOLO-Tiny has a fast detection speed, which is suitable for fast detection scenarios.