一种布匹瑕疵检测方法、装置及可读介质制造方法及图纸

本发明专利技术公开了一种布匹瑕疵检测方法、装置及可读介质，通过获取待检测的布匹图像和模板图像；构建改进的Yolov5s网络并训练，得到布匹瑕疵检测模型，改进的Yolov5s网络包括主干特征提取网络、颈部特征融合网络和头部预测网络，主干特征提取网络采用CSPDarkNet，颈部特征融合网络包括特征过滤金字塔网络和融合卷积块注意力模块的路径聚合网络，在训练过程中采用多类聚焦置信度损失函数代替交叉熵置信度损失函数；将待检测的布匹图像和模板图像输入布匹瑕疵检测模型，得到布匹检测结果。该方法关注实际布匹瑕疵检测中存在的瑕疵种类分布不均、不同瑕疵的检测难易程度不同等问题，通过对冗余背景特征的过滤与关键前景特征的提纯，降低网络复杂度，提高检测精度和效率。提高检测精度和效率。提高检测精度和效率。

【技术实现步骤摘要】
一种布匹瑕疵检测方法、装置及可读介质


[0001]本专利技术涉及计算机视觉
，具体涉及一种布匹瑕疵检测方法、装置及可读介质。

技术介绍

[0002]随着智能制造的发展，服装生产行业通过布匹瑕疵检测设备检测面料瑕疵，提高面料的生产质量。生产线的速度较快，传统布匹瑕疵检测检测方法的人工检测速度太慢、效率低，误检率和错检率高，并会对质检工人眼部造成不可逆的损伤，因此提出了基于机器学习的布匹瑕疵检测方法和基于深度学习的布匹瑕疵检测方法。
[0003]基于机器学习的布匹瑕疵检测方法中最重要的因素为手工特征的设计与选择。由于机器学习方法在提取特征时需要手工设计，往往手工设计的特征语义层次低，导致在一些差异小、复杂纹理的布匹图像上检测精度低。
[0004]基于深度学习的布匹瑕疵检测方法不再采用人工设计的浅层特征，而是利用卷积神经网络自动提取深层特征。基于深度学习的检测方法均衡了检测精度和检测效率。其中基于YOLOv5s的深度学习网络模型在大尺度、简单纹理、背景干扰小、类别分布均匀的瑕疵布匹数据集中能够获得较为满意的效果。但是，现有的基于YOLOv5s网络的布匹瑕疵检测方法在背景干扰大、瑕疵种类分布不均的瑕疵布匹数据集中的表现较差，导致模型的检测精确度和泛化能力受到限制。

技术实现思路

