donggong1 / memae-anomaly-detection

MemAE for anomaly detection. -- Gong, Dong, et al. "Memorizing Normality to Detect Anomaly: Memory-augmented Deep Autoencoder for Unsupervised Anomaly Detection". ICCV 2019.
https://donggong1.github.io/anomdec-memae.html
MIT License
465 stars 103 forks source link

MemAE for 2D datasets (MNIST, CIFAR-10) #34

Open samanmohammadraouf opened 2 months ago

samanmohammadraouf commented 2 months ago

Can you please provide the code for 2D version? The training code of the model for 2D datasets (like MNIST, CIFAR-10)? Thanks.