基于快速扩散生成模型的地铁轨道异物入侵检测算法研究
铁道标准设计
页数: 10 2023-04-19
摘要: 针对现有基于无监督的地铁轨道异物检测算法中,对不同视角需要进行单独训练模型的问题,提出一种基于快速扩散生成模型的统一模型检测算法。该模型基于扩散模型,在仅使用正常样本训练一个模型的条件下,实现对多个视角异物入侵检测。检测时,扩散模型对存在异物的图像部分重构为无异物,然后将重构图像和原图进行比对得出存在异物的位置。研究发现,通过合理设计扩散步长和采样时刻,扩散模型能够在不需要扩散... (共10页)