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 *