Official code for the VicTr-3D

First set up the environment:

conda create -n victr3d python==3.11

conda activate victr3d

pip3 install -r requirements.txt

Download weights

chmod +x download.sh

./download.sh

Inference for class condition generation

python3 evaluation/class_conditional_generation.py \
--AE-ckpt checkpoints/3DVAE.ckpt \
--model-ckpt checkpoints/VicTr3D_MRI_CT_Pretrained.pt \
--output-dir save_gen