Данный бот создан для переноса стиля с одного изображения на другое
В бота встроены две функции:
- Прислать изображение стиля и изображение контента
- Есди нет идей, на какое изображение нужно переносить стиль, то бот может скачать и предложить случайное фото из интернета, пользователю останется прислать только фото с необычным стилем
Кроме полученного изображения на выходе, бот также пришлёт коротенькую гифку - то, как на стиль из одного изображения переносился на другое
В этом проекте был использован алгоритм переноса стиля, разработанный Леоном Гатисом и Александром Шисмендером. Для обработки изображений в процессе переноса стиля была использована нейронная сеть VGG-19. Для оптимизации процесса использовался квазиньютоновский метод с ограниченной памятью BFGS. Функция потерь состоит из двух компонентов: контент-потери и стилевой потери. Контент-потеря измеряет среднеквадратичную ошибку (MSE) между изображением контента и изображением стиля. Стилевая потеря вычисляется путем вычисления MSE между грамматрицами этих изображений (для этого они сначала преобразуются в векторы). Соответствующие слои потерь добавляются после сверточных слоев внутри сети VGG-19, и сама сеть VGG-19 усечена, чтобы последний слой потери был последним слоем в сети.
- Изображение стиля
- Изображение контента
- Изображение на выходе
- Гифка