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