PhotoSketch

Фотография: вывод контурных рисунков из изображений

Этот репо содержит код обучения и тестирования для нашего генератора эскизов. Мы также предоставляем [предварительно обученную модель].

Для получения технических данных и набора данных, пожалуйста, обратитесь к [Paper] и [Project Page] .

Настройка

Теперь код обновлен для использования Pytorch 0.4 и работает в Windows, Mac и Linux. Для устаревшей версии с Pytorch 0,3 (только Linux), пожалуйста, ознакомьтесь с Branch Pytorch-0.3-Obsolete.

Пользователи Windows должны найти соответствующие файлы .cmd вместо файлов .sh упомянутых ниже.

Установка с одной линией (с условиями Conda)

conda env create -f environment.yml

Затем активируйте окружающую среду (эскиз), и вы готовы к работе!

Смотрите здесь для получения дополнительной информации о средах Conda.

Ручная установка

См. environment.yml для списка зависимостей.

Использование предварительно обученной модели

• Загрузите предварительно обученную модель
• Изменить путь в scripts/test_pretrained.sh
• Из корневого каталога репо, запустите scripts/test_pretrained.sh

Он поддерживает папку изображений в качестве входных.

Train & Test в нашем наборе данных о контурном рисунке

• Загрузите изображения и обозначенный набросок со страницы проекта
• Расписки и организуйте их в следующую структуру:

• Измените путь в scripts/train.sh и scripts/test.sh
• Из корневого каталога репо, запустите scripts/train.sh для обучения модели
• Из корневого каталога репо, запустите scripts/test.sh для проверки на наборе VAL или наборе тестирования (указанный флагом Phase)

Цитирование

Если вы используете код или данные для вашего исследования, пожалуйста, укажите статью:

 @article{LIPS2019,
  title={Photo-Sketching: Inferring Contour Drawings from Images},
  author={Li, Mengtian and Lin, Zhe and Mv ech, Radom'ir and and Yumer, Ersin and Ramanan, Deva},
  journal={WACV},
  year={2019}
}

Подтверждение

Этот код основан на старой версии Pix2pix.