/ Модель

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

Параметры
113 млн
Лицензия
CC0 1.0
Архитектура
vit · dense
Опубликована
30 сен 2022

vit-base-patch16-224-hentai

Классификация изображенийИзображение → label3 профессии

Классификатор на базе ViT для маркировки аниме-изображений как безопасных, suggestive или explicit.

43скачиваний за 30 дней22лайкавеса обновлены4 июл 2023

Бенчмарки

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

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

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

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

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

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

Модель carbon225/vit-base-patch16-224-hentai — классификатор изображений на базе Google ViT-Base с патчами 16×16, дообученный для анализа аниме-изображений. Модель относит изображение к одному из трёх классов: safe, suggestive или explicit; она предназначена для маркировки контента сексуального характера в изображениях в стиле аниме, в том числе визуальных новелл.

Обучение проводилось на полноразмерных скриншотах из базы vndb.org; набор данных доступен как carbon225/vndb_img. Модель содержит 113,2 млн параметров и распространяется в форматах PyTorch и safetensors. Лицензия — CC0 1.0. Карточка также указывает возможность дообучения для других задач, связанных с аниме-изображениями.

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

  • фильтрация изображений с сексуальным содержанием в аниме-стиле
  • разметка скриншотов визуальных новелл по уровню откровенности
  • предварительная сортировка аниме-изображений для модерации
  • подготовка меток safe, suggestive и explicit для набора изображений

Ограничения

Модель обучена на скриншотах из базы vndb.org и рассчитана на аниме-изображения; качество на других типах изображений не подтверждено. Классы ограничены тремя категориями: safe, suggestive и explicit.

Для каких профессий подходит3

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

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

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

Открыть на Hugging Face

ViT for NSFW classification

Model info

This is Google's vit-base-patch16-224-in21k finetuned for flagging images according to vndb.org with 3 classes:

  • safe
  • suggestive
  • explicit

Training data

The model was trained on the vndb.org database dump using full size screenshots (sf in the database dump). The dataset can be loaded from carbon225/vndb_img.

Intended use

The model can be used for flagging anime-style images for sexual content. It can also be finetuned on other tasks related to anime images.

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

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

GPU в ailist

Универсальный рантайм · image-classification

Можно запустить

Подбираем GPU…

Класс архитектуры
ViTForImageClassification
Ревизия весов
d072322

0 моделей