moymix / 3d-vq-vae-2

3D VQ-VAE-2 for high-resolution CT scan synthesis

Install

conda env create -f environment.yml
pip install .

Checkpoints / Codes / Samples

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

Due Credit

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