[go: up one dir, main page]

Skip to content

HunchPounch/style-transfer-bot

Repository files navigation

Style-Transfer Bot

Style-Transfer телеграмм бот создает новые изображения с помощью глубоких нейронных сетей, используя как классический алгоритм Гатиса на базе предобученной VGG19, так и CycleGAN. Бот написан с помощью актуальной библиотеки aiogram 3.x для построения асинхронной архитектуры.

Основной код находится в handlers/bot_messages.py. Бот упакован в Docker контейнер и задеплоен на арендованный сервер. Логи собираются логгером в stdout. Бот оснащен антифлудом.

Функционал бота:

Перенос стиля с помощью алгоритма Гатиса:

  • Необходимо отправить боту сначала фотографию, стиль которой хотим изменить, а затем фотографию, с которой хотим пересонить стиль.
  • Принцип прост: определяется два расстояния, до исходной картинки и до стилевой, и третье изображение преобразуется по градиентному спуску, минимизируя вышеупомянутые расстояния

Ray Image

Перенос стиля с помощью CycleGAN:

  • Необходимо отправить боту изображение и получить в ответ "картину Ван Гога"
  • Здесь принцип работы сложней, используется два генератора и два дискриминатора, работающие над переводами из одного домена в другой, а так же cycle consistency losses для сохранения структуры изображений. Благодаря данной реализации модели можно работать с непарными данными. Подробнее в статье

Ray Image

Установка

Docker + Docker Compose

  1. Возьмите файл docker-compose.yml из репозитория;
  2. Возьмите файл env_example там же, переименуйте как .env, откройте и заполните переменные;
  3. Запустите бота: docker compose up -d (или docker-compose up -d на старых версиях Docker);
  4. Проверьте, что контейнер поднялся: docker compose ps

Обычный запуск

  1. Клонируйте репозиторий;
  2. Скопируйте env_example под именем .env, откройте его и заполните переменные;
  3. Установите соответвующие модули из файла requirements.txt;
  4. Переместите файл main.py из директории bot в корень проекта;
  5. Запустите бота командой python __main__.py;