Основная идея была реализовать следующий пайплайн:
- Генерация описаний content + style с помощью BLIP
- Удаления фона с картинок
- Style Transfer с помощью kadinsky на картинках с удаленным фоном
- Генерация фона для 3), опять же с помощью Kadinsky
Также результаты будут представлены обычной модели kadinsky, без использования пайплайна выше
pytorch >=1.6.0 builds with cuda support
pip3 install torch torchvision torchaudio
pip install -r requirements.txt
pip install transformers -Upython run.py images/boromir.jpeg images/vazovski.jpeg