基于改進RefineDet的織物疵點檢測
打開文本圖片集
摘要:為了實現(xiàn)織物疵點的自動檢測與分類,提出了一種基于改進RefineDet的疵點檢測方法。首先,將VGG16改為全卷積網(wǎng)絡對織物圖像特征進行提取;其次,為了獲取疵點重要的特征并抑制不必要的特征,在Anchor細化模塊(Anchor refinement module,ARM)中加入了注意力機制;為了提高網(wǎng)絡的分類性能,在傳輸連接塊(Transfer connection block,TCB)中加入了SE模塊(Squeeze and excitation,SE);最后,目標檢測模塊(Object detection module,ODM)將檢測的結果回歸到準確的目標位置,并預測疵點的類別,對疵點進行定位。(剩余11385字)