Steel Surface Defect Detection Based on Improved YOLOv7

被引:0
作者
Li, Ming [1 ]
Wei, Lisheng [2 ]
Zheng, Bowen [1 ]
机构
[1] Anhui Polytech Univ, Sch Elect Engn, Wuhu, Peoples R China
[2] Anhui Key Lab Elect Drive & Control, Wuhu, Peoples R China
来源
2024 4TH INTERNATIONAL CONFERENCE ON COMPUTER, CONTROL AND ROBOTICS, ICCCR 2024 | 2024年
关键词
target detection; defect detection; YOLOv7; GAMAttention; loss function;
D O I
10.1109/ICCCR61138.2024.10585576
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Aiming at the current problem of steel surface defect detection with low accuracy and slow speed, which can easily lead to misdetection and omission, an algorithm for steel surface defect detection based on improved YOLOv7 is proposed. Firstly, the GAM (Global Attention Mechanism) attention mechanism is introduced, while CNeB and C3C2 are added to improve the feature extraction ability of the model by reducing the information approximation and amplifying the global interaction representation. Secondly, the WIoU (Wise-IoU) loss function is used to improve the convergence speed at the late stage of model training. Finally, the improved YOLOv7 is compared with other models. The experimental results show that the algorithm of this paper has an average detection accuracy (mAP) of 72.9% on the NEU-DET dataset, which is 4.1% higher compared with the original YOLOv7 algorithm, and the detection time is reduced by 63.6% under the same conditions, which verifies the effectiveness and feasibility of this paper's algorithm, and it has a certain value of application in industrial applications.
引用
收藏
页码:51 / 55
页数:5
相关论文
共 50 条
  • [41] Steel surface defect detection algorithm based on ESI-YOLOv8
    Zhang, Xinrong
    Wang, Yanlong
    Fang, Huaisong
    MATERIALS RESEARCH EXPRESS, 2024, 11 (05)
  • [42] Lightweight Surface Defect Detection Algorithm Based on Improved YOLOv5
    Yang, Kaijun
    Chen, Tao
    2024 5TH INTERNATIONAL CONFERENCE ON MECHATRONICS TECHNOLOGY AND INTELLIGENT MANUFACTURING, ICMTIM 2024, 2024, : 798 - 802
  • [43] An Improved YOLOv8 Model for Strip Steel Surface Defect Detection
    Wang, Jinwen
    Chen, Ting
    Xu, Xinke
    Zhao, Longbiao
    Yuan, Dijian
    Du, Yu
    Guo, Xiaowei
    Chen, Ning
    APPLIED SCIENCES-BASEL, 2025, 15 (01):
  • [44] Steel Surface Defect Detection Based on YOLOv8-TLC
    Liu, Chunling
    Cheng, Hui
    APPLIED SCIENCES-BASEL, 2024, 14 (21):
  • [45] Wood surface defect detection based on improved YOLOv8
    Yilin Ge
    Haowen Ji
    Xingli Liu
    Signal, Image and Video Processing, 2025, 19 (8)
  • [46] A lightweight algorithm for steel surface defect detection using improved YOLOv8
    Ma, Shuangbao
    Zhao, Xin
    Wan, Li
    Zhang, Yapeng
    Gao, Hongliang
    SCIENTIFIC REPORTS, 2025, 15 (01):
  • [47] Lightweight multi-attribute target detection for dogs and cats based on improved YOLOv7
    Cao, Danyang
    Liu, Fangfang
    PROCEEDINGS OF 2024 3RD INTERNATIONAL CONFERENCE ON CYBER SECURITY, ARTIFICIAL INTELLIGENCE AND DIGITAL ECONOMY, CSAIDE 2024, 2024, : 395 - 400
  • [48] YOLO-PD: Abnormal Signal Detection in Gas Pipelines Based on Improved YOLOv7
    Yan, Wendi
    Liu, Wei
    Bi, Hongbo
    Jiang, Chunlei
    Zhang, Qiao
    Wang, Tao
    Dong, Taiji
    Ye, Xiaohui
    Sun, Yu
    IEEE SENSORS JOURNAL, 2023, 23 (17) : 19737 - 19746
  • [49] Improving the lightweight VTG-YOLOv7-tiny for steel surface defect detection
    Liang L.
    Long P.
    Feng Y.
    Lu B.
    Guangxue Jingmi Gongcheng/Optics and Precision Engineering, 2024, 32 (08): : 1227 - 1240
  • [50] Lightweight-detection: The strip steel surface defect identification based on improved YOLOv5d
    Lu, Yan
    Huang, Zhi-Chao
    Jiang, Yu-Qiang
    Liu, Guang-Rui
    Wang, Jia-Xuan
    Ma, Qi-Ming
    MATERIALS TODAY COMMUNICATIONS, 2024, 40