cnulab / RealNet

Offical implementation of "RealNet: A Feature Selection Network with Realistic Synthetic Anomaly for Anomaly Detection (CVPR 2024)"
MIT License
239 stars 14 forks source link

作者您好!关于def validate返回的c和 recon_images #49

Open leejq666 opened 3 months ago

leejq666 commented 3 months ago

我看代码上gen_images是不同程度异常的拼接,然后 recon_images中的x0_det也涉及使用DDIM进行异常合成。这两部分返回值的区别是什么?如果我想合成最终的异常图像应该以哪个为主?期待您的回复!

leejq666 commented 3 months ago

还有就是,在训练了17881iter之后,diffusion_vis文件夹中还是空的(可能和我早停有关系),没有保存可视化结果。问一下您这个文件夹的作用是什么

cnulab commented 3 months ago

你好!

  1. 关于那个拼接图像的含义,你可以参考https://github.com/cnulab/RealNet/issues/40 .
  2. 默认每10个epoch设置一个checkpoint,保存可视化结果到diffusion_vis,如果是本文的数据集,那17881个iter肯定到10个epoch了,应该保存了。如果使用了自己的数据集,17881没到10个epoch,那就没保存。
leejq666 commented 3 months ago

你好!

  1. 关于那个拼接图像的含义,你可以参考关于自己数据集生成的问题 #40 .
  2. 默认每10个epoch设置一个checkpoint,保存可视化结果到diffusion_vis,如果是本文的数据集,那17881个iter肯定到10个epoch了,应该保存了。如果使用了自己的数据集,17881没到10个epoch,那就没保存。

ok明白了!感谢