Acta Optica Sinica, Volume. 41, Issue 15, 1515001(2021)

Occluded Pedestrian Detection Algorithm Based on Attention Mechanism

Ziyin Zou1,2, Shaoyan Gai1,2、*, Feipeng Da1,2,3, and Yu Li1,2
Author Affiliations
  • 1School of Automation, Southeast University, Nanjing, Jiangsu 210096, China
  • 2Key Laboratory of Measurement and Control of Complex Systems of Engineering, Ministry of Education, Southeast University, Nanjing, Jiangsu 210096, China
  • 3Shenzhen Research Institute, Southeast University, Shenzhen, Guangdong 518063, China
  • show less

    In light of the situation that it is difficult to accurately detect pedestrians in real scenes due to mutual occlusion, a feature extraction enhanced detection algorithm based on attention mechanism is proposed. Firstly, attention modules are added to learn the relationship between feature channels and the spatial information of feature maps, so as to enhance feature extraction in the visual area of pedestrian targets. Secondly, according to the actual size of pedestrian data, the k-means++ algorithm is used to cluster pedestrian labels, so as to determine the size and proportion of anchors. Distance-intersection over union loss function (DIOULoss) is used to design the loss function of the detector, so that the regression of the detection box pays more attention to the intersection over union between the candidate box and the real box, as well as the center distance between the two boxes. Finally, a new non-maximum suppression algorithm (DSoft-NMS) is presented to preserve more accurate prediction boxes. The proposed method has been tested on CityPersons and WiderPerson datasets, and the results show that the proposed method with a simple network structure has higher detection accuracy in occluded pedestrian detection, which is convenient for subsequent research.

    Tools

    Get Citation

    Copy Citation Text

    Ziyin Zou, Shaoyan Gai, Feipeng Da, Yu Li. Occluded Pedestrian Detection Algorithm Based on Attention Mechanism[J]. Acta Optica Sinica, 2021, 41(15): 1515001

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Machine Vision

    Received: Jan. 13, 2021

    Accepted: Mar. 8, 2021

    Published Online: Aug. 11, 2021

    The Author Email: Gai Shaoyan (qxxymm@163.com)

    DOI:10.3788/AOS202141.1515001

    Topics