[發明專利]一種紡織物表面缺陷檢測方法及相關設備

在審

• 申請號: 202310476352.3
• 申請日: 2023-04-28
• 公開（公告）號: CN116385426A
• 公開（公告）日: 2023-07-04
• 發明（設計）人: 李海霞;鄧曉衡
• 申請（專利權）人: 中南大學
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06V10/82;G06N3/045;G06N3/0464;G06N3/08
• 代理公司: 長沙軒榮專利代理有限公司 43235
• 代理人: 張慧敏

1.一種紡織物表面缺陷檢測方法，其特征在于，包括：

步驟1，采集紡織物的表面缺陷圖像測試集；

步驟2，將所述表面缺陷圖像測試集輸入至改進的Yolov7網絡進行訓練，得到目標檢測最優模型；

步驟3，將待檢測的紡織物表面圖像輸入至所述目標檢測最優模型進行表面缺陷檢測，得到檢測結果；

所述改進的Yolov7網絡包括：輸入模塊、用于特征提取的骨干模塊、用于特征融合的頭部模塊、用于檢測小缺陷的檢測模塊和用于缺陷類別預測的Transformer預測頭；

所述輸入模塊的輸出端與所述骨干模塊的輸入端連接，所述骨干模塊的第一輸出端與所述檢測模塊的第一輸入端連接，所述骨干模塊的第二輸出端與所述頭部模塊的輸入端連接，所述頭部模塊的輸出端與所述檢測模塊的第二輸入端連接，所述檢測模塊的輸出端與所述頭部模塊的輸入端連接，所述檢測模塊的輸出端與所述頭部模塊的輸出端均與所述Transformer預測頭連接；

所述輸入模塊作為所述改進的Yolov7網絡的輸入端，所述Transformer預測頭的輸出端作為改進的Yolov7網絡的輸出端。

2.根據權利要求1所述的紡織物表面缺陷檢測方法，其特征在于，所述骨干模塊包括：

依次連接的第一CBS子模塊、第二CBS子模塊、第三CBS子模塊、第四CBS子模塊、第一高效層聚合網絡子模塊、由最大池化層和卷積層構成的第一MPC子模塊、第二高效層聚合網絡子模塊、由最大池化層和卷積層構成的第二MPC子模塊、第三高效層聚合網絡子模塊、由最大池化層和卷積層構成的第三MPC子模塊和第四高效層聚合網絡子模塊；

所述第一CBS子模塊的輸入端與所述輸入模塊的輸出端連接，所述第一高效層聚合網絡子模塊的輸出端與所述檢測模塊的第一輸入端連接，所述第二高效層聚合網絡子模塊的輸出端與所述頭部模塊的第一輸入端連接，所述第三高效層聚合網絡子模塊的輸出端與所述頭部模塊的第二輸入端連接，所述第四高效層聚合網絡子模塊的輸出端與所述頭部模塊的第三輸入端連接。

3.根據權利要求2所述的紡織物表面缺陷檢測方法，其特征在于，所述頭部模塊包括：

第五CBS子模塊、第六CBS子模塊、第七CBS子模塊、第八CBS子模塊、第一拼接子模塊、第二拼接子模塊、第三拼接子模塊、第四拼接子模塊、第一上采樣子模塊、第二上采樣子模塊、第五高效層聚合網絡子模塊、第六高效層聚合網絡子模塊、第七高效層聚合網絡子模塊、第八高效層聚合網絡子模塊、由最大池化層和卷積層構成的第四MPC子模塊、由最大池化層和卷積層構成的第五MPC子模塊、第一卷積子模塊、第二卷積子模塊、第三卷積子模塊和空間金字塔池化子模塊；

所述第五CBS子模塊的輸入端與所述第二高效層聚合網絡子模塊的輸出端連接，所述第五CBS子模塊的輸出端與所述第一拼接子模塊的輸入端連接，所述第七CBS子模塊的輸入端與所述第三高效層聚合網絡子模塊的輸出端連接，所述第七CBS子模塊的輸出端與所述第三拼接子模塊的輸入端連接，所述空間金字塔池化子模塊的輸入端與所述第四高效層聚合網絡子模塊的輸出端連接，所述金字塔池化子模塊的輸出端分別與所述第八CBS子模塊的輸入端、所述第四拼接子模塊的輸入端連接；

所述第八CBS子模塊的輸出端與所述第二上采樣子模塊的輸入端連接，所述第二上采樣子模塊的輸出端與所述第三拼接子模塊、所述第五高效層聚合網絡子模塊、所述第六CBS子模塊、所述第一上采樣子模塊、所述第一拼接子模塊依次連接，所述第一拼接子模塊的輸出端與所述檢測模塊的第二輸入端連接；

所述檢測模塊的第一輸出端與所述第六高效層聚合網絡子模塊的輸入端連接，所述第六高效層聚合網絡子模塊的輸出端分別與所述第一卷積子模塊的輸入端、所述第四MPC子模塊的輸入端連接，所述第四MPC子模塊的輸出端、所述第六CBS子模塊的輸出端均與所述第二拼接子模塊的輸入端連接，所述第二拼接子模塊的輸出端與所述第七高效層聚合網絡子模塊的輸入端連接，所述第七高效層聚合網絡子模塊的輸出端分別與所述第二卷積子模塊的輸入端、所述第五MPC子模塊的輸入端，所述第五MPC子模塊的輸出端與所述第四拼接子模塊的輸入端連接，所述第四拼接子模塊的輸出端與所述第八高效層聚合網絡子模塊的輸入端連接，所述第八高效層聚合網絡子模塊的輸出端與所述第三卷積子模塊的輸入端連接，所述第三卷積子模塊的輸出端、所述第二卷積子模塊的輸出端、所述第一卷積子模塊的輸出端均與所述Transformer預測頭的輸入端連接。