/ Модель

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

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

finetuned-ViT-human-action-recognition-v1

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

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

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

Бенчмарки

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

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

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

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

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

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

DrishtiSharma выпустила модель ViT для классификации изображений, дообученную на датасете Human_Action_Recognition на базе google/vit-base-patch16-224-in21k. Модель содержит 113,2 млн параметров и предназначена для распознавания человеческих действий по изображениям.

В качестве оценочной метрики в карточке указана точность 0,0791 и функция потерь 3,1427. Эти результаты указывают на низкую точность на оценочном наборе, поэтому модель требует проверки перед практическим применением. Модель распространяется в формате PyTorch по лицензии Apache-2.0.

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

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

Ограничения

В README указана низкая точность на оценочном наборе — 0,0791; пригодность для реальных сценариев не подтверждена. Описание датасета, меток классов и ограничений применения не приведено.

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

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

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

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

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

Открыть на Hugging Face

finetuned-ViT-human-action-recognition-v1

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

  • Loss: 3.1427
  • Accuracy: 0.0791

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 0.0002
  • train_batch_size: 16
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 4
  • mixed_precision_training: Native AMP

Training results

Training LossEpochStepValidation LossAccuracy
1.49860.131003.14270.0791
1.19290.252003.40830.0726
1.26730.383003.46150.0769
0.98050.514003.91920.0824
1.1580.635004.26480.0698
1.25440.766004.55360.0574
1.00730.897004.03100.0819
0.93151.028004.51540.0702
0.90631.149004.71620.0633
0.67561.2710004.64820.0626
1.02391.411004.64370.0635
0.76341.5212004.56250.0752
0.83651.6513004.99120.0561
0.89791.7814005.17390.0356
0.94481.915004.89460.0541
0.6972.0316004.95160.0741
0.78612.1617005.00900.0776
0.64042.2818005.39050.0643
0.79392.4119004.91590.1015
0.63312.5420005.30830.0589
0.60822.6621004.85380.0857
0.62292.7922005.30860.0689
0.69642.9223005.37450.0713
0.52463.0524005.03690.0796
0.60973.1725005.29350.0743
0.57783.326005.54310.0709
0.41963.4327005.55080.0759
0.54953.5528005.57280.0813
0.59323.6829005.79920.0663
0.43823.8130005.80100.0643
0.48273.9331005.75290.0680

Framework versions

  • Transformers 4.21.2
  • Pytorch 1.12.1+cu113
  • Datasets 2.4.0
  • Tokenizers 0.12.1

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей