/ Модель

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

Контекст
1 024 токена
Параметры
6,3 млн
Лицензия
Apache 2.0
Архитектура
detr · dense
Базовая модель
detr-resnet-50
Библиотека
transformers
Опубликована
10 окт 2024

detr

Поиск объектовИзображение → bounding_boxes

Дообученная DETR на ResNet-50 для обнаружения людей, лиц, частей тела и предметов на изображениях.

19скачиваний за 30 дней0лайковвеса обновлены11 окт 2024

Бенчмарки

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

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

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

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

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

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

Модель b09501048/detr — дообученная версия DETR на ResNet-50 от Facebook, предназначенная для обнаружения объектов на изображениях и предсказания их классов и ограничивающих рамок. В карточке приведены метрики оценки для классов person, face, head, hands и ряда предметов одежды и средств защиты; результаты заметно различаются между классами, а для многих классов mAP равен нулю.

Модель использует архитектуру DetrForObjectDetection и опубликована в формате safetensors для библиотеки Transformers. Она дообучена на неизвестном наборе данных; карточка не описывает набор данных, предполагаемые сценарии применения или дополнительные языковые возможности. Лицензия — Apache-2.0.

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

  • обнаружение людей и лиц на изображениях
  • поиск головы и рук на изображениях
  • выявление средств индивидуальной защиты на изображениях
  • локализация обуви, инструментов и других объектов на изображениях

Ограничения

Набор данных и условия применения не описаны. Метрики низкие для многих классов: например, mAP равен нулю для средств защиты, инструментов и обуви; качество следует проверять на целевых данных.

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

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

Открыть на Hugging Face

detr

This model is a fine-tuned version of facebook/detr-resnet-50 on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 2.1588
  • Map: 0.0662
  • Map 50: 0.1137
  • Map 75: 0.0667
  • Map Small: 0.0
  • Map Medium: 0.0013
  • Map Large: 0.0706
  • Mar 1: 0.0705
  • Mar 10: 0.1303
  • Mar 100: 0.1374
  • Mar Small: 0.0
  • Mar Medium: 0.009
  • Mar Large: 0.1536
  • Map Person: 0.5545
  • Mar 100 Person: 0.6988
  • Map Ear: 0.0068
  • Mar 100 Ear: 0.1599
  • Map Earmuffs: 0.0
  • Mar 100 Earmuffs: 0.0
  • Map Face: 0.1494
  • Mar 100 Face: 0.3924
  • Map Face-guard: 0.0
  • Mar 100 Face-guard: 0.0
  • Map Face-mask-medical: 0.0
  • Mar 100 Face-mask-medical: 0.0
  • Map Foot: 0.0
  • Mar 100 Foot: 0.0
  • Map Tools: 0.0012
  • Mar 100 Tools: 0.0722
  • Map Glasses: 0.0
  • Mar 100 Glasses: 0.0
  • Map Gloves: 0.0
  • Mar 100 Gloves: 0.0
  • Map Helmet: 0.0
  • Mar 100 Helmet: 0.0
  • Map Hands: 0.1315
  • Mar 100 Hands: 0.4111
  • Map Head: 0.274
  • Mar 100 Head: 0.5671
  • Map Medical-suit: 0.0
  • Mar 100 Medical-suit: 0.0
  • Map Shoes: 0.0073
  • Mar 100 Shoes: 0.035
  • Map Safety-suit: 0.0
  • Mar 100 Safety-suit: 0.0
  • Map Safety-vest: 0.0
  • Mar 100 Safety-vest: 0.0

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: 5e-05
  • train_batch_size: 32
  • eval_batch_size: 32
  • seed: 42
  • gradient_accumulation_steps: 2
  • total_train_batch_size: 64
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: cosine
  • num_epochs: 10
  • mixed_precision_training: Native AMP

Training results

