A light weight multi-scale feature fusion steel surface defect detection model based on YOLOv8

被引:24
|
作者
Xie, Weining [1 ]
Sun, Xiaoyong [1 ]
Ma, Weifeng [1 ]
机构
[1] Zhejiang Univ Sci & Technol, Sch Informat & Elect Engn, 318 Liuhe Rd, Hangzhou 310023, Zhejiang, Peoples R China
关键词
defect detection; YOLOv8; deep learning; feature fusion;
D O I
10.1088/1361-6501/ad296d
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
In industrial production, the steel surface may incur different defects owing to the influence of external factors, thereby affecting the performance of steel. With the increasing requirements for steel quality, achieving efficient detection of steel surface defects is a difficult problem that urgently needs to be solved. Traditional steel surface defect detection methods are limited by poor detection performance and slow detection speed. Therefore, a model named LMS-YOLO, based on YOLOv8, is proposed in this paper for achieving efficient steel surface defect detection. Firstly, in backbone, the light weight multi-scale mixed convolution (LMSMC) module is designed to fuse with C2f to obtain C2f_LMSMC, so as to extract the features of different scales for fusion and achieve the light weight of the network. Meanwhile, the proposed efficient global attention mechanism was added to backbone to enhance cross dimensional information interaction and feature extraction capabilities, and to achieve a more efficient attention mechanism. In neck, using channel tuning to achieve better cross scale fusion in BiFPN. Finally, the model uses three independent decoupled heads for regression and classification, and replaces CIoU with NWD as the regression loss to enhance the effect of detecting small scale defects. The experimental results showed that LMS-YOLO achieved 81.1 mAP and 61.3 FPS on NEU-DET, 80.5 mAP and 61.3 FPS on GC10-DET, respectively. The mAP increased by 2.8 and 4.7 compared to YOLOv8, and decreased by 17.4% in floating point operations (GFLOPs) and 34.2% in parameters (Params), which indicates that the model proposed in this paper has a better comprehensive performance compared with other methods in steel surface defect detection.
引用
收藏
页数:17
相关论文
共 50 条
  • [21] LSKA-YOLOv8: A lightweight steel surface defect detection algorithm based on YOLOv8 improvement
    Tie, Jun
    Zhu, Chengao
    Zheng, Lu
    Wang, Haijiao
    Ruan, Chongwei
    Wu, Mian
    Xu, Ke
    Liu, Jiaqing
    ALEXANDRIA ENGINEERING JOURNAL, 2024, 109 : 201 - 212
  • [22] Multi-Scale Feature Fusion Model for Bridge Appearance Defect Detection
    Pang, Rong
    Yang, Yan
    Huang, Aiguo
    Liu, Yan
    Zhang, Peng
    Tang, Guangwu
    BIG DATA MINING AND ANALYTICS, 2024, 7 (01): : 1 - 11
  • [23] A Lightweight Strip Steel Surface Defect Detection Network Based on Improved YOLOv8
    Chu, Yuqun
    Yu, Xiaoyan
    Rong, Xianwei
    SENSORS, 2024, 24 (19)
  • [24] Improved Lightweight YOLOv8 Model for Rice Disease Detection in Multi-Scale Scenarios
    Wang, Jinfeng
    Ma, Siyuan
    Wang, Zhentao
    Ma, Xinhua
    Yang, Chunhe
    Chen, Guoqing
    Wang, Yijia
    AGRONOMY-BASEL, 2025, 15 (02):
  • [25] Road Surface Defect Detection Algorithm Based on YOLOv8
    Sun, Zhen
    Zhu, Lingxi
    Qin, Su
    Yu, Yongbo
    Ju, Ruiwen
    Li, Qingdang
    ELECTRONICS, 2024, 13 (12)
  • [26] An efficient re-parameterization feature pyramid network on YOLOv8 to the detection of steel surface defect
    Xie, Weining
    Ma, Weifeng
    Sun, Xiaoyong
    NEUROCOMPUTING, 2025, 614
  • [27] Defect Detection of Photovoltaic Modules Based on Multi-Scale Feature Fusion
    Tian, Hao
    Zhou, Qiang
    He, Chenlong
    Computer Engineering and Applications, 2024, 60 (03) : 340 - 347
  • [28] Surface defect detection of hot rolled steel based on multi-scale feature fusion and attention mechanism residual block
    Zhang, Hongkai
    Li, Suqiang
    Miao, Qiqi
    Fang, Ruidi
    Xue, Song
    Hu, Qianchuan
    Hu, Jie
    Chan, Sixian
    SCIENTIFIC REPORTS, 2024, 14 (01)
  • [29] Surface defect detection of hot rolled steel based on multi-scale feature fusion and attention mechanism residual block
    Hongkai Zhang
    Suqiang Li
    Qiqi Miao
    Ruidi Fang
    Song Xue
    Qianchuan Hu
    Jie Hu
    Sixian Chan
    Scientific Reports, 14
  • [30] Improving Defect Detection in YOLOv8 for Steel Strip
    Ma, Jinlin
    Cao, Haojie
    Ma, Ziping
    Lin, Baobao
    Yang, Jipeng
    Computer Engineering and Applications, 2024, 60 (21) : 183 - 193