Дообученный 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
Разделы каталога профессий, в задачах которых используется эта модель
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 Loss
Epoch
Step
Validation Loss
Accuracy
0.1218
1.0
147
0.0251
0.9953
0.0186
1.99
294
0.0239
0.9953
Framework versions
Transformers 4.38.2
Pytorch 2.1.2
Datasets 2.18.0
Tokenizers 0.15.2
Как запустить
На своей GPU в ailist. Цена — по текущим предложениям провайдеров