基于改進YOLO v3的軸承端面缺陷檢測算法

摘要： 為提高軸承端面缺陷檢測的速度以及檢測精度，提出一種基于改進YOLO v3的軸承端面缺陷檢測算法。首先，對圖像數(shù)據(jù)集進行數(shù)據(jù)增強處理以防止產(chǎn)生過擬合現(xiàn)象；其次，通過改進K-means聚類算法重新聚類出目標檢測的Anchor Boxes,并引入SKNet注意力機制模塊對原網(wǎng)絡(luò)結(jié)構(gòu)以及輸出層結(jié)構(gòu)進行改進；最后對改進的YOLO v3算法進行實驗驗證，并與原YOLO v3算法進行對比分析...