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