基于改進YOLOv6模型的微特電機電樞表面缺陷檢測

摘要： 針對傳統(tǒng)工業(yè)存在對微特電機電樞表面缺陷檢測任務人工成本高、工作量大的問題，提出了一種改進YOLOv6模型的微特電機表面缺陷檢測算法。首先，在主干網(wǎng)絡加入SimAM注意力模塊，加強網(wǎng)絡信息傳遞，提高模型對特征的敏感程度；其次，Neck端使用GSConv新型卷積方式，以減少模型計算量；最后，使用CIoU損失函數(shù)解決GIoU損失函數(shù)的局限性，以提升模型檢測精度。將所提改進算法在微特電... （共6頁）