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

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