search

sjtuplayer / anomalydiffusion

[AAAI 2024] AnomalyDiffusion: Few-Shot Anomaly Image Generation with Diffusion Model
MIT License
114 stars 14 forks source link

Trained checkpoints for sampling anomalous images #49

Closed supriyasathya closed 1 month ago

supriyasathya commented 2 months ago

Hi,

I have a question: How do we use the trained checkpoints for anomaly-checkpoints uploaded on the drive to sample the anomalous images?

Thanks

sjtuplayer commented 2 months ago

Hi, you can refer to run-mvtec.py which uses the checkpoints by running generate_with_mask.py.