H2K804 / CycleGAN-medical-image-segmentation

Implementation of CycleGAN for unsupervised image segmentaion, performed on brain tumor scans
46 stars 13 forks source link

CycleGAN

As proposed by Jun-Yan Zhu in Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks.

Prerequisites

Train

python main.py --dataset_dir=med_image

Test

python main.py --dataset_dir=med-image --phase=test --which_direction=AtoB

References