sara-nl / 3D-VQ-VAE-2

3D VQ-VAE-2 for high-resolution CT scan synthesis
https://scripties.uba.uva.nl/search?id=722710
44 stars 8 forks source link
3d computed-tomography generative-model pixelcnn pixelsnail pytorch pytorch-lightning vq-vae-2

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: