• Laser & Optoelectronics Progress
  • Vol. 58, Issue 22, 2212001 (2021)
Hui Luo, Jian Li*, and Chen Jia
Author Affiliations
  • School of Information Engineering, East China JiaoTong University, Nanchang, Jiangxi 330013, China
  • show less
    DOI: 10.3788/LOP202158.2212001 Cite this Article Set citation alerts
    Hui Luo, Jian Li, Chen Jia. Rail Surface Defect Detection Based on Image Enhancement and Improved Cascade R-CNN[J]. Laser & Optoelectronics Progress, 2021, 58(22): 2212001 Copy Citation Text show less
    Overall structure of the method in this paper
    Fig. 1. Overall structure of the method in this paper
    Improved Retinex rail surface image enhancement algorithm framework
    Fig. 2. Improved Retinex rail surface image enhancement algorithm framework
    Overall structure diagram of rail surface defect detection algorithm
    Fig. 3. Overall structure diagram of rail surface defect detection algorithm
    RoIAlign diagram
    Fig. 4. RoIAlign diagram
    Examples of rail surface defect image data augmentation
    Fig. 5. Examples of rail surface defect image data augmentation
    Changes of training losses based on improved Cascade R-CNN+ResNet-50
    Fig. 6. Changes of training losses based on improved Cascade R-CNN+ResNet-50
    Contrast diagrams of rail surface image enhancement effect
    Fig. 7. Contrast diagrams of rail surface image enhancement effect
    Comparison diagrams of rail surface defect detection result
    Fig. 8. Comparison diagrams of rail surface defect detection result
    Experiment datasetAP /%
    Original dataset91.21
    Augmentation dataset95.74
    Table 1. Experimental results of data augmentation effectiveness
    AlgorithmAP /%
    --95.74
    HE92.56
    CLAHE95.62
    MSR96.13
    Improved Retinex96.61
    Table 2. Experimental results of detection performance for different image enhancement algorithms
    BackboneImage enhancementIoU balanced samplingRoIAlignCIoUAP /%Time /ms
    ResNet-50--------95.74151.8
    ------96.61145.0
    ----97.03145.1
    ----97.66146.2
    ----97.83145.0
    --97.85146.3
    --98.19145.1
    --98.42146.2
    98.75146.3
    ResNet-101--------96.23170.5
    ------96.95163.2
    ----97.34163.3
    ----97.82164.5
    ----98.23163.2
    --98.12164.6
    --98.41163.3
    --98.74164.5
    98.96164.6
    Table 3. Experimental results of different architectures and improved algorithms
    AlgorithmBackboneAP /%Time /ms
    SSDResNet-10181.2358.3
    YOLOv3Darknet-5385.5439.5
    YOLOv4CSPDarknet-5390.3738.4
    Faster R-CNNResNet-10194.43153.6
    R-FCNResNet-10187.6395.8
    Cascade R-CNNResNet-10196.23170.5
    Proposed methodResNet-5098.75146.3
    Table 4. Performance comparison of different algorithms
    Hui Luo, Jian Li, Chen Jia. Rail Surface Defect Detection Based on Image Enhancement and Improved Cascade R-CNN[J]. Laser & Optoelectronics Progress, 2021, 58(22): 2212001
    Download Citation