├── .gitignore ├── LICENSE ├── README.md ├── data_root ├── .gitkeep └── CIFAR10 │ └── cifar-10-python.tar.gz ├── images └── .gitkeep ├── logs └── .gitkeep ├── main.py ├── readme_imgs ├── model_graphs.png ├── srvae_generations_x.jpg ├── srvae_generations_y.jpg ├── srvae_super_res.jpg ├── svae_inter1.jpg ├── svae_inter2.jpg ├── vae_generations.jpg ├── vae_inter1.jpg ├── vae_inter2.jpg ├── vae_inter3.jpg └── vae_recon.jpg └── src ├── __init__.py ├── data ├── __init__.py ├── dataloaders.py └── datasets.py ├── models ├── __init__.py ├── srvae │ ├── __init__.py │ ├── image_networks │ │ └── densenet16x32.py │ └── srvae.py └── vae │ ├── __init__.py │ ├── image_networks │ └── densenet32.py │ └── vae.py ├── modules ├── __init__.py ├── distributions.py ├── loss.py ├── nn_layers.py ├── optim.py ├── priors │ ├── __init__.py │ ├── mog.py │ ├── prior.py │ ├── realnvp │ │ ├── __init__.py │ │ ├── distributions │ │ │ ├── __init__.py │ │ │ ├── mog.py │ │ │ └── standard_normal.py │ │ ├── model │ │ │ ├── __init__.py │ │ │ ├── coupling_layer.py │ │ │ └── real_nvp.py │ │ └── util │ │ │ ├── __init__.py │ │ │ ├── array_util.py │ │ │ └── norm_util.py │ └── standard_normal.py └── train.py └── utils ├── __init__.py ├── args.py ├── plotting.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/README.md -------------------------------------------------------------------------------- /data_root/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_root/CIFAR10/cifar-10-python.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/data_root/CIFAR10/cifar-10-python.tar.gz -------------------------------------------------------------------------------- /images/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /logs/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/main.py -------------------------------------------------------------------------------- /readme_imgs/model_graphs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/readme_imgs/model_graphs.png -------------------------------------------------------------------------------- /readme_imgs/srvae_generations_x.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/readme_imgs/srvae_generations_x.jpg -------------------------------------------------------------------------------- /readme_imgs/srvae_generations_y.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/readme_imgs/srvae_generations_y.jpg -------------------------------------------------------------------------------- /readme_imgs/srvae_super_res.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/readme_imgs/srvae_super_res.jpg -------------------------------------------------------------------------------- /readme_imgs/svae_inter1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/readme_imgs/svae_inter1.jpg -------------------------------------------------------------------------------- /readme_imgs/svae_inter2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/readme_imgs/svae_inter2.jpg -------------------------------------------------------------------------------- /readme_imgs/vae_generations.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/readme_imgs/vae_generations.jpg -------------------------------------------------------------------------------- /readme_imgs/vae_inter1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/readme_imgs/vae_inter1.jpg -------------------------------------------------------------------------------- /readme_imgs/vae_inter2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/readme_imgs/vae_inter2.jpg -------------------------------------------------------------------------------- /readme_imgs/vae_inter3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/readme_imgs/vae_inter3.jpg -------------------------------------------------------------------------------- /readme_imgs/vae_recon.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/readme_imgs/vae_recon.jpg -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/data/__init__.py -------------------------------------------------------------------------------- /src/data/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/data/dataloaders.py -------------------------------------------------------------------------------- /src/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/data/datasets.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/models/__init__.py -------------------------------------------------------------------------------- /src/models/srvae/__init__.py: -------------------------------------------------------------------------------- 1 | from .srvae import srVAE 2 | -------------------------------------------------------------------------------- /src/models/srvae/image_networks/densenet16x32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/models/srvae/image_networks/densenet16x32.py -------------------------------------------------------------------------------- /src/models/srvae/srvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/models/srvae/srvae.py -------------------------------------------------------------------------------- /src/models/vae/__init__.py: -------------------------------------------------------------------------------- 1 | from .vae import VAE 2 | -------------------------------------------------------------------------------- /src/models/vae/image_networks/densenet32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/models/vae/image_networks/densenet32.py -------------------------------------------------------------------------------- /src/models/vae/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/models/vae/vae.py -------------------------------------------------------------------------------- /src/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/__init__.py -------------------------------------------------------------------------------- /src/modules/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/distributions.py -------------------------------------------------------------------------------- /src/modules/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/loss.py -------------------------------------------------------------------------------- /src/modules/nn_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/nn_layers.py -------------------------------------------------------------------------------- /src/modules/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/optim.py -------------------------------------------------------------------------------- /src/modules/priors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/__init__.py -------------------------------------------------------------------------------- /src/modules/priors/mog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/mog.py -------------------------------------------------------------------------------- /src/modules/priors/prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/prior.py -------------------------------------------------------------------------------- /src/modules/priors/realnvp/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import RealNVP 2 | -------------------------------------------------------------------------------- /src/modules/priors/realnvp/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/realnvp/distributions/__init__.py -------------------------------------------------------------------------------- /src/modules/priors/realnvp/distributions/mog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/realnvp/distributions/mog.py -------------------------------------------------------------------------------- /src/modules/priors/realnvp/distributions/standard_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/realnvp/distributions/standard_normal.py -------------------------------------------------------------------------------- /src/modules/priors/realnvp/model/__init__.py: -------------------------------------------------------------------------------- 1 | from .real_nvp import RealNVP 2 | -------------------------------------------------------------------------------- /src/modules/priors/realnvp/model/coupling_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/realnvp/model/coupling_layer.py -------------------------------------------------------------------------------- /src/modules/priors/realnvp/model/real_nvp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/realnvp/model/real_nvp.py -------------------------------------------------------------------------------- /src/modules/priors/realnvp/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/realnvp/util/__init__.py -------------------------------------------------------------------------------- /src/modules/priors/realnvp/util/array_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/realnvp/util/array_util.py -------------------------------------------------------------------------------- /src/modules/priors/realnvp/util/norm_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/realnvp/util/norm_util.py -------------------------------------------------------------------------------- /src/modules/priors/standard_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/priors/standard_normal.py -------------------------------------------------------------------------------- /src/modules/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/modules/train.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/utils/args.py -------------------------------------------------------------------------------- /src/utils/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/utils/plotting.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ioangatop/srVAE/HEAD/src/utils/utils.py --------------------------------------------------------------------------------