/ Модель

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

Параметры
113 млн
Лицензия
Apache 2.0
Архитектура
vit · dense
Базовая модель
vit-base-patch16-224-in21k
Опубликована
29 мая 2024

Facemask-detection

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

Дообученный Vision Transformer для классификации изображений с целью обнаружения масок на лицах.

134скачиваний за 30 дней0лайковвеса обновлены29 мая 2024

Бенчмарки

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

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

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

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

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

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

Hemgg/Facemask-detection — дообученная модель Vision Transformer для классификации изображений с целью обнаружения масок на лицах. Она основана на google/vit-base-patch16-224-in21k и содержит 113,2 млн параметров. Модель принимает изображения и выдаёт классы; в карточке приведены результаты оценки: accuracy 0,9953 и loss 0,0239.

Модель подойдёт для классификации изображений лиц по наличию маски, например, при анализе фотографий или видеокадров после извлечения отдельных изображений. Данные об используемом датасете и перечне классов не приведены. Распространяется по лицензии Apache-2.0.

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

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

Ограничения

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

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

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

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

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

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

Открыть на Hugging Face

Face-Mask-Detection

This model is a fine-tuned version of google/vit-base-patch16-224-in21k on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0239
  • Accuracy: 0.9953

Training results

Training LossEpochStepValidation LossAccuracy
0.12181.01470.02510.9953
0.01861.992940.02390.9953

Framework versions

  • Transformers 4.38.2
  • Pytorch 2.1.2
  • Datasets 2.18.0
  • Tokenizers 0.15.2

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей