基于改進(jìn)Skip-GAN的高速鐵路巡檢圖像異常檢測(cè)
鐵道學(xué)報(bào)
頁(yè)數(shù): 8 2024-09-15
摘要: 針對(duì)高速鐵路探傷巡檢過(guò)程中普遍存在的訓(xùn)練樣本不平衡和打標(biāo)簽復(fù)雜等問(wèn)題,提出一種改進(jìn)的Skip-GAN算法,對(duì)巡檢圖像中存在的多類病害進(jìn)行無(wú)監(jiān)督檢測(cè)。對(duì)高速鐵路巡檢圖像進(jìn)行預(yù)處理,包括軌道板分割和數(shù)據(jù)增強(qiáng),減少排水溝和明暗環(huán)境對(duì)網(wǎng)絡(luò)重構(gòu)圖像的影響;對(duì)Skip-GAN結(jié)構(gòu)進(jìn)行改進(jìn),包括增加注意力機(jī)制模塊、修改判別器為雙自動(dòng)編碼器結(jié)構(gòu)等;改進(jìn)卷積神經(jīng)網(wǎng)絡(luò)的損失函數(shù),增加網(wǎng)絡(luò)對(duì)圖像的重... (共8頁(yè))