A lightweight algorithm for steel surface defect detection using improved YOLOv8

被引:0
|
作者
Ma, Shuangbao [1 ,2 ]
Zhao, Xin [2 ]
Wan, Li [3 ]
Zhang, Yapeng [1 ,2 ]
Gao, Hongliang [4 ]
机构
[1] Wuhan Text Univ, Hubei Key Lab Digital Text Equipment, Wuhan 430073, Peoples R China
[2] Wuhan Text Univ, Sch Mech Engn & Automat, Wuhan 430073, Peoples R China
[3] Wuhan Text Univ, Sch Econ, Wuhan 430073, Peoples R China
[4] Hubei Normal Univ, Sch Elect Engn & Automat, Huangshi, Peoples R China
来源
SCIENTIFIC REPORTS | 2025年 / 15卷 / 01期
基金
中国国家自然科学基金;
关键词
YOLOv8; Lightweight; Defect detection; GhostNet; Attention mechanisms; SIoU;
D O I
10.1038/s41598-025-93469-5
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
In response to the issues of low precision, a large number of parameters and high model complexity in steel surface defect detection, a lightweight algorithm using improved YOLOv8 is proposed. Firstly, GhostNet is utilized as the backbone network in order to reduce the number of model parameters and computational complexity. Secondly, the MPCA (MultiPath Coordinate Attention) attention mechanism is integrated to enhance feature extraction capabilities. Finally, the SIoU (Simplified IoU ) is used to replace the traditional CIoU loss function, which can make the anchor frame more fast and accurate in the regression process, to improve the stability and the robustness of detection. The experimental results indicate that these enhancements have led to a reduction of 37% in calculation amount for the improved YOLOv8n algorithm, a decrease of 32% in parameter count, and an increase in average detection accuracy ( mAP ) by 1.2%. This model achieves a balance between lightweighting and detection accuracy while providing a viable solution for deployment in computationally resource-constrained edge computing environments such as embedded systems and mobile devices.
引用
收藏
页数:13
相关论文
共 50 条
  • [1] Improved Steel Surface Defect Detection Algorithm Based on YOLOv8
    You, Congzhe
    Kong, Haozheng
    IEEE ACCESS, 2024, 12 : 99570 - 99577
  • [2] Lightweight rail surface defect detection algorithm based on an improved YOLOv8
    Xu, CanYang
    Liao, Yingying
    Liu, Yongqiang
    Tian, Runliang
    Guo, Tao
    MEASUREMENT, 2025, 242
  • [3] Improved Lightweight Bearing Defect Detection Algorithm of YOLOv8
    Yao, Jingli
    Cheng, Guang
    Wan, Fei
    Zhu, Deping
    Computer Engineering and Applications, 2024, 60 (21) : 205 - 214
  • [4] A Lightweight Strip Steel Surface Defect Detection Network Based on Improved YOLOv8
    Chu, Yuqun
    Yu, Xiaoyan
    Rong, Xianwei
    SENSORS, 2024, 24 (19)
  • [5] 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
  • [6] Lightweight insulator defect detection algorithm based on improved YOLOv8
    Tang, Mingyue
    Wu, Hang
    PROCEEDINGS OF 2024 3RD INTERNATIONAL CONFERENCE ON CYBER SECURITY, ARTIFICIAL INTELLIGENCE AND DIGITAL ECONOMY, CSAIDE 2024, 2024, : 197 - 201
  • [7] Steel surface defect detection based on improved YOLOv8
    Lu, Xin-ya
    Qu, Mei-xia
    ENGINEERING RESEARCH EXPRESS, 2025, 7 (01):
  • [8] Steel Surface Defect Detection Algorithm Based on YOLOv8
    Song, Xuan
    Cao, Shuzhen
    Zhang, Jingwei
    Hou, Zhenguo
    ELECTRONICS, 2024, 13 (05)
  • [9] Improved YOLOv8 Algorithm for Industrial Surface Defect Detection
    Su, Jia
    Jia, Ze
    Qin, Yichang
    Zhang, Jianyan
    Computer Engineering and Applications, 2024, 60 (14) : 187 - 196
  • [10] An Improved YOLOv8 Algorithm for Rail Surface Defect Detection
    Wang, Yan
    Zhang, Kehua
    Wang, Ling
    Wu, Lintong
    IEEE ACCESS, 2024, 12 : 44984 - 44997