├── .gitignore ├── README.md ├── experiment_runner.py ├── experiments_launcher.sh ├── generation_launcher.sh ├── generation_parser.py ├── make_experiment.py ├── models ├── __init__.py ├── base.py ├── config.py ├── generation.py ├── riemann_tools.py └── vae.py ├── notebooks └── example.ipynb ├── requirements.txt ├── trained_models ├── RHVAE_train_EMNIST_M_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001 │ └── RHVAE_train_EMNIST_M.model ├── RHVAE_train_Fashion_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001 │ └── RHVAE_train_Fashion.model ├── RHVAE_train_Fashion_ldim_2_nlf_3_epslf_0.01_T_0.8_lbd_0.001 │ └── RHVAE_train_Fashion.model ├── RHVAE_train_MNIST_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001 │ └── RHVAE_train_MNIST.model ├── RHVAE_train_MNIST_ldim_2_nlf_3_epslf_0.01_T_0.8_lbd_0.001 │ └── RHVAE_train_MNIST.model ├── RHVAE_train_Shapes_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001 │ └── RHVAE_train_Shapes.model ├── RHVAE_train_Shapes_ldim_2_nlf_3_epslf_0.01_T_0.8_lbd_0.001 │ └── RHVAE_train_Shapes.model ├── VAE_train_EMNIST_M_ldim_10 │ └── VAE_train_EMNIST_M.model ├── VAE_train_Fashion_ldim_10 │ └── VAE_train_Fashion.model ├── VAE_train_Fashion_ldim_2 │ └── VAE_train_Fashion.model ├── VAE_train_MNIST_ldim_10 │ └── VAE_train_MNIST.model ├── VAE_train_MNIST_ldim_2 │ └── VAE_train_MNIST.model ├── VAE_train_Shapes_ldim_2 │ └── VAE_train_Shapes.model └── peers │ ├── VAMP_train_EMNIST_M_ldim_10 │ ├── VAMP_train_Fashion_ldim_10 │ ├── VAMP_train_Fashion_ldim_2 │ ├── VAMP_train_MNIST_ldim_10 │ ├── VAMP_train_MNIST_ldim_2 │ ├── VAMP_train_Shapes_ldim_10 │ └── VAMP_train_Shapes_ldim_2 ├── trainers ├── __init__.py └── training.py ├── training_data ├── train_EMNIST_M ├── train_Fashion ├── train_MNIST └── train_Shapes └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/README.md -------------------------------------------------------------------------------- /experiment_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/experiment_runner.py -------------------------------------------------------------------------------- /experiments_launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/experiments_launcher.sh -------------------------------------------------------------------------------- /generation_launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/generation_launcher.sh -------------------------------------------------------------------------------- /generation_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/generation_parser.py -------------------------------------------------------------------------------- /make_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/make_experiment.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/models/base.py -------------------------------------------------------------------------------- /models/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/models/config.py -------------------------------------------------------------------------------- /models/generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/models/generation.py -------------------------------------------------------------------------------- /models/riemann_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/models/riemann_tools.py -------------------------------------------------------------------------------- /models/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/models/vae.py -------------------------------------------------------------------------------- /notebooks/example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/notebooks/example.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/requirements.txt -------------------------------------------------------------------------------- /trained_models/RHVAE_train_EMNIST_M_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001/RHVAE_train_EMNIST_M.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/RHVAE_train_EMNIST_M_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001/RHVAE_train_EMNIST_M.model -------------------------------------------------------------------------------- /trained_models/RHVAE_train_Fashion_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001/RHVAE_train_Fashion.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/RHVAE_train_Fashion_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001/RHVAE_train_Fashion.model -------------------------------------------------------------------------------- /trained_models/RHVAE_train_Fashion_ldim_2_nlf_3_epslf_0.01_T_0.8_lbd_0.001/RHVAE_train_Fashion.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/RHVAE_train_Fashion_ldim_2_nlf_3_epslf_0.01_T_0.8_lbd_0.001/RHVAE_train_Fashion.model -------------------------------------------------------------------------------- /trained_models/RHVAE_train_MNIST_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001/RHVAE_train_MNIST.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/RHVAE_train_MNIST_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001/RHVAE_train_MNIST.model -------------------------------------------------------------------------------- /trained_models/RHVAE_train_MNIST_ldim_2_nlf_3_epslf_0.01_T_0.8_lbd_0.001/RHVAE_train_MNIST.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/RHVAE_train_MNIST_ldim_2_nlf_3_epslf_0.01_T_0.8_lbd_0.001/RHVAE_train_MNIST.model -------------------------------------------------------------------------------- /trained_models/RHVAE_train_Shapes_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001/RHVAE_train_Shapes.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/RHVAE_train_Shapes_ldim_10_nlf_3_epslf_0.01_T_1.5_lbd_0.001/RHVAE_train_Shapes.model -------------------------------------------------------------------------------- /trained_models/RHVAE_train_Shapes_ldim_2_nlf_3_epslf_0.01_T_0.8_lbd_0.001/RHVAE_train_Shapes.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/RHVAE_train_Shapes_ldim_2_nlf_3_epslf_0.01_T_0.8_lbd_0.001/RHVAE_train_Shapes.model -------------------------------------------------------------------------------- /trained_models/VAE_train_EMNIST_M_ldim_10/VAE_train_EMNIST_M.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/VAE_train_EMNIST_M_ldim_10/VAE_train_EMNIST_M.model -------------------------------------------------------------------------------- /trained_models/VAE_train_Fashion_ldim_10/VAE_train_Fashion.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/VAE_train_Fashion_ldim_10/VAE_train_Fashion.model -------------------------------------------------------------------------------- /trained_models/VAE_train_Fashion_ldim_2/VAE_train_Fashion.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/VAE_train_Fashion_ldim_2/VAE_train_Fashion.model -------------------------------------------------------------------------------- /trained_models/VAE_train_MNIST_ldim_10/VAE_train_MNIST.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/VAE_train_MNIST_ldim_10/VAE_train_MNIST.model -------------------------------------------------------------------------------- /trained_models/VAE_train_MNIST_ldim_2/VAE_train_MNIST.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/VAE_train_MNIST_ldim_2/VAE_train_MNIST.model -------------------------------------------------------------------------------- /trained_models/VAE_train_Shapes_ldim_2/VAE_train_Shapes.model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/VAE_train_Shapes_ldim_2/VAE_train_Shapes.model -------------------------------------------------------------------------------- /trained_models/peers/VAMP_train_EMNIST_M_ldim_10: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/peers/VAMP_train_EMNIST_M_ldim_10 -------------------------------------------------------------------------------- /trained_models/peers/VAMP_train_Fashion_ldim_10: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/peers/VAMP_train_Fashion_ldim_10 -------------------------------------------------------------------------------- /trained_models/peers/VAMP_train_Fashion_ldim_2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/peers/VAMP_train_Fashion_ldim_2 -------------------------------------------------------------------------------- /trained_models/peers/VAMP_train_MNIST_ldim_10: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/peers/VAMP_train_MNIST_ldim_10 -------------------------------------------------------------------------------- /trained_models/peers/VAMP_train_MNIST_ldim_2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/peers/VAMP_train_MNIST_ldim_2 -------------------------------------------------------------------------------- /trained_models/peers/VAMP_train_Shapes_ldim_10: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/peers/VAMP_train_Shapes_ldim_10 -------------------------------------------------------------------------------- /trained_models/peers/VAMP_train_Shapes_ldim_2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trained_models/peers/VAMP_train_Shapes_ldim_2 -------------------------------------------------------------------------------- /trainers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trainers/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/trainers/training.py -------------------------------------------------------------------------------- /training_data/train_EMNIST_M: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/training_data/train_EMNIST_M -------------------------------------------------------------------------------- /training_data/train_Fashion: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/training_data/train_Fashion -------------------------------------------------------------------------------- /training_data/train_MNIST: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/training_data/train_MNIST -------------------------------------------------------------------------------- /training_data/train_Shapes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/training_data/train_Shapes -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clementchadebec/Data_Augmentation_with_VAE-DALI/HEAD/utils.py --------------------------------------------------------------------------------