Характеристики

Лицензия: Apache 2.0
Опубликована: 10 авг 2022

Ссылки

Hugging Face

fbanime-gan

skytnt

Генерация изображенийТекст → изображение

Модель на основе StyleGAN2 для безусловной генерации аниме-изображений, обученная на fbanimehq v2.0.

0скачиваний за 30 дней9лайковвеса обновлены12 дек 2022

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

fbanime-gan — модель генерации изображений из проекта fbanimegan, основанная на StyleGAN2 и обученная на датасете fbanimehq v2.0. Автор указывает, что код StyleGAN2 изменён для поддержки неквадратных разрешений; для модели заявлен FID 1.4.

В репозитории представлены веса в fp16 и fp32, а также ONNX-версии mapping- и synthesis-сетей. README также упоминает encoder на основе e4e и ResNet-34 и модель обнаружения waifudect на YOLOv5. Основной сценарий — безусловная генерация аниме-изображений; fp16-вариант работает только на GPU, тогда как fp32-вариант поддерживает GPU и CPU.

Лицензия репозитория — Apache-2.0. В карточке указана задача unconditional-image-generation; сведения о размере модели и длине контекста неприменимы или не приведены.

Сценарии применения

генерация аниме-изображений без текстового запроса
создание неквадратных аниме-иллюстраций
получение вариаций изображений в стиле обучающего набора

Ограничения

Модель генерирует изображения без текстового управления содержанием. fp16-вариант работает только на GPU; README не приводит оценки качества вне заявленного FID 1.4.

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

Model info

Project fbanimegan

fbanime.pkl

StyleGan2 model trained with official StyleGan3. But I modified the code (networks_stylegan2.py and dataset.py) to support non-square resolutions.

FID: 1.4

fbanime_fp32.pkl

fp32 version of fbanime.pkl

Note: Fp16 version (fbanime.pkl) only works on gpu. And fp32 version works on gpu and cpu.

g_mapping.onnx

onnx format mapping network of fbanime_fp32.pkl

g_synthesis.onnx

onnx format synthesis network of fbanime_fp32.pkl

encoder.onnx

e4e model trained with encoder4editing-stylegan3. I add support for official StyleGan2 model and change backbone to ResNet-34 in restyle-encoder.

waifu_dect.onnx

YOLOv5 model trained with official YOLOv5

Usage

see demo

Dataset

fbanimehq v2.0

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм

Нельзя запустить

Безусловная генерация изображений пока не поддерживается.

Ревизия весов: 79c6af6