Laser & Optoelectronics Progress, Volume. 59, Issue 12, 1215006(2022)

Research on Train Key Components Detection Based on Improved RetinaNet

Kai Yang1, Rui Li1、*, Lin Luo1, and Liming Xie2
Author Affiliations
  • 1School of Physical Science and Technology, Southwest Jiaotong University, Chengdu 610031, Sichuan , China
  • 2Chengdu Leading Technology Co., Ltd., Chengdu 610073, Sichuan , China
  • show less

    The key components of the train are essential for ensuring the safe operation of the train. The current detection algorithm based on deep learning has poor detectability under poor lighting conditions and small component size. To solve this problem, this study proposes a detection algorithm for key components of a train based on improved RetinaNet. First, a receptive field block module was introduced after shallow feature P3 to improve the receptive field and feature quality of the P3 feature layer. Then, the feature pyramid network was replaced with a pixel aggregation net and the positioning ability of the feature pyramid was enhanced by adding a bottom-up feature fusion path. Finally, by adjusting the experimental parameters and the location of the network detection layer, a network model suitable for detecting key components of the train was obtained. Results show that the proposed model is superior to the original RetinaNet in the open dataset PASCAL VOC. Furthermore, it is superior to the current mainstream algorithm in detecting the key components of the train.

    Tools

    Get Citation

    Copy Citation Text

    Kai Yang, Rui Li, Lin Luo, Liming Xie. Research on Train Key Components Detection Based on Improved RetinaNet[J]. Laser & Optoelectronics Progress, 2022, 59(12): 1215006

    Download Citation

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

    Category: Machine Vision

    Received: May. 6, 2021

    Accepted: Jun. 19, 2021

    Published Online: May. 23, 2022

    The Author Email: Li Rui (14781866710@163.com)

    DOI:10.3788/LOP202259.1215006

    Topics