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