data_prepare.md 3.0 KB
Newer Older
L
LielinJiang 已提交
1
## 数据准备
L
LielinJiang 已提交
2

L
LielinJiang 已提交
3
现有的配置默认数据集的路径是在`$PaddleGAN/data`下,目录结构如下图所示。如果你已经下载好数据集了,建议将数据集软链接到 `$PaddleGAN/data`
L
LielinJiang 已提交
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30

```
PaddleGAN
|-- configs
|-- data
|   |-- cityscapes
|   |   ├── test
|   |   ├── testA
|   |   ├── testB
|   |   ├── train
|   |   ├── trainA
|   |   └── trainB
|   ├── horse2zebra
|   |   ├── testA
|   |   ├── testB
|   |   ├── trainA
|   |   └── trainB
|   └── facades
|       ├── test
|       ├── train
|       └── val
|-- docs
|-- ppgan
|-- tools

```

L
LielinJiang 已提交
31 32 33 34 35 36 37 38 39 40 41 42 43
如果将数据集放在其他位置,比如 ```your/data/path```
你可以修改配置文件中的 ```dataroot``` 参数:

```
dataset:
  train:
    name: PairedDataset
    dataroot: your/data/path
    num_workers: 4
```

### CycleGAN模型相关的数据集下载

Q
qingqing01 已提交
44
#### 从网页下载
L
LielinJiang 已提交
45

L
LielinJiang 已提交
46
cyclgan模型相关的数据集可以在[这里](https://people.eecs.berkeley.edu/~taesung_park/CycleGAN/datasets/)下载
L
LielinJiang 已提交
47

Q
qingqing01 已提交
48 49
#### 使用脚本下载

L
LielinJiang 已提交
50 51 52 53
我们在 ```PaddleGAN/data``` 文件夹下提供了一个脚本 ```download_cyclegan_data.py``` 方便下载CycleGAN相关的
数据集。执行如下命令可以下载相关的数据集,目前支持的数据集名称有:apple2orange, summer2winter_yosemite,horse2zebra, monet2photo, cezanne2photo, ukiyoe2photo, vangogh2photo, maps, cityscapes, facades, iphone2dslr_flower, ae_photos, cityscapes。

执行如下命令,可以下载对应的数据集到 ```~/.cache/ppgan``` 并软连接到 ```PaddleGAN/data/``` 下。
Q
qingqing01 已提交
54

L
LielinJiang 已提交
55 56 57
```
python data/download_cyclegan_data.py --name horse2zebra
```
Q
qingqing01 已提交
58

L
LielinJiang 已提交
59 60 61 62 63 64 65 66 67 68 69 70
#### 使用自己的数据集
如果你使用自己的数据集,需要构造成如下目录的格式。注意 ```xxxA``````xxxB```文件数量,文件内容无需一一对应。
```
custom_datasets
├── testA
├── testB
├── trainA
└── trainB
```

### Pix2Pix相关的数据集下载

Q
qingqing01 已提交
71
#### 从网页下载
L
LielinJiang 已提交
72

L
LielinJiang 已提交
73
pixel2pixel模型相关的数据集可以在[这里](https://people.eecs.berkeley.edu/~tinghuiz/projects/pix2pix/datasets/)下载
L
LielinJiang 已提交
74

Q
qingqing01 已提交
75 76
#### 使用脚本下载

L
LielinJiang 已提交
77 78 79
我们在 ```PaddleGAN/data``` 文件夹下提供了一个脚本 ```download_pix2pix_data.py``` 方便下载pix2pix模型相关的数据集。执行如下命令可以下载相关的数据集,目前支持的数据集名称有:apple2orange, summer2winter_yosemite,horse2zebra, monet2photo, cezanne2photo, ukiyoe2photo, vangogh2photo, maps, cityscapes, facades, iphone2dslr_flower, ae_photos, cityscapes。

执行如下命令,可以下载对应的数据集到 ```~/.cache/ppgan``` 并软连接到 ```PaddleGAN/data/``` 下。
Q
qingqing01 已提交
80

L
LielinJiang 已提交
81 82 83 84 85
```
python data/download_pix2pix_data.py --name cityscapes
```

#### 使用自己的数据集
Q
qingqing01 已提交
86

L
LielinJiang 已提交
87 88 89 90 91 92 93 94
如果你使用自己的数据集,需要构造成如下目录的格式。同时图片应该制作成下图的样式,即左边为一种风格,另一边为相应转换的风格。

```
facades
├── test
├── train
└── val
```