Improvement of Pedestrian Detection Algorithm Based on YOLO

被引:7
作者
Li, Xuan [1 ]
Li, Jing [1 ]
机构
[1] Shenyang Aerosp Univ, Sch Elect Informat Engn, Shenyang 110136, Liaoning, Peoples R China
来源
2019 2ND INTERNATIONAL CONFERENCE ON MECHANICAL ENGINEERING, INDUSTRIAL MATERIALS AND INDUSTRIAL ELECTRONICS (MEIMIE 2019) | 2019年
关键词
deep learning; object detection; detection boxes; penalty factor; YOLO V2;
D O I
10.23977/meimie.2019.43009
中图分类号
TH [机械、仪表工业];
学科分类号
0802 ;
摘要
The non-maximum suppression algorithm is usually used in the post-position of deep learning object detection algorithm. It suppresses the detection boxes with high overlap rate while using the algorithm. In order to avoid the missed and false detection caused by non-maximum suppression algorithm, an improved maximum value suppression algorithm is proposed. When the IOU of the suppression window and the suppressed window is greater than the given threshold, the confidence multiply by the penalty factor instead of discarding it directly. After multiple iterations, we need to remove the lower scores detection boxes. Experiments show that the YOLO V2 deep learning model with improved algorithm has improved accuracy on different data sets as well as strong versatility and robustness.
引用
收藏
页码:44 / 49
页数:6
相关论文
共 8 条
[1]  
[Anonymous], 2017, P IEEE C COMP VIS PA
[2]  
[Anonymous], P C WORKSH NEUR INF
[3]  
[Anonymous], 2016 IEEE Conf. Comp. Vis. Patt. Recog. (CVPR)
[4]  
[Anonymous], 2015, NEURAL INFORM PROCES
[5]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[6]   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
[7]   SSD: Single Shot MultiBox Detector [J].
Liu, Wei ;
Anguelov, Dragomir ;
Erhan, Dumitru ;
Szegedy, Christian ;
Reed, Scott ;
Fu, Cheng-Yang ;
Berg, Alexander C. .
COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 :21-37
[8]   Selective Search for Object Recognition [J].
Uijlings, J. R. R. ;
van de Sande, K. E. A. ;
Gevers, T. ;
Smeulders, A. W. M. .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2013, 104 (02) :154-171