Optimized YOLOv5 and Spatial Pyramid Pooling to Enhance Industrial Defect Detection

被引:0
作者
Su, Ming-Hsiang [1 ]
Jian, Zhi-Juan [1 ]
机构
[1] Soochow Univ, Dept Data Sci, Taipei, Taiwan
来源
2024 11TH INTERNATIONAL CONFERENCE ON CONSUMER ELECTRONICS-TAIWAN, ICCE-TAIWAN 2024 | 2024年
关键词
Defect detection; YOLOv5; Spatial Pyramid Pooling; NEU-DET dataset;
D O I
10.1109/ICCE-Taiwan62264.2024.10674059
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
This study addresses industrial appearance defect detection challenges by enhancing the YOLOv5 algorithm using the NEU-DET dataset. Leveraging optimized Spatial Pyramid Pooling (SPPCSPS), the model exhibits improved accuracy in recognizing various defect types, mitigating issues of omission and misdetection. Experimental results demonstrate a 1.94% average accuracy improvement over the original YOLOv5. The proposed modifications enhance the algorithm's ability to detect defects, contributing to more accurate and efficient industrial inspection processes.
引用
收藏
页码:349 / 350
页数:2
相关论文
共 9 条
  • [1] Surface defect detection of industrial components based on vision
    Chen, Zhendong
    Feng, Xuefeng
    Liu, Li
    Jia, Zhenhong
    [J]. SCIENTIFIC REPORTS, 2023, 13 (01):
  • [2] He KM, 2014, LECT NOTES COMPUT SC, V8691, P346, DOI [arXiv:1406.4729, 10.1007/978-3-319-10578-9_23]
  • [3] Surface Defect Detection Model for Aero-Engine Components Based on Improved YOLOv5
    Li, Xin
    Wang, Cheng
    Ju, Haijuan
    Li, Zhuoyue
    [J]. APPLIED SCIENCES-BASEL, 2022, 12 (14):
  • [4] Investigation Into Recognition Algorithm of Helmet Violation Based on YOLOv5-CBAM-DCN
    Wang, Lijun
    Cao, Yunyu
    Wang, Song
    Song, Xiaona
    Zhang, Shenfeng
    Zhang, Jianyong
    Niu, Jinxing
    [J]. IEEE ACCESS, 2022, 10 : 60622 - 60632
  • [5] Wang XIB, 2023, Arxiv, DOI arXiv:2302.04521
  • [6] Light-YOLOv5: A Lightweight Algorithm for Improved YOLOv5 in Complex Fire Scenarios
    Xu, Hao
    Li, Bo
    Zhong, Fei
    [J]. APPLIED SCIENCES-BASEL, 2022, 12 (23):
  • [7] Underwater object detection algorithm based on attention mechanism and cross-stage partial fast spatial pyramidal pooling
    Yan, Jinghui
    Zhou, Zhuang
    Zhou, Dujuan
    Su, Binghua
    Zhe, Xuanyuan
    Tang, Jialin
    Lai, Yunting
    Chen, Jiongjiang
    Liang, Wanxin
    [J]. FRONTIERS IN MARINE SCIENCE, 2022, 9
  • [8] Real-time detection of particleboard surface defects based on improved YOLOV5 target detection
    Zhao, Ziyu
    Yang, Xiaoxia
    Zhou, Yucheng
    Sun, Qinqian
    Ge, Zhedong
    Liu, Dongfang
    [J]. SCIENTIFIC REPORTS, 2021, 11 (01)
  • [9] Object Detection for Hazardous Material Vehicles Based on Improved YOLOv5 Algorithm
    Zhu, Pengcheng
    Chen, Bolun
    Liu, Bushi
    Qi, Zifan
    Wang, Shanshan
    Wang, Ling
    [J]. ELECTRONICS, 2023, 12 (05)