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: