Title: Research on small target pedestrian detection based on improved YOLO

Authors: Xing Xu; Kaiyao Wang; Yun Zhao

Addresses: School of Mechanical and Energy Engineering, Zhejiang University of Science and Technology, Hangzhou 310023, Zhejiang, China ' School of Mechanical and Energy Engineering, Zhejiang University of Science and Technology, Hangzhou 310023, Zhejiang, China ' School of Information and Electronic Engineering, Zhejiang University of Science and Technology, Hangzhou 310023, Zhejiang, China

Abstract: Aiming at the problems of low detection accuracy and speed for small target pedestrian in traffic scenes, the YOLO-SP based on YOLO-v4 is proposed. Firstly, the KITTI and INRIA data sets are used to make the new data set, and k-means algorithm is used to cluster the anchor box. Secondly, in the feature fusion phase (Neck), we increase the number of fused channels and simplify the number of output channels. Finally, to optimise the loss function, GIOU is used to calculate the coordinate loss, Focal is used to calculate the confidence loss. The test shows that all the improvement measures show better effect on small and overlapping pedestrians, the final detection accuracy (AP) is increased by 4.0%, the detection speed is accelerated by 11.3%. YOLO-SP has a certain practicality in the small target pedestrian detection.

Keywords: small target; pedestrian detection; YOLO; deep learning.

DOI: 10.1504/IJWMC.2021.115659

International Journal of Wireless and Mobile Computing, 2021 Vol.20 No.3, pp.281 - 289

Received: 09 Nov 2020
Accepted: 11 Dec 2020

Published online: 15 Jun 2021 *

Full-text access for editors Full-text access for subscribers Purchase this article Comment on this article