Pedestrian detection based on attention mechanism and feature enhancement with SSD

被引:0
作者
Feng, T. T. [1 ]
Ge, H. Y. [1 ]
机构
[1] Donghua Univ, Sch Informat Sci & Technol, Shanghai, Peoples R China
来源
2020 5TH INTERNATIONAL CONFERENCE ON COMMUNICATION, IMAGE AND SIGNAL PROCESSING (CCISP 2020) | 2020年
关键词
SSD; CBAM; pedestrian detection; feature enhancement;
D O I
10.1109/ccisp51026.2020.9273507
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
Some factors such as low resolution of small targets, limited target features, and noise interference affect the effect of pedestrian detection. Therefore, a pedestrian detection algorithm based on attention mechanism and feature enhancement with SSD is presented in this paper. It uses channel feature fusion to fuse non-adjacent convolutional layers to obtain significant edge gradient features and semantic information features. Finally, through the optimization of the attention mechanism CBAM, the channel features and spatial features are coupled under different fusion detection layers to improve the feature weight of the pedestrian's salient region, so as to the detection accuracy of the algorithm has taken a big step forward. The improved algorithm is verified on the fusion dataset and the VOC2007TEST dataset, compared with the SSD algorithm, the detection accuracy of the improved algorithm model on the fusion dataset reaches 60.1%, with an increase of 7.2%, and that on the VOC2007TEST dataset reaches 78%, with an increase of 4.1%. The experimental results show that this method can effectively detect the small target pedestrian in the image, and reduce the error detection and omission detection, thus verifying its feasibility.
引用
收藏
页码:145 / 148
页数:4
相关论文
共 11 条
  • [1] Cheng YF, 2017, COMPUTER VISION PATT
  • [2] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [3] Hai T Z, 2020, COMPUTER ENG, P1
  • [4] Jin P L., 2020, SCI TECHNOLOGY ENG, P1498
  • [5] SSD: Single Shot MultiBox Detector
    Liu, Wei
    Anguelov, Dragomir
    Erhan, Dumitru
    Szegedy, Christian
    Reed, Scott
    Fu, Cheng-Yang
    Berg, Alexander C.
    [J]. COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 : 21 - 37
  • [6] Mo R J, 2020, J OPTICS, P1
  • [7] You Only Look Once: Unified, Real-Time Object Detection
    Redmon, Joseph
    Divvala, Santosh
    Girshick, Ross
    Farhadi, Ali
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 779 - 788
  • [8] Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
    Ren, Shaoqing
    He, Kaiming
    Girshick, Ross
    Sun, Jian
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (06) : 1137 - 1149
  • [9] Face detection using deep learning: An improved faster RCNN approach
    Sun, Xudong
    Wu, Pengcheng
    Hoi, Steven C. H.
    [J]. NEUROCOMPUTING, 2018, 299 : 42 - 50
  • [10] Tao W, 2020, SCI TECHNOLOGICAL IN, P62