Julian-Wyatt / AnoDDPM

CVPR Workshop paper - AnoDDPM: Anomaly Detection with Denoising Diffusion Probabilistic Models using Simplex Noise
https://julianwyatt.co.uk/anoddpm
MIT License
154 stars 27 forks source link

How to train and eval AnoDDPM #1

Closed jinkun-hao closed 1 year ago

jinkun-hao commented 2 years ago

Hi,this is a nice work! I have read your paper and is interested in applying Diffusion model to anomaly detection task. Could you add more detailed information about how to train and evaluate AnoDDPM in README file?

Julian-Wyatt commented 1 year ago

Hi, apologies for the late reply

I've since updated my readme, but could likely still be lacking the detail required. The project was written with mri data files in mind and would need to refactor the code to add your own dataset, by modifying the existing data loaders, and therefore also where the loaders are instantiated throughout the training and evaluation scripts (located in detection.py)

Please reach out if there are any further issues?

jinkun-hao commented 1 year ago

Thanks for the update! I'll take a closer look.