Keine Beschreibung

karan hora e4e6857a51 Merge branch 'master' of https://github.com/H2K804/CycleGAN-medical-image-segmentation vor 5 Jahren
imgs 9d0cb59da7 initial commit vor 5 Jahren
.gitattributes c0c20cf92b Initial commit vor 5 Jahren
.gitignore 9d0cb59da7 initial commit vor 5 Jahren
README.md da7a13c2b5 added new files vor 5 Jahren
dice_score.m 9d0cb59da7 initial commit vor 5 Jahren
dice_score_multiple.m 9d0cb59da7 initial commit vor 5 Jahren
main.py 9d0cb59da7 initial commit vor 5 Jahren
model.py 9d0cb59da7 initial commit vor 5 Jahren
module.py 9d0cb59da7 initial commit vor 5 Jahren
ops.py 9d0cb59da7 initial commit vor 5 Jahren
prepareimgs.m 9d0cb59da7 initial commit vor 5 Jahren
requirements.txt 9d0cb59da7 initial commit vor 5 Jahren
utils.py 9d0cb59da7 initial commit vor 5 Jahren

README.md

CycleGAN

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

Prerequisites

  • tensorflow r1.1
  • numpy 1.11.0
  • scipy 0.17.0
  • pillow 3.3.0

Train

python main.py --dataset_dir=med_image

Test

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

References