Showing
1 changed file
with
39 additions
and
1 deletions
| 1 | # Domain Transfer Network (DTN) | 1 | # Domain Transfer Network (DTN) |
| 2 | TensorFlow implementation of [Unsupervised Cross-Domain Image Generation.](https://arxiv.org/abs/1611.02200) | 2 | TensorFlow implementation of [Unsupervised Cross-Domain Image Generation.](https://arxiv.org/abs/1611.02200) |
| 3 | - | ||
| 4 |  | 3 |  |
| 5 | 4 | ||
| 5 | +## Requirements | ||
| 6 | +* Python 2.7 | ||
| 7 | +* [TensorFlow 0.12](https://github.com/tensorflow/tensorflow/tree/r0.12) | ||
| 8 | + | ||
| 9 | + | ||
| 6 | ## Usage | 10 | ## Usage |
| 7 | 11 | ||
| 12 | +#### Clone the repository | ||
| 13 | +```bash | ||
| 14 | +git clone https://github.com/yunjey/dtn-tensorflow.git | ||
| 15 | +cd dtn-tensorflow | ||
| 16 | +``` | ||
| 17 | + | ||
| 18 | +#### Download the dataset | ||
| 19 | +```bash | ||
| 20 | +chmod +x download.sh | ||
| 21 | +./download.sh | ||
| 22 | +``` | ||
| 23 | + | ||
| 24 | +#### Resize MNIST dataset to 32x32 | ||
| 25 | +```bash | ||
| 26 | +python prepro.py | ||
| 27 | +``` | ||
| 28 | + | ||
| 29 | +#### Pretrain the model | ||
| 30 | +```bash | ||
| 31 | +python main.py --mode='pretrain' | ||
| 32 | +``` | ||
| 33 | + | ||
| 34 | +#### Train the model | ||
| 35 | +```bash | ||
| 36 | +python main.py --mode='train' | ||
| 37 | +``` | ||
| 38 | + | ||
| 39 | +#### Transfer SVHN to MNIST | ||
| 40 | +```bash | ||
| 41 | +python main.py --mode='eval' | ||
| 42 | +``` | ||
| 43 | + | ||
| 44 | +## Results | ||
| 45 | + | ||
| 8 | 46 | ... | ... |
-
Please register or login to post a comment