基于改进YOLOv5s的轻量化布匹瑕疵检测算法  

A lightweight fabric defect detection algorithm based on improved YOLOv5s

作　　者：邹宏睿 任佳[1,2] 潘海鹏[1,2] 周传辉 ZOU Hongrui;REN Jia;PAN Haipeng;ZHOU Chuanhui(School of Information Science and Engineering,Zhejiang Sci-Tech University,Hangzhou 310018,China;Zhejiang Sci-Tech University Changshan Research Institute Co.,Ltd.,Quzhou 324299,China)

机构地区：[1]浙江理工大学信息科学与工程学院,杭州310018 [2]浙江理工大学常山研究院有限公司,衢州324299

出　　处：《浙江理工大学学报（自然科学版）》2024年第3期389-398,共10页Journal of Zhejiang Sci-Tech University(Natural Sciences)

基　　金：浙江省“尖兵”研发攻关计划项目(2023C01002);浙江省教育厅一般科研项目(Y202147717);浙江理工大学研究生优秀学位论文培育基金(LW-YP2022080)。

摘　　要：针对纺织生产中布匹瑕疵检测高精度、实时性的需求,提出了一种基于改进YOLOv5s的轻量化布匹瑕疵检测算法(GhostNet-CBAM-Partial convolution-YOLOv5s,GCP-YOLOv5s)。该算法首先引入GhostNet中的GhostConv模块,对原主干网络进行优化重构,大幅减少网络参数;其次,在主干特征提取网络中加入CBAM(Convolutional block attention module)注意力机制,增加网络的特征提取能力;最后,设计了基于Partial convolution的改进C3模块(C3-Partial convolution,C3-P),在降低模型参数量的同时提高特征融合能力。在自建布匹瑕疵数据集上进行了对比测试,结果表明:与基准模型YOLOv5s相比,GCP-YOLOv5s的参数量降低了41.6%,计算量降低了43.1%,检测速度提高了12 FPS,检测精度提升了1.7%。GCP-YOLOv5s算法在保证模型轻量化的同时具有较高的检测精度,可以满足布匹瑕疵检测的高精度和实时性要求。A lightweight fabric defect detection algorithm,GhostNet-CBAM-Partial convolution-YOLOv5s(GCP-YOLOv5s)based on improved YOLOv5s was proposed to meet the high-precision and real-time requirements of fabric defect detection in textile production.Firstly,this algorithm introduced the GhostConv module in GhostNet to optimize and reconstruct the original backbone network,significantly reducing network parameters.Secondly,CBAM(convolutional block attention module)attention mechanism was incorporated into the backbone network to increase its feature extraction capability.Finally,an improved C3 module C3-P(C3-Partial convolotion)module based on Partial convolution was designed to reduce the number of model parameters while improving the feature fusion capability.Comparative tests were conducted on a dataset of self-built fabric defects on site,and the results showed that compared with the benchmark model YOLOv5s,the parameters and computational complexity of GCP-YOLOv5s were reduced by 41.6%and 43.1%,respectively,while the detection speed and detection accuracy were increased by 12 FPS and 1.7%,respectively.The GCP-YOLOv5s algorithm has high detection accuracy while ensuring model lightweighting,meeting the requirements of high precision and real-time performance in fabric defect detection at the same time.

关 键 词：布匹瑕疵检测 YOLOv5s GhostNet 注意力机制 高精度 实时性 

分 类 号：TP183[自动化与计算机技术—控制理论与控制工程]