Detection Algorithm of Pedestrian Shoe Area Based on Improved YOLOv4

被引:2
作者
Yang Zhixiong [1 ]
Tang Yunqi [1 ]
Zhang Jiajun [1 ]
Geng Pengzhi [1 ]
机构
[1] Peoples Publ Secur Univ China, Sch Invest, Beijing 100038, Peoples R China
关键词
image processing; shoes detection; YOLOv4; feature fusion; spatial pyramid pooling; video surveillance;
D O I
10.3788/LOP202259.0810007
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
One of the important tactics used by the public security bureau in a criminal investigation is to combine the related surveillance video and shoeprints on the spot to identify criminal suspects. However, the low level automation of such a method is so labor-intensive and time-consuming, which limits its application. Therefore, this paper proposes an object detection method based on the YOLOv4 algorithm to realize the automatic detection of pedestrian shoes in surveillance video. According to the characteristics of the pedestrian shoe area, first, the K-means clustering algorithm is used to determine the scale of the anchor box and confirm its quantity; second, an appropriate detection layer was selected based on the datasets in this paper to improve the learning of shoe features; finally, a multifeature fusion method is used and the adjusted spatial pyramid pooling structure is transferred into the pruned network to improve the learning ability of the model. The experimental results demonstrate that the training weight of the YOLOv4_shoe algorithm proposed is only 39. 56 MB, which is approximately one-sixth of the original model; and its mean average precision reaches 97. 93%, which is 2. 07% higher than that of the original YOLOv4 model.
引用
收藏
页数:10
相关论文
共 26 条
[1]  
[Anonymous], 2017, GUANGDONG GONGAN KEJ
[2]  
Bochkovskiy A., 2020, PREPRINT
[3]   Histograms of oriented gradients for human detection [J].
Dalal, N ;
Triggs, B .
2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, :886-893
[4]  
Felzenszwalb P, 2008, 2008 IEEE C COMPUTER
[5]   Pedestrian Shoes Detection Algorithm Based on SSD [J].
Geng Pengzhi ;
Yang Zhixiong ;
Zhang Jiajun ;
Tang Yunqi .
LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (06)
[6]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[7]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[8]  
He K., 2017, P IEEE INT C COMPUTE, P2980, DOI [DOI 10.1109/ICCV.2017.322, 10.1109/ICCV.2017.322]
[9]  
He KM, 2014, LECT NOTES COMPUT SC, V8691, P346, DOI [arXiv:1406.4729, 10.1007/978-3-319-10578-9_23]
[10]   Surface Defect Detection of Aeroengine Components Based on Improved YOLOv4 Algorithm [J].
Li Bin ;
Wang Cheng ;
Wu Jing ;
Liu Jichao ;
Tong Lijia ;
Guo Zhenping .
LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (14)