Hao SUN
7d73cf04f0
migrate to new scipy
|
há 4 anos atrás | |
|---|---|---|
| datasets | há 6 anos atrás | |
| imgs | há 6 anos atrás | |
| .gitattributes | há 6 anos atrás | |
| .gitignore | há 6 anos atrás | |
| README.md | há 4 anos atrás | |
| dice_score.m | há 6 anos atrás | |
| dice_score_multiple.m | há 6 anos atrás | |
| main.py | há 4 anos atrás | |
| model.py | há 4 anos atrás | |
| module.py | há 6 anos atrás | |
| ops.py | há 6 anos atrás | |
| prepareimgs.m | há 6 anos atrás | |
| requirements.txt | há 6 anos atrás | |
| utils.py | há 4 anos atrás |
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
- CycleGAN implementation, https://github.com/XHUJOY/CycleGAN-tensorflow
- Torch CycleGAN, https://github.com/junyanz/CycleGAN