Style transfer telegram bot

Этот бот для телеграма переносит стиль с одних фотографий на другие.

Режимы работы бота

У бота есть 5 режимов работы:

• Простой перенос стиля с одного изображения на другое (NST)
• Двойной перенос стиля с двух изображений на третье (NST)
• Перекрашивание лошадей в зебр (GAN)
• Стилизация изображений под картины Ван Гога (GAN)
• Стилизация изображений под картины Моне (GAN)

Простой перенос стиля с одного изображения на другое

В данном режиме бот переносит стиль с первого изображения на второй с заданными настройками.

Данный режим использует технологию Neural style transfer.

Для этого режима возможны 2 дополнительные настройки:

• Количество эпох (влияет на степень переноса изображения)
  • 25 эпох
  • 50 эпох
  • 100 эпох
  • 200 эпох
  • 300 эпох
  • 400 эпох
• Размер выходного изображения
  • 64х64 пикселя
  • 128х128 пикселей
  • 256х256 пикселей
  • 512х512 пикселей

Возможные результаты работы данного режима бота:

Изначальное изображение	Переносимый стиль	Итоговое изображение

Двойной перенос стиля с двух изображений на третье

В данном режиме бот переносит первый стиль на левую половину фото, второй стиль -- на правую.

Данный режим использует технологию Neural style transfer.

Для этого режима возможны 2 дополнительные настройки:

• Количество эпох (влияет на степень переноса изображения)
  • 25 эпох
  • 50 эпох
  • 100 эпох
  • 200 эпох
  • 300 эпох
  • 400 эпох
• Размер выходного изображения
  • 64х64 пикселя
  • 128х128 пикселей
  • 256х256 пикселей
  • 512х512 пикселей

Возможные результаты работы данного режима бота:

Изначальное изображение	Первый переносимый стиль	Второй переносимый стиль	Итоговое изображение

Перекрашивание лошадей в зебр

В данном режиме бот раскрашивает лошадей на фото в полоски так, чтобы они были похожи на зебр. При этом бот использует предобученную генеративную сеть из этого проекта: https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix.

Данный режим использует технологию Generative adversarial networks.

Для этого режима возможна 1 дополнительная настройка:

• Размер выходного изображения
  • 64х64 пикселя
  • 128х128 пикселей
  • 256х256 пикселей
  • 512х512 пикселей

Возможные результаты работы данного режима бота:

Изначальное изображение	Итоговое изображение

Стилизация изображений под картины Ван Гога

В данном режиме бот перерисовывает данное ему изображение так, чтобы оно было похоже по стилю на картины Ван Гога. При этом бот использует предобученную генеративную сеть из этого проекта: https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix.

Данный режим использует технологию Generative adversarial networks.

Для этого режима возможна 1 дополнительная настройка:

• Размер выходного изображения
  • 64х64 пикселя
  • 128х128 пикселей
  • 256х256 пикселей
  • 512х512 пикселей

Возможные результаты работы данного режима бота:

Изначальное изображение	Итоговое изображение

Стилизация изображений под картины Моне

В данном режиме бот перерисовывает данное ему изображение так, чтобы оно было похоже по стилю на картины Моне. При этом бот использует предобученную генеративную сеть из этого проекта: https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix.

Данный режим использует технологию Generative adversarial networks.

Для этого режима возможна 1 дополнительная настройка:

• Размер выходного изображения
  • 64х64 пикселя
  • 128х128 пикселей
  • 256х256 пикселей
  • 512х512 пикселей

Возможные результаты работы данного режима бота:

Изначальное изображение	Итоговое изображение

Информация по запуску бота

Для запуска данного бота у себя необходимо добавить в основной каталог файл config.py со следующим содержанием:

  API_TOKEN = '<YOUR TOKEN>'

  DEBUG_ID = '<YOUR_DEBUG_ID>'
  
  GET_DEBUG_INFO = True

  LOGGING = True

  MODE = 'NORMAL'

  CONNECTION_TYPE = 'POLLING'

  WEBHOOK_HOST = '<YOUR_WEBHOOK_HOST>'

  WEBAPP_PORT = '<YOUR_WEBAPP_PORT>'

Где:

• <YOUR TOKEN> -- токен вашего бота, который можно получить у официального бота сервиса Telegram для создания собственных ботов: @BotFather,
• <YOUR_DEBUG_ID> -- id человека, которому будут приходить сообщения в телеграм об ошибках,
• GET_DEBUG_INFO -- будет ли на указанный выше id отправляться информация об ошибках (если поставить False, то можно не указывать id в строке выше),
• LOGGING -- будет ли выводиться в консоль информация о том, какие действия сейчас совершает бот,
• MODE -- может принимать значения 'EASY' или 'NORMAL', в первом случае при попытке перенесения стиля бот будет сообщать, что он запущен на слабом устройстве, во втором случае все работает штатно.
• CONNECTION_TYPE -- может принимать значения 'POLLING' или 'WEBHOOKS', в зависимости от желаемого вами типа (для более простой работы стоит указать 'POLLING')
• <YOUR_WEBHOOK_HOST> -- адрес вашего webhook хоста (если выше выбрали 'POLLING', то можно не заполнять)
• <YOUR_WEBAPP_POST> -- порт вашего webhook хоста (если выше выбрали 'POLLING', то можно не заполнять)

Библиотека для работы с нейронными сетями: pytorch

---

По всем вопросам обращаться в телеграм: @alresing