search

sjtuplayer / anomalydiffusion

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

Dirty and so many repeated code #34

Closed DonghoonPark12 closed 3 months ago

DonghoonPark12 commented 3 months ago

and poorly organized code.

In 'Personalized_mvtec_mask' class why did author read 'mask' and name it as 'image' ??

There is something more confusing.

engrmusawarali commented 3 months ago

I think this is used to train the diffusion model to learn the mask distribution. Although I agree that the name is not correct. You can refer 'Personalized_mvtec_encoder' for the diffusion model which is used to generate anomalies.

Note: I am not the author of this paper, I am just replying to you based on my intuitions.