沈阳大学学报(自然科学版)2024,Vol.36Issue(2):112-120,封3,10.
基于改进YOLOv7的织物表面疵点检测技术
Fabric Surface Defect Detection Technology Based on Improved YOLOv7
摘要
Abstract
In view of the limitations of the current fabric defect detection technology in the textile industry,an improved YOLOv7 algorithm for automatic detection of fabric defects was proposed.Firstly,the SPD-Conv module was introduced into the neck network,which retained the feature discrimination information related to defects during convolution downsampling,and solved the problem of insufficient learning of feature information of small targets in the original network.Secondly,the backbone network of YOLOv7 introduced the CA attention mechanism,which not only took into account the channel attention,but also considered the location information,so as to identify defects more effectively.Finally,WIoU was used as the border loss function to make it more focused on the anchor box of general quality,so as to enhance the generalization ability of YOLOv7.Through experimental comparison,it was found that the mAP value and accuracy of the improved algorithm were 92.28%and 95.65%,which were 2.64%and 4.12%higher than the original YOLOv7 algorithm,respectively,which could meet the requirements of the textile industry in terms of defect detection.关键词
疵点检测/YOLOv7/SPD-Conv模块/WIoU/CA注意力机制Key words
defect detection/YOLOv7/SPD-Conv module/WIoU/CA attention mechanism分类
信息技术与安全科学引用本文复制引用
任经琦,张团善,赵浩铭..基于改进YOLOv7的织物表面疵点检测技术[J].沈阳大学学报(自然科学版),2024,36(2):112-120,封3,10.基金项目
国家自然科学基金资助项目(51735010) (51735010)
西安现代智能纺织设备重点实验室项目(2019220614SYS021CG043). (2019220614SYS021CG043)