karan hora
9d0cb59da7
initial commit
|
5 年 前 | |
|---|---|---|
| imgs | 5 年 前 | |
| .gitattributes | 5 年 前 | |
| .gitignore | 5 年 前 | |
| README.md | 5 年 前 | |
| dice_score.m | 5 年 前 | |
| dice_score_multiple.m | 5 年 前 | |
| download_dataset.sh | 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. For example in paper:
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