Laser & Optoelectronics Progress, Volume. 59, Issue 12, 1215002(2022)
Improved Tiny YOLOv4 Algorithm and Its Application in Pedestrian Detection
To solve the problem that the Tiny YOLOv4 target detection algorithm has low accuracy and low recall rate in pedestrian detection, the feature extraction network and prediction network are improved. In the part of feature extraction network, the traditional convolution network is replaced by a depthwise separable convolution network to reduce parameters and computation. The attention mechanism module is added in the feature extraction network to enhance the area of interest of detecting object and improve the detection accuracy. A prediction scale is added in the prediction network, and the added scale is enhanced by features to improve the recall of detection of objects. The experimental results show that compared with the original algorithm, the improved Tiny YOLOv4 algorithm improves the accuracy by 7.1%, and the recall rate also increases by 6.6%.
Get Citation
Copy Citation Text
Yong Xuan, Chao Han, Wenhan Sha. Improved Tiny YOLOv4 Algorithm and Its Application in Pedestrian Detection[J]. Laser & Optoelectronics Progress, 2022, 59(12): 1215002
Category: Machine Vision
Received: Apr. 14, 2021
Accepted: Jun. 2, 2021
Published Online: May. 23, 2022
The Author Email: Han Chao (hanchaozh@126.com)