基于快速擴(kuò)散生成模型的地鐵軌道異物入侵檢測算法研究
鐵道標(biāo)準(zhǔn)設(shè)計
頁數(shù): 10 2023-04-19
摘要: 針對現(xiàn)有基于無監(jiān)督的地鐵軌道異物檢測算法中,對不同視角需要進(jìn)行單獨(dú)訓(xùn)練模型的問題,提出一種基于快速擴(kuò)散生成模型的統(tǒng)一模型檢測算法。該模型基于擴(kuò)散模型,在僅使用正常樣本訓(xùn)練一個模型的條件下,實現(xiàn)對多個視角異物入侵檢測。檢測時,擴(kuò)散模型對存在異物的圖像部分重構(gòu)為無異物,然后將重構(gòu)圖像和原圖進(jìn)行比對得出存在異物的位置。研究發(fā)現(xiàn),通過合理設(shè)計擴(kuò)散步長和采樣時刻,擴(kuò)散模型能夠在不需要擴(kuò)散... (共10頁)