Karan Hora
d52e69ec62
update 2
|
5 лет назад | |
|---|---|---|
| imgs | 5 лет назад | |
| .gitattributes | 5 лет назад | |
| .gitignore | 5 лет назад | |
| README.md | 5 лет назад | |
| dice_score.m | 5 лет назад | |
| dice_score_multiple.m | 5 лет назад | |
| main.py | 5 лет назад | |
| model.py | 5 лет назад | |
| module.py | 5 лет назад | |
| ops.py | 5 лет назад | |
| prepareimgs.m | 5 лет назад | |
| requirements.txt | 5 лет назад | |
| utils.py | 5 лет назад |
README.md
CycleGAN
As proposed by Jun-Yan Zhu in Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networkssee.
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