Optimized YOLOv5 and Spatial Pyramid Pooling to Enhance Industrial Defect Detection

被引:1
作者
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 [J].
Chen, Zhendong ;
Feng, Xuefeng ;
Liu, Li ;
Jia, Zhenhong .
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 [J].
Li, Xin ;
Wang, Cheng ;
Ju, Haijuan ;
Li, Zhuoyue .
APPLIED SCIENCES-BASEL, 2022, 12 (14)
[4]   Investigation Into Recognition Algorithm of Helmet Violation Based on YOLOv5-CBAM-DCN [J].
Wang, Lijun ;
Cao, Yunyu ;
Wang, Song ;
Song, Xiaona ;
Zhang, Shenfeng ;
Zhang, Jianyong ;
Niu, Jinxing .
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 [J].
Xu, Hao ;
Li, Bo ;
Zhong, Fei .
APPLIED SCIENCES-BASEL, 2022, 12 (23)
[7]   Underwater object detection algorithm based on attention mechanism and cross-stage partial fast spatial pyramidal pooling [J].
Yan, Jinghui ;
Zhou, Zhuang ;
Zhou, Dujuan ;
Su, Binghua ;
Zhe, Xuanyuan ;
Tang, Jialin ;
Lai, Yunting ;
Chen, Jiongjiang ;
Liang, Wanxin .
FRONTIERS IN MARINE SCIENCE, 2022, 9
[8]   Real-time detection of particleboard surface defects based on improved YOLOV5 target detection [J].
Zhao, Ziyu ;
Yang, Xiaoxia ;
Zhou, Yucheng ;
Sun, Qinqian ;
Ge, Zhedong ;
Liu, Dongfang .
SCIENTIFIC REPORTS, 2021, 11 (01)
[9]   Object Detection for Hazardous Material Vehicles Based on Improved YOLOv5 Algorithm [J].
Zhu, Pengcheng ;
Chen, Bolun ;
Liu, Bushi ;
Qi, Zifan ;
Wang, Shanshan ;
Wang, Ling .
ELECTRONICS, 2023, 12 (05)