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