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