Image-to-Image при помощи cGAN

Данный репозиторий содержит реализацию Conditional Adversarial Networks для решения задачи "картинка по картинке" на наборе данных facades. Модель реализована на основе U-net генератора и 70x70 PatchGAN детектора (см. орининальную статью).

[Статья]

Пример сгенерированных изображений:

Запуск проекта

Для начала скачаем проект:

git clone https://github.com/Haru4me/pix2pix.git

Для запуска необходимо скачать оригинальный датасэт по [ссылке] и поместить его в корневую папку проекта. Папка facades должна содержать в себе 4 папки (trainA,trainB,testA,testB).

Теперь запустите через командную строку:

cd /pix2pix
sh run.sh train

После того как модель обучится, можно будет запустить ее так:

cd /pix2pix
sh run.sh test