[0005]针对上述提到的技术问题。本申请的实施例的目的在于提出了一种布匹瑕疵检测方法、装置及可读介质，来解决以上
技术介绍
部分提到的技术问题。
[0006]第一方面，本专利技术提供了一种布匹瑕疵检测方法，包括以下步骤：
[0007]获取待检测的布匹图像和模板图像；
[0008]构建改进的Yolov5s网络并训练，得到布匹瑕疵检测模型，改进的Yolov5s网络包括主干特征提取网络、颈部特征融合网络和头部预测网络，主干特征提取网络采用CSPDarkNet，颈部特征融合网络包括特征过滤金字塔网络和融合卷积块注意力模块的路径聚合网络，在训练过程中采用多类聚焦置信度损失函数代替交叉熵置信度损失函数；
[0009]将待检测的布匹图像和模板图像输入布匹瑕疵检测模型，得到布匹检测结果。
[0010]作为优选，CSPDarkNet包括依次连接的第一单元、第二单元、第三单元、第四单元和第五单元，第一单元包括依次串联的Focus模块和第一CBS模块，第二单元包括依次串联的第二CBS模块和第一CSP1_x模块，第三单元包括依次串联的第三CBS模块和第二CSP1_x模块，第四单元包括依次串联的第四CBS模块和第三CSP1_x模块，第五单元包括依次串联的第五CBS模块、第四CSP1_x模块和SPPF模块，将布匹图像输入CSPDarkNet，在第三单元、第四单元和第五单元分别输出得到第一有效特征图、第二有效特征图和第三有效特征图。
[0011]作为优选，特征过滤金字塔网络包括FPN特征融合网络、冗余特征过滤网络和模板
特征提取网络，第一有效特征图、第二有效特征图、第三有效特征图和模板图像分别输入FPN特征融合网络中，其中，第三有效特征图输入第六CBS模块，得到第三中间特征图，将第三中间特征图经由第一Upsample模块得到特征图与第二有效特征图进行Concat操作，再经由第一CSP2_3模块与第七CBS模块进行融合，得到第二中间特征图，将第二中间特征图经由第二Upsample模块得到的特征图与第一有效特征图进行Concat操作，再经由第二CSP2_3模块与第八CBS模块进行融合，得到第一中间特征图。
[0012]作为优选，模板图像输入模板特征提取网络中，模板图像输入第九CBS模块，得到第一模板特征图，将第一模板特征图输入Downsample1模块，得到第二模板特征图，第二模板特征图输入Downsample2模块，得到第三模板特征图。
[0013]作为优选，第一中间特征图、第二中间特征图、第三中间特征图、第一模板特征图、第二模板特征图和第三模板特征图输入冗余特征过滤网络中，将第一中间特征图和第一模板特征图进行Concat操作，再经过大小为1
×
1的卷积核，得到第一过滤特征图；将第二中间特征图和第二模板特征图进行Concat操作，再经过大小为1
×
1的卷积核，得到第二过滤特征图，将第三中间特征图和第三模板特征图进行Concat操作，再经过大小为1
×
1的卷积核，得到第三过滤特征图。
[0014]作为优选，第一过滤特征图、第二过滤特征图和第三过滤特征图输入融合卷积块注意力模块的路径聚合网络中，第一过滤特征图输入第一DWConv模块，得到第一融合特征图，将第一融合特征图经由第一CBAM模块得到的特征图与第二过滤特征图进行Concat操作，再经由5个第二DWConv模块进行特征融合，得到第二融合特征图，将第二融合特征图经由第三DWConv模块与第二CBAM模块得到的特征图与第三过滤特征图进行Concat操作，再经由通过5个第四DWConv模块进行特征融合，得到第三融合特征图，将第一融合特征图、第二融合特征图和第三融合特征图输入头部预测网络，得到布匹检测结果。
[0015]作为优选，多类聚焦置信度损失函数为：
[0016][0017]其中，y表示标签，p表示为置信度，(1
‑
p)
γ
表示为难易度平衡因子，i表示第i种布匹瑕疵类别，N表示布匹瑕疵类别总数，β
c
表示为类别平衡因子：
[0018]β
c
＝ω/N
c
；
[0019]其中，ω表示平衡系数，N
c
表示类别为c的样本数量。
[0020]第二方面，本专利技术提供了一种布匹瑕疵检测装置，包括：
[0021]图像获取模块，被配置为获取待检测的布匹图像和模板图像；
[0022]模型构建模块，被配置为构建改进的Yolov5s网络并训练，得到布匹瑕疵检测模型，改进的Yolov5s网络包括主干特征提取网络、颈部特征融合网络和头部预测网络，主干特征提取网络采用CSPDarkNet，颈部特征融合网络包括特征过滤金字塔网络和融合卷积块注意力模块的路径聚合网络，在训练过程中采用多类聚焦置信度损失函数代替交叉熵置信度损失函数；
[0023]预测模块，被配置为将待检测的布匹图像和模板图像输入布匹瑕疵检测模型，得到布匹检测结果。
[0024]第三方面，本专利技术提供了一种电子设备，包括一个或多个处理器；存储装置，用于存储一个或多个程序，当一个或多个程序被一个或多个处理器执行，使得一个或多个处理器实现如第一方面中任一实现方式描述的方法。
[0025]第四方面，本专利技术提供了一种计算机可读存储介质，其上存储有计算机程序，该计算机程序被处理器执行时实现如第一方面中任一实现方式描述的方法。
[0026]相比于现有技术，本专利技术具有以下有益效果：
[0027](1)本专利技术提出的布匹瑕疵检测方法通过特征过滤金字塔网络与融合卷积块注意力模块的路径聚合网络构成的本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种布匹瑕疵检测方法，其特征在于，包括以下步骤：获取待检测的布匹图像和模板图像；构建改进的Yolov5s网络并训练，得到布匹瑕疵检测模型，所述改进的Yolov5s网络包括主干特征提取网络、颈部特征融合网络和头部预测网络，所述主干特征提取网络采用CSPDarkNet，所述颈部特征融合网络包括特征过滤金字塔网络和融合卷积块注意力模块的路径聚合网络，在训练过程中采用多类聚焦置信度损失函数代替交叉熵置信度损失函数；将所述待检测的布匹图像和模板图像输入所述布匹瑕疵检测模型，得到布匹检测结果。2.根据权利要求1所述的布匹瑕疵检测方法，其特征在于，所述CSPDarkNet包括依次连接的第一单元、第二单元、第三单元、第四单元和第五单元，所述第一单元包括依次串联的Focus模块和第一CBS模块，所述第二单元包括依次串联的第二CBS模块和第一CSP1_x模块，所述第三单元包括依次串联的第三CBS模块和第二CSP1_x模块，所述第四单元包括依次串联的第四CBS模块和第三CSP1_x模块，所述第五单元包括依次串联的第五CBS模块、第四CSP1_x模块和SPPF模块，将所述布匹图像输入所述CSPDarkNet，在所述第三单元、所述第四单元和第五单元分别输出得到第一有效特征图、第二有效特征图和第三有效特征图。3.根据权利要求2所述的布匹瑕疵检测方法，其特征在于，所述特征过滤金字塔网络包括FPN特征融合网络、冗余特征过滤网络和模板特征提取网络，所述第一有效特征图、第二有效特征图、第三有效特征图和模板图像分别输入所述FPN特征融合网络中，其中，所述第三有效特征图输入第六CBS模块，得到第三中间特征图，将所述第三中间特征图经由第一Upsample模块得到特征图与所述第二有效特征图进行Concat操作，再经由第一CSP2_3模块与第七CBS模块进行融合，得到第二中间特征图，将所述第二中间特征图经由第二Upsample模块得到的特征图与所述第一有效特征图进行Concat操作，再经由第二CSP2_3模块与第八CBS模块进行融合，得到第一中间特征图。4.根据权利要求3所述的布匹瑕疵检测方法，其特征在于，所述模板图像输入所述模板特征提取网络中，所述模板图像输入第九CBS模块，得到第一模板特征图，将所述第一模板特征图输入Downsample1模块，得到第二模板特征图，所述第二模板特征图输入Downsample2模块，得到第三模板特征图。5.根据权利要求4所述的布匹瑕疵检测方法，其特征在于，所述第一中间特征图、第二中间特征图、第三中间特征图、第一模板特征图、第二模板特征图和第三模板特征图输入所述冗余特征过滤网络中，将所述第一中间特征图和第一模板特征图进行Concat操作，再经过大小为1
×
1的卷积核，得到第一过滤特征图；将所述第二中间特征图和第二模板...

【专利技术属性】
技术研发人员：黄德天，宋佳讯，林明昕，朱达欣，蔡丹玲，曾焕强，陈婧，
申请(专利权)人：华侨大学，
类型：发明
国别省市：