輕量化YOLOv5n的木材表面缺陷檢測(cè)模型

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