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