Laser & Optoelectronics Progress, Volume. 61, Issue 24, 2412002(2024)

Feature Enhancement and Metric Optimization for Defect Detection on Steel Surface

Junying Chen*, Hantao Huang, and Zhaoyang Li
Author Affiliations
  • College of Information and Control Engineering, Xi'an University of Architecture and Technology, Xi'an 710055, Shaanxi , China
  • show less
    Figures & Tables(10)
    Structure of STCSP
    Structure of ELAN-DS module
    Structure of improved YOLOv7 algorithm
    P-R curves of different algorithms. (a) Original algorithm; (b) improved algorithm
    Comparison of detection results before and after algorithm improvement
    • Table 1. Detection results with STCSP module added at different feature extraction layers

      View table

      Table 1. Detection results with STCSP module added at different feature extraction layers

      Number of layermAP@0.5AP@0.5
      CrInPaPsRsSc
      74.335.185.096.380.357.292.1
      2472.734.179.395.180.559.288.2
      3774.536.982.797.083.359.188.0
      5075.637.984.198.281.562.089.8
    • Table 2. Detection results with ELAN-DS module added at different positions

      View table

      Table 2. Detection results with ELAN-DS module added at different positions

      PositionmAP@0.5AP@0.5
      CrInPaPsRsSc
      74.335.185.096.380.357.292.1
      Backbone76.940.983.797.685.162.591.7
      Head74.137.983.296.482.458.786.0
      Backbone & Head75.538.383.896.082.861.890.1
    • Table 3. Detection results with different ratios of weighted NWD and CIoU loss function

      View table

      Table 3. Detection results with different ratios of weighted NWD and CIoU loss function

      NWD∶CIoUmAP@0.5AP@0.5
      CrInPaPsRsSc
      0∶1.074.335.185.096.380.357.292.1
      0.2∶0.876.637.386.997.684.662.291.2
      0.8∶0.277.843.285.698.182.16494.1
      1.0∶078.746.684.497.885.664.193.5
      0.5∶0.581.750.488.899.286.070.695.0
    • Table 4. Results of ablation experiments

      View table

      Table 4. Results of ablation experiments

      NumberSTCSPELAN-DSNWD∶CIoU=0.5∶0.5FPS /(frame·s-1mAP@0.5 /%
      N18274.3
      N2+7575.6
      N3+6676.9
      N4+8481.7
      N5++7974.4
      N6++7679.3
      N7++7781.9
      N8+++7183.1
    • Table 5. Comparison of experimental results for different object detection algorithms

      View table

      Table 5. Comparison of experimental results for different object detection algorithms

      AlgorithmParameters /MFPS /(frame·s-1mAP@0.5 /%
      Cascade R-CNN107.03759.6
      RetinaNet28.54261.7
      YOLOv5(m)21.27570.7
      Improved YOLOv55.884978.5
      YOLOX(m)25.38870.1
      CG-Net32.710575.9
      CenterNet2.34173.4
      YOLOv736.58274.3
      YOLOv830.111875.8
      YOLOv960.54074.4
      Proposed38.87183.1
    Tools

    Get Citation

    Copy Citation Text

    Junying Chen, Hantao Huang, Zhaoyang Li. Feature Enhancement and Metric Optimization for Defect Detection on Steel Surface[J]. Laser & Optoelectronics Progress, 2024, 61(24): 2412002

    Download Citation

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

    Category: Instrumentation, Measurement and Metrology

    Received: Mar. 12, 2024

    Accepted: Apr. 25, 2024

    Published Online: Dec. 10, 2024

    The Author Email: Junying Chen (chenjy@xauat.edu.cn)

    DOI:10.3788/LOP240876

    CSTR:32186.14.LOP240876

    Topics