Steel surface defect detection algorithm in complex background scenarios

被引:14
|
作者
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 条
  • [31] 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
  • [32] Sub-surface defect detection in a steel sheet
    Atzlesberger, J.
    Zagar, B. G.
    Cihal, R.
    Brummayer, M.
    Reisinger, P.
    MEASUREMENT SCIENCE AND TECHNOLOGY, 2013, 24 (08)
  • [33] Steel surface defect detection based on multi-layer fusion networks
    Li, Hanlin
    Liu, Ming
    Yin, Yanfang
    Sun, Weiliang
    SCIENTIFIC REPORTS, 2025, 15 (01):
  • [34] An efficient and real-time steel surface defect detection method based on single-stage detection algorithm
    Zhang H.
    Miao Q.
    Li S.
    Wang C.
    Chan S.
    Hu J.
    Bai C.
    Multimedia Tools and Applications, 2024, 83 (42) : 90595 - 90617
  • [35] y A New Self-Reference Image Decomposition Algorithm for Strip Steel Surface Defect Detection
    Liu, Kun
    Luo, Nana
    Li, Aimei
    Tian, Ying
    Sajid, Hasan
    Chen, Haiyong
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2020, 69 (07) : 4732 - 4741
  • [36] Weakly-Supervised Steel Plate Surface Defect Detection Algorithm by Integrating Multiple Level Features
    He Y.
    Song K.-C.
    Zhang D.-F.
    Yan Y.-H.
    Dongbei Daxue Xuebao/Journal of Northeastern University, 2021, 42 (05): : 687 - 692
  • [37] Surface Defect Segmentation Algorithm of Steel Plate Based on Geometric Median Filter Pruning
    Hao, Zhiqiang
    Wang, Zhigang
    Bai, Dongxu
    Tong, Xiliang
    FRONTIERS IN BIOENGINEERING AND BIOTECHNOLOGY, 2022, 10
  • [38] Improved Yolov5 Algorithm for Surface Defect Detection of Solar Cell
    Li, Pengjie
    Shan, Shuo
    Zeng, Pengzhong
    Wei, Haikun
    2023 35TH CHINESE CONTROL AND DECISION CONFERENCE, CCDC, 2023, : 3601 - 3605
  • [39] Improved SSD-assisted algorithm for surface defect detection of electromagnetic luminescence
    Xu, Zhenying
    Wu, Ziqian
    Fan, Wei
    PROCEEDINGS OF THE INSTITUTION OF MECHANICAL ENGINEERS PART O-JOURNAL OF RISK AND RELIABILITY, 2021, 235 (05) : 761 - 768
  • [40] Virtual-sample-based defect detection algorithm for aluminum tube surface
    Lang, Ning
    Wang, Decheng
    Cheng, Peng
    Zuo, Shanchao
    Zhang, Pengfei
    MEASUREMENT SCIENCE AND TECHNOLOGY, 2021, 32 (08)