/ Модель

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

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

modeversion1_m7_e4

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

Дообученный Vision Transformer для классификации изображений на классах набора sudo-s/herbier_mesuem7.

9скачиваний за 30 дней0лайковвеса обновлены23 июл 2022

Бенчмарки

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

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

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

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

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

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

modeversion1m7e4 — дообученная модель классификации изображений на архитектуре Vision Transformer (ViT). Она создана на основе google/vit-base-patch16-224-in21k и обучена на наборе данных sudo-s/herbier_mesuem7. Модель предназначена для отнесения изображений к классам; на оценочном наборе указаны loss 0,0902 и accuracy 0,9731.

Модель содержит 113,2 млн параметров и опубликована в формате PyTorch. README не раскрывает назначение классов и сценарии применения набора данных, поэтому область изображений, для которой модель специализирована, определить нельзя. Лицензия — Apache-2.0.

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

  • Классификация изображений по классам обучающего набора
  • Автоматическая категоризация изображений в экспериментальном проекте
  • Оценка изображений с помощью модели ViT

Ограничения

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

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

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

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

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

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

Открыть на Hugging Face

modeversion1_m7_e4

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

  • Loss: 0.0902
  • Accuracy: 0.9731

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
4.0730.061003.93700.1768
3.41860.122003.27210.2590
2.67450.183002.64650.3856
2.28060.234002.26000.4523
1.92750.295001.96530.5109
1.69580.356001.68150.6078
1.27970.417001.45140.6419
1.37720.478001.32120.6762
1.17650.539001.14760.7028
1.01520.5910001.03570.7313
0.78610.6411001.02300.7184
1.02620.712000.94690.7386
0.89050.7613000.81840.7756
0.69190.8214000.80830.7711
0.74940.8815000.76010.7825
0.50780.9416000.68840.8056
0.71341.017000.63110.8160
0.43281.0618000.57400.8252
0.49711.1119000.58560.8290
0.52071.1720000.62190.8167
0.40271.2321000.57030.8266
0.56051.2922000.52170.8372
0.27231.3523000.48050.8565
0.4011.4124000.48110.8490
0.34191.4725000.46190.8608
0.3011.5226000.43180.8712
0.28721.5827000.46980.8573
0.24511.6428000.42100.8729
0.22111.729000.36450.8851
0.31451.7630000.41390.8715
0.20011.8231000.36050.8864
0.30951.8832000.42740.8675
0.19151.9333000.29100.9101
0.24651.9934000.27260.9103
0.12182.0535000.27420.9129
0.07522.1136000.25720.9183
0.10672.1737000.25840.9203
0.08382.2338000.24580.9212
0.11062.2939000.24120.9237
0.0922.3440000.22320.9277
0.10562.441000.28170.9077
0.06962.4642000.23340.9285
0.04442.5243000.21420.9363
0.10462.5844000.20360.9352
0.0662.6445000.21150.9365
0.06492.746000.17300.9448
0.05132.7547000.21480.9339
0.09172.8148000.18100.9438
0.08792.8749000.19710.9388
0.10522.9350000.16020.9508
0.03622.9951000.14750.9556
0.0413.0552000.13280.9585
0.01563.1153000.13890.9571
0.00473.1754000.12240.9638
0.01743.2255000.11930.9651
0.00873.2856000.12760.9622
0.00843.3457000.11340.9662
0.01413.458000.12390.9631
0.02913.4659000.11990.9645
0.00493.5260000.11030.9679
0.00553.5861000.11200.9662
0.00613.6362000.10710.9668
0.00543.6963000.10320.9697
0.00413.7564000.09610.9711
0.00183.8165000.09300.9718
0.00323.8766000.09180.9730
0.00483.9367000.09060.9732
0.0023.9968000.09020.9731

Framework versions

  • Transformers 4.20.1
  • Pytorch 1.12.0
  • Datasets 2.3.2
  • Tokenizers 0.12.1

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей