基于機器視覺的高鐵碳滑板圖像分割算法研究
現(xiàn)代制造工程
頁數(shù): 9 2024-09-18
摘要: 針對語義分割模型識別碳滑板邊緣困難、復雜背景干擾性較大以及特征信息丟失嚴重等問題,提出一種新型編解碼結(jié)構(gòu)的Swin Transformer語義分割優(yōu)化算法。首先,主干網(wǎng)絡采用U型的編解碼結(jié)構(gòu),實現(xiàn)多尺度的信息融合;其次,添加注意力局部增強感知模塊來擴大感受野并提高模型泛化能力;然后,采用具有數(shù)據(jù)相關(guān)性的上采樣結(jié)構(gòu),以提高上采樣質(zhì)量,擺脫分辨率對預測結(jié)果的影響,加強圖像重建能力;... (共9頁)