Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

alresing/Style_transfer_telegram_bot

Open more actions menu

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
27 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

A bot for Telegram that can transfer style from one image to another

Topics

Resources

Stars

Watchers

Forks

Used by

Contributors

Languages

Morty Proxy This is a proxified and sanitized view of the page, visit original site.