https://github.com/takuseno/beta-vae

beta-VAE implementation with TensorFlow
https://github.com/takuseno/beta-vae

Last synced: about 1 month ago
JSON representation

beta-VAE implementation with TensorFlow

• Host: GitHub
• URL: https://github.com/takuseno/beta-vae
• Owner: takuseno
• Created: 2018-09-09T10:08:42.000Z (over 6 years ago)
• Default Branch: master
• Last Pushed: 2018-10-10T13:58:07.000Z (about 6 years ago)
• Last Synced: 2024-10-23T04:07:00.849Z (2 months ago)
• Language: Python
• Size: 25.4 KB
• Stars: 2
• Watchers: 3
• Forks: 2
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# beta-vae

Implementation of beta-VAE with TensorFlow.

https://arxiv.org/pdf/1606.05579.pdf

## dependencies

- Python3

- tensorflow

- opencv-python

## train

```

$ python train.py

```

## test trained models

```

$ python test.py --model saved_models/path/models.ckpt --config saved_models/path/constants.json

```

## test each element of the latent variable

```

$ python test_latent.py --model saved_models/path/models.ckpt --config saved_models/path/constants.json

```

### beta = 1.0

![image](https://user-images.githubusercontent.com/5235131/45580499-acfe8280-b8cc-11e8-94bf-be33f11e475d.png)

### beta = 4.0

![image](https://user-images.githubusercontent.com/5235131/45580489-84768880-b8cc-11e8-81d0-af7142e7ff2f.png)