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

    To address the challenges of low detection accuracy, frequent missed detections, and false detections in steel surface defect detection for small target samples, an improved YOLOv7 algorithm is proposed. First, the Swin-Transformer is introduced for feature extraction, and a dual-branch loop is employed to fuse global and local features interactively. Second, distribution shift convolution is utilized to improve the efficient layer aggregation network, thereby enhance its local feature extraction capability. Finally, the regression loss function incorporates the weighted normalized Wasserstein distance and complete intersection over union methods to reduce the sensitivity to small target deviations. Experimental comparisons on the NEU-DET dataset demonstrate that the improved algorithm increases the mean average precision by 8.8 percentage points, reaching 83.1%. Proposed algorithm improves the accuracy of steel surface defect detection and reduces false detections and missed detections for small target samples. In addition, the proposedd algorithm achieves a detection speed of 71 frames/s, meeting the requirements for real-time detection.

    Keywords
    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