Подготовка данных

1. Конфигурация маршрутов наборов данных

Конфигурация будет предполагать, что ваши данные находятся в $PaddleGAN/data. Вы можете создать символическую ссылку на свои наборы данных в `$PaddleGAN/data$.

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

Если вы поместите наборы данных в другое место, например your/data/path, вы также можете изменить dataroot в файле конфигурации:

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

2. Подготовка наборов данных

2.1 Загрузка наборов данных

2.1.1 Наборы данных CycleGAN

• Загрузка с веб-сайта

Наборы данных CycleGAN можно загрузить с здесь, не забудьте создать символическую ссылку на ваши наборы данных в `$PaddleGAN/data$.

• Загрузка через скрипт

Вы можете использовать download_cyclegan_data.py в PaddleGAN/data для загрузки необходимых вам наборов данных.

Поддерживаемые наборы данных: apple2orange, summer2winter_yosemite,horse2zebra, monet2photo, cezanne2photo, ukiyoe2photo, vangogh2photo, maps, cityscapes, facades, iphone2dslr_flower, ae_photos, cityscapes.

Выполните следующую команду. Набор данных будет загружен в ~/.cache/ppgan и символически связан с PaddleGAN/data/.

python data/download_cyclegan_data.py --name horse2zebra

Пользовательский набор данных

Данные должны быть организованы следующим образом, если вы используете пользовательский набор данных.

custom_datasets
├── testA
├── testB
├── trainA
└── trainB

2.1.2 Наборы данных Pix2Pix

• Загрузка с веб-сайта

Набор данных для pix2pix можно загрузить с здесь

• Загрузка через скрипт

Вы можете использовать download_pix2pix_data.py в PaddleGAN/data для загрузки нужных вам наборов данных. Поддерживаемые наборы данных: apple2orange, summer2winter_yosemite,horse2zebra, monet2photo, cezanne2photo, ukiyoe2photo, vangogh2photo, maps, cityscapes, facades, iphone2dslr_flower, ae_photos, cityscapes.

Набор данных будет загружен в ~/.cache/ppgan и символически связан с PaddleGAN/data/.

python data/download_pix2pix_data.py --name cityscapes

Пользовательские наборы данных

Данные должны быть организованы следующим образом, если вы используете пользовательский набор данных. И содержание изображения должно совпадать с примером изображения.

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