talafek96 / Diffusion-AD-Project

In this project we will attempt to develop a POC for detecting anomalies in images based on the ability or inability of a DDM to reconstruct them.
1 stars 1 forks source link

Thanks for your great work. Is there any code to train the model? Thank you very much #38

Closed 927514606 closed 1 year ago

talafek96 commented 1 year ago

Thank you. The current version is only a zero-shot (no training) POC for the task of anomaly detection using a pre-trained model. We are currently working on a few-shot version in order to also see the performance over few-shot optimization (however we are currently running in to some technical issues while trying to fine-tune the pretrained model).