棉纺织技术2024，Vol.52Issue(4)：30-39,10.
基于上下文信息聚合YOLOv5的织物缺陷检测
Fabric defect detection based on contextual information aggregation YOLOv5
摘要
Abstract
In response to the problem that the current mainstream inspection methods cannot effectively capture the rich contextual information of fabric defects,resulting in lower detection accuracy and higher leakage rate of defects similar with background,an improved YOLOv5 model(YOLO-CA model)that could aggregate contextual information was proposed.Defects with context-rich contextual relationships were captured by the YOLO-CA model by incorporating a dual cross-attention module(DCCA)in the backbone network of YOLOv5,which enabled the model to focus on the defective area and improved the detection rate of defects by the model.More detailed defect feature information was extracted in combination with the CSP-SoftSPPF module to improve the detection accuracy of the model.The experimental results showed that the YOLO-CA model has mAP values of 67.6%and 93.2%,as well as MR values of 38.4%and 10.5%on SDCFD and TILDA publicly available datasets.Compared with original YOLOv5 model,the mAP value of YOLO-CA model was increased by 20.8 percentage points and 4.5 percentage points,and the MR values were decreased by 16.5 percentage points and 6.9 percentage points respectively.It is considered that the YOLO-CA model has better detection performance,can effectively reduce the leakage rate of defects of similar classes to the background,improves the detection accuracy of several defects,and is suitable for fabric defect detection tasks.关键词
深度学习/织物缺陷检测/YOLOv5模型/注意力机制/SoftPool池化Key words
deep learning/fabric defect detection/YOLOv5 model/attention mechanism/SoftPool pooling分类
轻工业引用本文复制引用
李静,郑文斌..基于上下文信息聚合YOLOv5的织物缺陷检测[J].棉纺织技术,2024,52(4):30-39,10.基金项目
四川省自然科学基金(2022NSFSC0571) （2022NSFSC0571）
四川省应用基础研究项目(2018JY0273,2019YJ0532) （2018JY0273,2019YJ0532）
四川省重点实验室开放项目(SCVCVR2020.05VS) （SCVCVR2020.05VS）
