首页 | 本学科首页   官方微博 | 高级检索  
     

基于改进YOLOv5的冲压件缺陷检测方法研究
引用本文:夏巍,操乐文,苏亮亮. 基于改进YOLOv5的冲压件缺陷检测方法研究[J]. 安徽建筑大学学报, 2024, 32(1): 61-67
作者姓名:夏巍  操乐文  苏亮亮
作者单位:安徽建筑大学 电子与信息工程学院,安徽 合肥 230601
基金项目:国家自然科学基金项目(62001004)
摘    要:冲压件在生产过程中容易出现裂纹、划痕、起皱、凹凸点等缺陷。目前,生产线上的冲压件缺陷检测以人工检测为主,效率低,且容易造成漏检。为此,提出了一种基于改进YOLOv5模型的缺陷检测算法。为了提高缺陷部分的关注度,更好地聚焦缺陷,本文在YOLOv5模型的主干网络中引入CA注意力模块。为了进一步提升模型的精度,本文通过对比实验,将目标框损失函数改为 GIoU,提升了定位精度。实验表明,相较于原模型,改进后的YOLOv5模型精准度、召回率、mAP值均得到提升。

关 键 词:YOLOv5;冲压件;缺陷检测;注意力机制

Research on Defect Detection Method of Stamping Parts Based on Improved YOLOv5
XIA Wei,CAO Lewen,SU Liangliang. Research on Defect Detection Method of Stamping Parts Based on Improved YOLOv5[J]. Journal of Anhui Jianzhu University, 2024, 32(1): 61-67
Authors:XIA Wei  CAO Lewen  SU Liangliang
Affiliation:School of Electronics and Communication Engineering,Anhui Jianzhu University,Hefei 230601,China
Abstract:Stamped parts are prone to cracks, scratches, wrinkles, bumps and other defects in the production process. At present, the defect detection of stamped parts on the production line is based on manual detection, which is inefficient and prone to leakage. For this reason, a defect detection algorithm based on the improved YOLOv5 model is proposed. In order to improve the attention of the defective part and better focus the defects, this paper introduces the CA attention module in the backbone network of the YOLOv5model. To further improve the accuracy of the model, this paper improves the localization accuracy by changing the target frame loss function to GIoU through comparative experiments. The experiments show that compared with the original model, the improvedYOLOv5 model precision, recall, and mAP value are all improved.
Keywords:YOLOv5;stamping part;defect detection;attention mechanism
点击此处可从《安徽建筑大学学报》浏览原始摘要信息
点击此处可从《安徽建筑大学学报》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号