No such file or directory: 'logs/anomaly-checkpoints/checkpoints/spatial_encoder.pt'

sjtuplayer / anomalydiffusion

[AAAI 2024] AnomalyDiffusion: Few-Shot Anomaly Image Generation with Diffusion Model

MIT License

114 stars 14 forks source link

Closed xuyu666 closed 4 months ago

xuyu666 commented 4 months ago

作者您好：很高兴看到您发布了推理代码~

我在运行推理代码过程中遇到这个错误，貌似您提供的anomaly-checkpoints缺失了spatial_encoder.pt。或许其需要训练来获得，但为了快速验证论文效果，可能我也想获得一个预训练模型，先看看具体效果。请问可以提供下么，或者是我某个步骤遗漏了。

sjtuplayer commented 4 months ago

这是命名的问题，将之前下载的ht_encoder改名为spatial_encoder即可

xuyu666 commented 4 months ago

好的哈，谢谢作者快速的回复~ PS: 压缩包内部的文件夹的名称拼写貌似不对。（可能导致不易发现的错误）

sjtuplayer commented 4 months ago

感谢提醒，已修改