https://github.com/revsic/tf-survae-flows

Tensorflow implementation of SurVAE Flows, Nielsen et al., 2020.
https://github.com/revsic/tf-survae-flows

diffusion normalizing-flow survae-flows tensorflow tf2 vae

Last synced: about 1 month ago
JSON representation

Tensorflow implementation of SurVAE Flows, Nielsen et al., 2020.

• Host: GitHub
• URL: https://github.com/revsic/tf-survae-flows
• Owner: revsic
• License: mit
• Created: 2021-01-04T15:30:46.000Z (over 5 years ago)
• Default Branch: main
• Last Pushed: 2021-05-05T14:44:24.000Z (about 5 years ago)
• Last Synced: 2025-03-23T06:27:51.039Z (about 1 year ago)
• Topics: diffusion, normalizing-flow, survae-flows, tensorflow, tf2, vae
• Language: Jupyter Notebook
• Homepage:
• Size: 806 KB
• Stars: 2
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# tf-survae-flows

(Unofficial) Tensorflow implementation of SurVAE Flows, Nielsen et al., 2020. 

- Original paper, SurVAE Flows [[arXiv:2007.02731](https://arxiv.org/abs/2007.02731)]

- Full code is based on original repository [[GIT](https://github.com/didriknielsen/survae_flows)]

## Usage

For checking exeperiments, reference experiment page.

- [experiments/checkerboard.ipynb](./experiments/checkerboard.ipynb), checkerboard toy samples for AbsFlow.

- [experiments/vae.ipynb](./experiments/vae.ipynb), MNIST samples for VAE.

Checkerboard baseline

![checkerboard baseline](rsrc/checkerboard_baseline.png)

Checkerboard AbsFlow

![checkerboard abseflow](rsrc/checkerboard_absflow.png)

MNIST-VAE on survae-flow framework

![vae latent interpolation](rsrc/vae_latentinterp.png)