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 条
  • [21] Steel surface defect detection algorithm based on ESI-YOLOv8
    Zhang, Xinrong
    Wang, Yanlong
    Fang, Huaisong
    MATERIALS RESEARCH EXPRESS, 2024, 11 (05)
  • [22] Research on Steel Surface Defect Detection with Improved YOLOv7 Algorithm
    Gao, Chunyan
    Qin, Shen
    Li, Manhong
    Lyv, Xiaoling
    Computer Engineering and Applications, 2024, 60 (07) : 282 - 291
  • [23] A human eye detection algorithm in complex scenarios
    Cui J.
    Cao H.
    Zhang Y.
    Luo S.
    Li J.
    Wang H.
    Beijing Hangkong Hangtian Daxue Xuebao/Journal of Beijing University of Aeronautics and Astronautics, 2021, 47 (01): : 38 - 44
  • [24] Small target detection algorithm in complex background
    Zheng P.
    Bai H.-Y.
    Li W.
    Guo H.-W.
    Bai, Hong-yang (hongyang@njust.edu.cn), 1777, Zhejiang University (54): : 1777 - 1784
  • [25] A modified motion detection algorithm in complex background
    Yu, Zhen
    Wei, Zhen
    ICCSSE 2009: PROCEEDINGS OF 2009 4TH INTERNATIONAL CONFERENCE ON COMPUTER SCIENCE & EDUCATION, 2009, : 98 - 102
  • [26] Saliency detection algorithm under complex background
    Pang, Xin
    Dong, Mingfang
    Yu, Zhezhou
    Journal of Information and Computational Science, 2015, 12 (02): : 423 - 429
  • [27] Research on Optimization of YOLOv5s Detection Algorithm for Steel Surface Defect
    Xu, Hongjun
    Tang, Ziqiang
    Zhang, Jindong
    Zhu, Peihua
    Computer Engineering and Applications, 60 (07): : 306 - 314
  • [28] Steel surface defect detection based on the lightweight improved RT-DETR algorithm
    Mao, Haojie
    Gong, Yongwang
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2025, 22 (01)
  • [29] Steel Surface Defect Detection Algorithm Based on Improved YOLOv8n
    Zhang, Tian
    Pan, Pengfei
    Zhang, Jie
    Zhang, Xiaochen
    APPLIED SCIENCES-BASEL, 2024, 14 (12):
  • [30] Fusion Lightweight Steel Surface Defect Detection Algorithm Based on Improved Deep Learning
    Ren, Fei
    Fei, Jiajie
    Li, HongSheng
    Doma, Bonifacio T., Jr.
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2024, 15 (05) : 377 - 382