/cycle-consistent-VAE

Primary LanguagePython

Cycle Consistent VAE

This is the code for Cycle Consistent VAE (ECCV).

Official Code Repo - Github

Architecture

Forward Cycle

Reverse Cycle

Installation

Use the package manager pip to install all required libraries.

pip install requirements.txt

or

conda create --name <env> --file requirements.txt

Training

All the models are train on 2D Sprites Dataset.

Autoencoder

For training the autoencoer use main_vae.py

Classifier Model

For training classifier based specified and unspecified use main_classifier.py

Prediction Model

For training prediction based specified and unspecified use main_prediction.py

Results

Autoencoder

image info

Style transfer output

Linear Interpolation

Classifier Model Training

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.