https://github.com/4catalyzer/cyclegan
https://github.com/4catalyzer/cyclegan
Last synced: about 1 month ago
JSON representation
- Host: GitHub
- URL: https://github.com/4catalyzer/cyclegan
- Owner: 4Catalyzer
- License: mit
- Created: 2017-07-18T15:48:28.000Z (almost 8 years ago)
- Default Branch: master
- Last Pushed: 2019-03-29T03:13:33.000Z (about 6 years ago)
- Last Synced: 2025-03-26T18:03:23.870Z (about 2 months ago)
- Language: Python
- Size: 991 KB
- Stars: 13
- Watchers: 4
- Forks: 123
- Open Issues: 4
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# CycleGAN in TensorFlow
This is the TensorFlow implementation for CycleGAN. The code was written by [Harry Yang](https://www.harryyang.org) and [Nathan Silberman](https://github.com/nathansilberman).
CycleGAN: [[Project]](https://junyanz.github.io/CycleGAN/) [[Paper]](https://arxiv.org/pdf/1703.10593.pdf)
## Introduction
This code contains two versions of the network architectures and hyper-parameters. The first one is based on the [TensorFlow implementation](https://github.com/hardikbansal/CycleGAN). The second one is based on the [official PyTorch implementation](https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix). The differences are minor and we observed both versions produced good results. You may need to train several times as the quality of the results are sensitive to the initialization.
Below is a snapshot of our result at the 50th epoch on one training instance:
## Getting Started
### Prepare dataset
* You can either download one of the defaults CycleGAN datasets or use your own dataset.
* Download a CycleGAN dataset (e.g. horse2zebra):
```bash
bash ./download_cyclegan_dataset.sh horse2zebra
```
* Using your own dataset: put images from each domain at folder_a and folder_b respectively.
* Create the csv file as input to the data loader.
* Edit the cyclegan_datasets.py file. For example, if you have a face2ramen_train dataset which contains 800 face images and 1000 ramen images both in PNG format, you can just edit the cyclegan_datasets.py as following:
```python
DATASET_TO_SIZES = {
'face2ramen_train': 1000
}
PATH_TO_CSV = {
'face2ramen_train': './CycleGAN/input/face2ramen/face2ramen_train.csv'
}
DATASET_TO_IMAGETYPE = {
'face2ramen_train': '.png'
}
```
* Run create_cyclegan_dataset.py:
```bash
python -m CycleGAN_TensorFlow.create_cyclegan_dataset --image_path_a=folder_a --image_path_b=folder_b --dataset_name="horse2zebra_train" --do_shuffle=0
```
### Training
* Create the configuration file. The configuration file contains basic information for training/testing. An example of the configuration file could be fond at configs/exp_01.json.
* Start training:
```bash
python -m CycleGAN_TensorFlow.main \
--to_train=1 \
--log_dir=CycleGAN_TensorFlow/output/cyclegan/exp_01 \
--config_filename=CycleGAN_TensorFlow/configs/exp_01.json
```
* Check the intermediate results.
* Tensorboard
```bash
tensorboard --port=6006 --logdir=CycleGAN_TensorFlow/output/cyclegan/exp_01/#timestamp#
```
* Check the html visualization at CycleGAN_TensorFlow/output/cyclegan/exp_01/#timestamp#/epoch_#id#.html.
### Restoring from the previous checkpoint.
```bash
python -m CycleGAN_TensorFlow.main \
--to_train=2 \
--log_dir=CycleGAN_TensorFlow/output/cyclegan/exp_01 \
--config_filename=CycleGAN_TensorFlow/configs/exp_01.json \
--checkpoint_dir=CycleGAN_TensorFlow/output/cyclegan/exp_01/#timestamp#
```
### Testing
* Create the testing dataset.
* Edit the cyclegan_datasets.py file the same way as training.
* Create the csv file as the input to the data loader.
```bash
python -m CycleGAN_TensorFlow.create_cyclegan_dataset --image_path_a=folder_a --image_path_b=folder_b --dataset_name="horse2zebra_test" --do_shuffle=0
```
* Run testing.
```bash
python -m CycleGAN_TensorFlow.main \
--to_train=0 \
--log_dir=CycleGAN_TensorFlow/output/cyclegan/exp_01 \
--config_filename=CycleGAN_TensorFlow/configs/exp_01_test.json \
--checkpoint_dir=CycleGAN_TensorFlow/output/cyclegan/exp_01/#old_timestamp#
```
The result is saved in CycleGAN_TensorFlow/output/cyclegan/exp_01/#new_timestamp#.