基于改進YOLOv5的軸承表面缺陷檢測

摘要： 傳統(tǒng)的軸承表面缺陷檢測由于缺陷目標較小，錯檢漏檢率高，檢測效率低等問題，為此提出一種基于YOLOv5網絡改進的算法模型。首先，在主干網絡中添加高效通道注意力機制(efficient channel attention, ECA),增強網絡的特征提取能力，集中關注各種影響軸承質量的重點信息；其次，在YOLOv5網絡基礎上添加小目標檢測層，通過補充融合特征層和引入額外檢測頭，提高網...