Hao SUN
7d73cf04f0
migrate to new scipy
|
4 лет назад | |
|---|---|---|
| datasets | 6 лет назад | |
| imgs | 6 лет назад | |
| .gitattributes | 6 лет назад | |
| .gitignore | 6 лет назад | |
| README.md | 4 лет назад | |
| dice_score.m | 6 лет назад | |
| dice_score_multiple.m | 6 лет назад | |
| main.py | 4 лет назад | |
| model.py | 4 лет назад | |
| module.py | 6 лет назад | |
| ops.py | 6 лет назад | |
| prepareimgs.m | 6 лет назад | |
| requirements.txt | 6 лет назад | |
| utils.py | 4 лет назад |
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