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 | ![alt text](jpg/dtn.jpg) | 3 | ![alt text](jpg/dtn.jpg) |
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