ITСooky

IT-рецепты съедобные и не очень!

Как делать аниме фотки с нейросетью AnimeGANv2!

дата 10.12.2022

Еще одна нейросеть, старая уже, прошлогодняя, но ею можно воспользоваться свободно. Качество мне кажется все ближе к Meitu Ai Art тут конечно все дело в тренеровки и врядли кто выложит такуюже натренерованную нейросеть, а по проще есть! Устаналиваю PyTorch Implementation of AnimeGANv2 это вот руторкх реализация AnimeGANv2.

Все делаю в Conda чтобы не было проблем с совместимыми версия, они все равно слетают, но проблем действительно меньше.

Клонирую гит
git clonе https://github.com/bryandlee/animegan2-pytorch
cd animegan2-pytorch

Сама конда
conda create -n agv2
conda activate agv2

Тут собственно если ставить то что хочет PyTorch Implementation of AnimeGANv2 то у мея не работает

The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
If you want to use the NVIDIA RTX A4500 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

Поставил все по умолчанию новейшее
pip3 install torch torchvision

Нам дают 4-ре натренерованных модели
celeba_distill
face_paint_512_v1
face_paint_512_v2
paprika

Запускать надо файл test.py и там надо прописывать в 62-ой строке какой эффект брать

        default='./weights/face_paint_512_v2.pt',

Создаю в папке со скриптами папку /input туда клад исходные фотки, а результат будет появлятся в /samples/result

Беру нашего тестового конана

И запускаю
python test.py --input_dir ./input --device cuda

Уппппппс

Да лицо надо покрупнее а фотку поменьше, потому что на огромных фотак будет ошибка нехватки памяти на видеокарте, я засовывал где-то 1024×1024 фотки.

Вырезаем лицо Конана и увеличиваем нейросетью GFPGAN (Увеличиваем улучшаем старые фотки с GFPGAN на Windows через WSL2 и картой с CUDA(можно и без)!)

Новый тестовый лицо-Конан

Пробуем эффект face_paint_512_v2

Ха, хаааа! Нейросеть умная но глупая, а зачем такие длинные волосы носить — поэтотму придает женсике черты!!!

Пробуем эффект face_paint_512_v1

Мммм стал еще более женственным, собственно уже Конаниха получается совсем!

Пробуем эффект paprika

По мне так это уже не умный эффект а просто выкручивание цветов, но может кто-то из знатаков оценить паприку!

Пробуем эффект celeba_distill

Ну это да фигня какая-то!

Ну вообщем эффект от эффекта есть, можно использовать, посмотрим что Друзья скажут

ААА! Лица надо больше а фотку меньше жееее!


Добавить комментарий

Ваш адрес email не будет опубликован. Обязательные поля помечены *