• 中国期刊全文数据库
  • 中国学术期刊综合评价数据库
  • 中国科技论文与引文数据库
  • 中华核心期刊(遴选)数据库
HUANG Yi, DONG Rongsheng. Steel surface defect detection based on multi-scale feature fusion and attention mechanism[J]. Journal of Guilin University of Electronic Technology, 2022, 42(5): 398-404.
Citation: HUANG Yi, DONG Rongsheng. Steel surface defect detection based on multi-scale feature fusion and attention mechanism[J]. Journal of Guilin University of Electronic Technology, 2022, 42(5): 398-404.

Steel surface defect detection based on multi-scale feature fusion and attention mechanism

  • In order to solve the difficulty of defect location and classification in steel surface defect detection and improve the efficiency of steel surface defect detection, a steel surface defect detection algorithm based on multi-scale feature fusion and attention mechanism is proposed. The model of the algorithm is based on encoder-decoder structure to achieve the task of defect classification and segmentation. The encoder structure uses the residual network ResNet50 as the backbone and then uses the multi-scale feature fusion module to capture rich multi-scale spatial information. The decoder structure is based on the Global Attention Upsample module, and uses the global context weights generated by high-level semantic information to guide the shallow details to achieve more accurate selection of detailed information. Finally, the segmentation results are refined through 3×3 convolutional blocks, and gradually recover defect information and make predictions. Using the steel surface defect dataset provided by the kaggle competition platform to experiment with the algorithm, the Dice coefficient of defect detection can reach 94.22%. Compared with semantic segmentation models such as U-net, the defect detection effect is better.
  • loading

Catalog

    Turn off MathJax
    Article Contents

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return