Training LossEpochStepValidation LossMapMap 50Map 75Map SmallMap MediumMap LargeMar 1Mar 10Mar 100Mar SmallMar MediumMar LargeMap PersonMar 100 PersonMap EarMar 100 EarMap EarmuffsMar 100 EarmuffsMap FaceMar 100 FaceMap Face-guardMar 100 Face-guardMap Face-mask-medicalMar 100 Face-mask-medicalMap FootMar 100 FootMap ToolsMar 100 ToolsMap GlassesMar 100 GlassesMap GlovesMar 100 GlovesMap HelmetMar 100 HelmetMap HandsMar 100 HandsMap HeadMar 100 HeadMap Medical-suitMar 100 Medical-suitMap ShoesMar 100 ShoesMap Safety-suitMar 100 Safety-suitMap Safety-vestMar 100 Safety-vest
No log0.9913572.99950.02530.03910.02710.00.00010.02630.03190.05690.06840.00.00050.07250.37570.65610.00020.00460.00.00.00060.00420.00.00.00.00.00.00.00.00.00.00.00.00.00.00.02110.30650.03330.19190.00.00.00.00.00.00.00.0
No log2.01152.69460.03120.05180.03230.00.00020.03270.04110.07750.08740.00.0030.0940.420.67620.00070.02460.00.00.00530.05150.00.00.00.00.00.00.00.00.00.00.00.00.00.00.03730.34270.06690.39160.00.00.00.00.00.00.00.0
No log2.99131722.55270.03250.05950.03080.00.0020.03430.04410.08530.09610.00.00570.10580.36490.6520.00210.07980.00.00.01420.07480.00.00.00.00.00.00.00220.00730.00.00.00.00.00.00.05080.33840.11780.4820.00.00.00.00.00.00.00.0
No log4.02302.46370.04250.07880.03960.00.00060.04490.05030.09680.10590.00.00490.11680.4510.65640.00180.08670.00.00.05480.17640.00.00.00.00.00.00.00170.02250.00.00.00.00.00.00.07250.35540.14020.50270.00.00.00.00.00.00.00.0
No log4.99132872.34550.05270.09290.0520.00.00060.05610.06250.11540.12340.00.00420.13750.49240.67710.00680.13460.00.00.10610.35050.00.00.00.00.00.00.00110.02780.00.00.00.00.00.00.09160.38020.19180.52320.00.00.00580.00470.00.00.00.0
No log6.03452.28590.05380.09380.05340.00.00110.05720.06240.11750.12680.00.00510.14120.51060.69450.00370.12840.00.00.11080.33330.00.00.00.00.00.00.00060.05030.00.00.00.00.00.00.10480.40160.17930.53940.00.00.00550.00880.00.00.00.0
No log6.99134022.21260.06090.10560.05960.00.00190.0650.06810.12570.13190.00.00860.14690.52880.69150.00640.14120.00.00.13660.38860.00.00.00.00.00.00.00060.05170.00.00.00.00.00.00.12040.39050.23710.56370.00.00.00550.01540.00.00.00.0
No log8.04602.17940.06410.110.06430.00.00150.06850.06990.12940.13720.00.00890.15330.55250.69640.00640.15990.00.00.13750.38920.00.00.00.00.00.00.00120.07750.00.00.00.00.00.00.12730.40890.25860.5670.00.00.0070.03330.00.00.00.0
2.84818.99135172.16420.06580.1130.06610.00.00160.07030.07020.12960.13690.00.00910.15320.55290.69690.00670.16060.00.00.14750.38750.00.00.00.00.00.00.00120.07250.00.00.00.00.00.00.130.41020.27270.56590.00.00.00730.03440.00.00.00.0
2.84819.91305702.15880.06620.11370.06670.00.00130.07060.07050.13030.13740.00.0090.15360.55450.69880.00680.15990.00.00.14940.39240.00.00.00.00.00.00.00120.07220.00.00.00.00.00.00.13150.41110.2740.56710.00.00.00730.0350.00.00.00.0

Framework versions

  • Transformers 4.44.2
  • Pytorch 2.4.1+cu121
  • Datasets 3.0.1
  • Tokenizers 0.19.1

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

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

GPU в ailist

Универсальный рантайм · transformers · object-detection

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

Подбираем GPU…

Класс архитектуры
DetrForObjectDetection
Ревизия весов
0168c83

0 моделей