基于改進(jìn)YOLO v3的軸承端面缺陷檢測算法

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