conda env create -f environment.yml
pip install .
All checkpoints, codes, and samples are available at https://surfdrive.surf.nl/files/index.php/s/xY7bwjrgfnhPCAt.
Simply copy+pasting the folder structure over the root of this reposity is sufficient.
For completeness, the relative path (from root) of each checkpoint/code/sample is listed in the following table:
| Checkpoints | Full size (512×512×128) | Downscaled (256×256×128) |
|---|---|---|
| AE | 3-layer | 2-layer |
| Codes | 3-layer-codes | 2-layer-codes |
| Pixel Model | Top Mid Bottom |
Top Bottom |
| Samples | Unconditional | Conditional Unconditional |
This implementation alters and re-uses ideas from a few places:
- VQ-VAE & Pixel-Model: https://github.com/rosinality/vq-vae-2-pytorch
- VQ-VAE: https://github.com/danieltudosiu/nmpevqvae
- Fixup: https://github.com/hongyi-zhang/Fixup