CycleGAN used for multimodal MRI image translation
pytorch>=1.5, torchvision, PIL, SimpleITK
downloand data from here and unzip it outside the code folder
python main.py --mode train
optional (visualize training process in TensorBoard): tensorbooard --logdir logs/
python main.py --mode test
--data_root: data root stores all training dataset and test dataset (e.g. trainT1, trainT2, testT1, testT2...)
--model_root: pre-trained G_A2B (generator modal A -> modal B) model used for test
--epochs: number of epochs to train
--output_root: output root stores all test outputs
--ssim: use ssim as reconstruction loss