Steel surface defect detection algorithm in complex background scenarios

被引:13
|
作者
Zhao, Baiting [1 ]
Chen, Yuran [1 ]
Jia, Xiaofen [2 ,3 ]
Ma, Tianbing [3 ]
机构
[1] Anhui Univ Sci & Technol, Sch Elect & Informat Engn, Huainan 232001, Peoples R China
[2] Anhui Univ Sci & Technol, Sch Artificial Intelligence, Huainan 232001, Peoples R China
[3] Anhui Univ Sci & Technol, State Key Lab Min Response & Disaster Prevent & Co, Huainan 232001, Peoples R China
基金
中国国家自然科学基金;
关键词
Defect detection; YOLOv8; Deep learning; Multi-scale feature extraction;
D O I
10.1016/j.measurement.2024.115189
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Detecting surface defects on steel poses a significant challenge attributed to factors such as poor contrast, diverse defect types, complex background clutter, and noise interference present in images of steel surface defects. Current detection techniques face challenges in quickly and accurately identifying defects within complex backgrounds. To address the deployment of high -precision detection models on edge devices with limited resources, particularly for identifying steel surface defects, this study introduces a Multi -Scale Adaptive Fusion (MSAF) YOLOv8n defect detection algorithm designed for complex backgrounds. This algorithm effectively balances detection speed and accuracy. Firstly, a Multi -Scale Adaptive Fusion Block (MS -AFB) is proposed for the extraction of multi -scale features. Secondly, a Dynamic Coordinate Attention Ghostconv Space Pooling Pyramidfast Cross -stage Partial Convolutional (DCA-GSPPFCSPC) is devised to significantly improve detection accuracy. Furthermore, the detection head has been redesigned utilizing Lightweight Multi -scale Convolutional (LMSC) approach, and an Adaptive Pyramid Receptive Field Block (AP-RFB) has been introduced to improve the receptive field efficiently. Meanwhile, Normalized Weighted Distance (NWD) and Weighted Intersection over Union (WIoU) are employed as the boundary box loss functions, serving as substitutes for Complete Intersection over Union (CIoU) loss function with a ratio of 2:8. The experimental results obtained from the improved Northeastern University Defect Dataset (NEU-DET) dataset demonstrate that MSAF-YOLOv8n model, despite having 40.4 % of the parameters and 28.8 % of Floating Point Operations (FLOPs) of YOLOv8s, achieves a mAP@.5 that is 0.9 % higher than that of YOLOv8s. Additionally, MSAF-YOLOv8n demonstrates robust generalization capabilities in Pascal VOC2007, self -constructed datasets, and various other datasets. Subsequently, the model is implemented on embedded systems, namely Jeston TX2 NX and Orange Pi 5 +, both of which demonstrate real-time detection capabilities.
引用
收藏
页数:14
相关论文
共 50 条
  • [41] Strip steel surface defect detection algorithm based on improved Faster R-CNN
    齐继阳
    吴宇帆
    China Welding, 2024, 33 (02) : 11 - 22
  • [42] Improved YOLOv7 Algorithm for Small Sample Steel Plate Surface Defect Detection
    Dou, Zhi
    Hu, Chenguang
    Li, Qinghua
    Zheng, Liming
    Computer Engineering and Applications, 2023, 59 (23) : 283 - 292
  • [43] Surface Defect Detection Algorithm for Strip Steel Based on Improved YOLOv7 Model
    Wang, Zhu
    Liu, Weisheng
    IAENG International Journal of Computer Science, 2024, 51 (03) : 308 - 316
  • [44] Slim-YOLOv8: A fast and accurate algorithm for surface defect detection of steel strips
    Zhao, Jia
    Liu, Song
    Tao, Han
    Liu, Wanming
    IRONMAKING & STEELMAKING, 2024,
  • [45] Optimization Algorithm of Steel Surface Defect Detection Based on YOLOv8n-SDEC
    Jiang, Xing
    Cui, Yihao
    Cui, Yongcheng
    Xu, Ruikang
    Yang, Jingqi
    Zhou, Jishuai
    IEEE ACCESS, 2024, 12 : 95106 - 95117
  • [46] Progressive refined redistribution pyramid network for defect detection in complex scenarios
    Yu, Xuyi
    Lyu, Wentao
    Wang, Chengqun
    Guo, Qing
    Zhou, Di
    Xu, Weiqiang
    KNOWLEDGE-BASED SYSTEMS, 2023, 260
  • [47] FMR-YOLO: An improved YOLOv8 algorithm for steel surface defect detection
    Ni, Yongjing
    Wu, Qi
    Zhang, Xiuqing
    IET IMAGE PROCESSING, 2025, 19 (01)
  • [48] A Fast Surface Defect Detection Method Based on Background Reconstruction
    Lv, Chengkan
    Zhang, Zhengtao
    Shen, Fei
    Zhang, Feng
    Su, Hu
    INTERNATIONAL JOURNAL OF PRECISION ENGINEERING AND MANUFACTURING, 2020, 21 (03) : 363 - 375
  • [49] Study on Reduction of Background Fringes for Defect Detection of Specular Surface
    Wei, An-Chi
    Chang, Yi-Cheng
    Sze, Jyh-Rou
    2023 ASIA PACIFIC SIGNAL AND INFORMATION PROCESSING ASSOCIATION ANNUAL SUMMIT AND CONFERENCE, APSIPA ASC, 2023, : 2163 - 2167
  • [50] A Fast Surface Defect Detection Method Based on Background Reconstruction
    Chengkan Lv
    Zhengtao Zhang
    Fei Shen
    Feng Zhang
    Hu Su
    International Journal of Precision Engineering and Manufacturing, 2020, 21 : 363 - 375