• Laser & Optoelectronics Progress
  • Vol. 59, Issue 12, 1215006 (2022)
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
    DOI: 10.3788/LOP202259.1215006 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

    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.
    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