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

Параметры: 113 млн
Архитектура: vit · dense
Опубликована: 9 дек 2024

Ссылки

Hugging Face

violence_classification

locih

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

Дообученная ViT-модель классифицирует изображения как содержащие или не содержащие сцены насилия.

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

Бенчмарки

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

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

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

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

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

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

locih/violence_classification — дообученная модель Vision Transformer для классификации изображений по наличию насилия. Она предназначена для выявления сцен насилия и разделения изображений на категории «насильственные» и «ненасильственные». Модель содержит 113,2 млн параметров и представлена в формате Safetensors.

Обучение проводилось на Violence Detection Dataset с Kaggle, содержащем изображения насильственных и ненасильственных сцен. Карточка не приводит результаты оценки и подробности о применении. Лицензия в доступных данных не указана.

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

фильтрация изображений со сценами насилия
первичная модерация пользовательских изображений
классификация изображений на насильственные и ненасильственные

Ограничения

В карточке нет результатов оценки, описания известных ограничений или сведений о лицензии.

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

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

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

IV.9Отраслевые задачиМедиа, журналистика, культура, архивы и игрыAI помогает создавать, искать, сохранять и персонализировать медиа, культурный и игровой контент.44 задачи14 938моделей II.10Разработка и данныеРазработка и эксплуатация AI/ML-системAI помогает создавать, оценивать, развёртывать и контролировать AI/ML-системы на всём жизненном цикле.30 задач49 000моделей II.11Разработка и данныеКибербезопасность и приватностьAI помогает обнаруживать киберугрозы, защищать данные и помогать реагировать на инциденты.25 задач4 232модели

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

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

Открыть на Hugging Face

violence_detection

This repository contains a fine-tuned Vision Transformer (ViT) model for detecting violence in images. The model is trained on the Violence Detection Dataset from Kaggle, which consists of images categorized into violent and non-violent scenes.

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training results

Training Loss	Epoch	Validation Loss	Accuracy
0.167900	1	0.164447	0.931034
0.010100	2	0.170160	0.942529
0.002200	3	0.170347	0.948276
0.001400	4	0.170811	0.948276
0.001100	5	0.172756	0.948276

Framework versions

Transformers 4.42.4
Pytorch 2.3.1+cu121
Datasets 2.20.0
Tokenizers 0.19.1

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

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

GPU в ailist

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

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

Подбираем GPU…

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