基于改進(jìn)Swin-Transformer模型的銅礦X射線圖像分類研究
有色金屬(選礦部分)
頁數(shù): 8 2024-11-04
摘要: 針對銅礦圖像分類中傳統(tǒng)神經(jīng)網(wǎng)絡(luò)因感受野限制和維度信息阻塞面臨的問題,提出了基于X射線透射成像技術(shù)改進(jìn)的Swin-Transformer模型。該模型以Swin-Transformer為基礎(chǔ)框架,在主干網(wǎng)絡(luò)的第二和第三階段中添加Mixing Block,通過局部窗口自注意力和深度卷積之間的的雙向交互,使模型的感受野得到顯著增大,從而增強(qiáng)了特征表示和建模能力;同時,引入的EMA(Ef... (共8頁)