Laser & Optoelectronics Progress, Volume. 58, Issue 16, 1610020(2021)

Anchor-Free Traffic Sign Recognition Algorithm Based on Attention Model

Jinghui Chu, Hao Huang, and Wei Lü*
Author Affiliations
  • School of Electrical and Information Engineering, Tianjin University, Tianjin 300072, China
  • show less
    Figures & Tables(11)
    Structure of AAFCNN model
    Structure of semantic connection path
    Different attention modules. (a) Channel attention module; (b) spatial attention module
    Structure of attention model
    Size distribution of traffic signs in TT100K dataset
    Accuracy-recall curves of traffic signs at three scales. (a) Pixel interval of (0,32); (b) pixel interval of (32,96]; (c) pixel interval of (96,400]
    Part of visual recognition results of AAFCNN model
    • Table 1. Performance comparison of different traffic sign recognition methods

      View table

      Table 1. Performance comparison of different traffic sign recognition methods

      MethodBackboneParams /106IndexS /%M /%L /%
      Faster R-CNNResNet-10152.2Recall72.091.391.5
      Precision76.187.586.1
      F1-score74.089.488.7
      Faster R-CNN +FPNResNet-10160.1Recall86.695.595.1
      Precision85.092.992.3
      F1-score85.894.293.7
      Ref. [15]81.2Recall87.493.687.7
      Precision81.790.890.6
      F1-score84.592.089.1
      RetinaNetResNeXt-10194.7Recall87.495.193.1
      Precision84.395.994.2
      F1-score85.895.593.6
      FCOSResNeXt-10189.7Recall88.795.692.4
      Precision85.696.493.5
      F1-score86.896.093.0
      CenterNetHourglassNet191.3Recall89.796.092.4
      Precision90.196.794.9
      F1-score89.996.393.6
      AAFCNNDenseNet-12148.1Recall90.695.693.1
      Precision91.297.396.8
      F1-score90.996.494.9
    • Table 2. Effect of depth of densely connected network on recognition performance

      View table

      Table 2. Effect of depth of densely connected network on recognition performance

      BackboneParams /106AP /%
      SML
      DenseNet-12148.163.480.186.1
      DenseNet-16965.462.579.986.1
      DenseNet-201101.461.779.785.7
      DenseNet-264154.861.980.085.0
    • Table 3. Effect of location of attention model on recognition performance

      View table

      Table 3. Effect of location of attention model on recognition performance

      LocationParams /106AP /%
      SML
      In coding path48.163.480.186.1
      In decoding path47.862.180.085.8
      Both coding path and decoding path48.262.680.085.1
    • Table 4. Performance comparison of each module

      View table

      Table 4. Performance comparison of each module

      ModelParams /106AP /%
      SML
      Base14.160.879.785.9
      Base+AM14.261.779.885.1
      Base+SCP47.861.980.087.2
      Base+AM+SCP48.163.480.186.1
    Tools

    Get Citation

    Copy Citation Text

    Jinghui Chu, Hao Huang, Wei Lü. Anchor-Free Traffic Sign Recognition Algorithm Based on Attention Model[J]. Laser & Optoelectronics Progress, 2021, 58(16): 1610020

    Download Citation

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

    Category: Image Processing

    Received: Aug. 18, 2020

    Accepted: Sep. 30, 2020

    Published Online: Aug. 16, 2021

    The Author Email: Wei Lü (luwei@tju.edu.cn)

    DOI:10.3788/LOP202158.1610020

    Topics