andreaskapou / spatialDGM

Spatial deep generative models

spatialDGM

Spatial deep generative models.

Implementation of spatialVAE paper.

Working directory structure:

.
├── LICENSE
├── README.md
├── configs
│	└── vae_mnist.yaml
├── data
│	└── MNIST
├── models 				# spatialVAE and VAE models
├── notebooks 			# Sample notebooks for quick analysis
├── output              # Pretrained models with different latent dimensions z
└── src        			# Main scripts to run VAE models