輕量化YOLOv5n的木材表面缺陷檢測模型

摘要： 為解決當(dāng)前木材表面缺陷檢測模型計算量大，在移動端或嵌入式設(shè)備部署困難的問題，本文提出了一種基于輕量化改進(jìn)YOLOv5n的模型——FCS-YOLOv5n。首先，引入FasterNet替換主干網(wǎng)絡(luò)，減少計算冗余及內(nèi)存訪問數(shù)量。其次，使用采樣算子Carafe，進(jìn)一步壓縮模型參數(shù)量，保證語義信息獲取的全面性。然后，添加參數(shù)量和計算量較小的SE注意力機(jī)制，提高特征提取和融合效果。最后，用... （共9頁）