/ Модель

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

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

detr-resnet-50-finetuned-10k-cppe5

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

Дообученная на cppe-5 модель DETR с ResNet-50 обнаруживает средства индивидуальной защиты на изображениях.

127скачиваний за 30 дней1лайквеса обновлены8 мая 2024

Бенчмарки

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

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

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

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

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

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

Это модель обнаружения объектов на изображениях, разработанная путём дообучения facebook/detr-resnet-50 на наборе данных cppe-5. Модель основана на архитектуре DETR с ResNet-50 и выдаёт классы объектов и ограничивающие рамки. В репозитории указано 6,3 млн параметров; веса доступны в формате safetensors.

Модель предназначена для обнаружения средств индивидуальной защиты на изображениях: комбинезонов, защитных экранов для лица, перчаток, очков и масок. На оценочном наборе автор приводит mAP 0,3578 и mAP@50 0,6781; метрики различаются по классам, например, mAP для комбинезонов составляет 0,6584, а для очков — 0,1766. Эти результаты относятся к cppe-5 и не гарантируют такое же качество на других изображениях.

Лицензия — Apache-2.0. Модель принимает изображения и возвращает обнаруженные объекты с классами и рамками; в README не приведены сведения о языковой поддержке.

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

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

Ограничения

Качество зависит от класса: приведённая mAP варьируется от 0,1766 для очков до 0,6584 для комбинезонов. Результаты указаны для cppe-5 и могут не переноситься на другие условия съёмки и наборы данных.

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

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

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

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

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

Открыть на Hugging Face

detr-resnet-50-finetuned-10k-cppe5

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

  • Loss: 0.9865
  • Map: 0.3578
  • Map 50: 0.6781
  • Map 75: 0.3105
  • Map Small: 0.3578
  • Map Medium: -1.0
  • Map Large: -1.0
  • Mar 1: 0.365
  • Mar 10: 0.535
  • Mar 100: 0.5483
  • Mar Small: 0.5483
  • Mar Medium: -1.0
  • Mar Large: -1.0
  • Map Coverall: 0.6584
  • Mar 100 Coverall: 0.7772
  • Map Face Shield: 0.3691
  • Mar 100 Face Shield: 0.6063
  • Map Gloves: 0.2477
  • Mar 100 Gloves: 0.4266
  • Map Goggles: 0.1766
  • Mar 100 Goggles: 0.4655
  • Map Mask: 0.3371
  • Mar 100 Mask: 0.4661

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: 8
  • eval_batch_size: 8
  • seed: 1337
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 100.0
  • mixed_precision_training: Native AMP

Training results

Training LossEpochStepValidation LossMapMap 50Map 75Map SmallMap MediumMap LargeMar 1Mar 10Mar 100Mar SmallMar MediumMar LargeMap CoverallMar 100 CoverallMap Face ShieldMar 100 Face ShieldMap GlovesMar 100 GlovesMap GogglesMar 100 GogglesMap MaskMar 100 Mask
3.71981.01073.18690.00360.01440.00060.0036-1.0-1.00.0170.04260.06470.0647-1.0-1.00.01690.19940.00.00.00010.01770.00.00.00080.1062
3.03932.02142.89780.0070.0210.0030.007-1.0-1.00.01920.07960.12380.1238-1.0-1.00.03330.52160.00.00.00010.02920.00.00.00170.0684
2.78123.03212.54450.01380.04140.00850.0138-1.0-1.00.03070.09940.12580.1258-1.0-1.00.06550.49380.00.00.00020.03540.00.00.00330.1
2.59924.04282.38280.02320.06010.01550.0232-1.0-1.00.04230.12020.15180.1518-1.0-1.00.10210.54810.00.00.00060.04950.00590.01090.00720.1503
2.38285.05352.26720.02830.07030.01790.0283-1.0-1.00.05210.12830.17370.1737-1.0-1.00.13440.58460.00.00.0010.08330.00.00.00630.2006
2.26336.06422.06180.04790.09960.04160.0479-1.0-1.00.07820.16790.20350.2035-1.0-1.00.20990.63330.0030.01590.00180.11870.00520.02180.01950.2277
2.18377.07492.11000.04550.11590.02550.0455-1.0-1.00.07470.15820.18940.1894-1.0-1.00.20680.61850.00850.05560.0010.07340.00020.00180.01130.1977
2.06898.08562.00000.0540.13890.03010.054-1.0-1.00.09540.18460.21590.2159-1.0-1.00.21550.55370.03140.13970.00490.14060.00020.00180.01810.2435
2.04179.09631.87020.06970.16310.05010.0697-1.0-1.00.10740.21730.2570.257-1.0-1.00.28260.60860.02790.1810.00460.17340.01020.04180.02340.2802
1.997210.010701.85630.07420.15680.05410.0742-1.0-1.00.11960.24160.27860.2786-1.0-1.00.29330.60860.02330.19210.00530.16720.02390.08910.0250.3362
1.893111.011771.67780.10540.22480.08980.1054-1.0-1.00.14560.27640.30330.3033-1.0-1.00.39550.6710.04980.26030.01080.21880.01490.03820.0560.3282
1.826912.012841.69050.11110.23990.09420.1111-1.0-1.00.15430.29490.32570.3257-1.0-1.00.41130.6790.0690.3190.00870.20210.0150.09090.05140.3373
1.803613.013911.64060.11490.24070.0970.1149-1.0-1.00.16360.31080.33720.3372-1.0-1.00.42550.67590.07710.33810.01090.21820.01370.13090.0470.3226
1.746314.014981.71690.11060.24210.08750.1106-1.0-1.00.17760.32050.35110.3511-1.0-1.00.39960.70.04040.24760.01170.24580.02570.20360.07570.3582
1.76315.016051.59610.12450.25770.10180.1245-1.0-1.00.18170.33840.36770.3677-1.0-1.00.45750.66790.07750.36980.01070.25050.03180.19640.04470.3537
1.646716.017121.53650.13760.30730.10620.1376-1.0-1.00.21640.380.4080.408-1.0-1.00.4550.68520.07390.38730.02150.27190.04420.28910.09340.4068
1.622217.018191.59900.12950.26960.10260.1295-1.0-1.00.18020.34090.36930.3693-1.0-1.00.45770.66540.07860.36190.02970.29580.02110.22180.06030.3017
1.623918.019261.41640.1590.35430.12620.159-1.0-1.00.2350.39290.41380.4138-1.0-1.00.47530.72040.09210.39680.0390.29220.03230.26360.15650.396
1.544819.020331.46890.16280.37250.13140.1628-1.0-1.00.2050.38110.40640.4064-1.0-1.00.47940.68950.10380.4190.03980.28280.03330.280.15780.3605
1.502620.021401.40930.17980.3970.13690.1798-1.0-1.00.23360.41250.43490.4349-1.0-1.00.48510.68580.14940.45080.03410.28590.04340.33820.18690.4136
1.479721.022471.46050.16520.36050.12540.1652-1.0-1.00.22950.38230.40410.4041-1.0-1.00.49780.69570.09680.38250.05290.27970.02630.32360.15220.339
1.429822.023541.42310.1630.35580.1150.163-1.0-1.00.22560.38510.41080.4108-1.0-1.00.49020.70930.10330.41590.05150.3130.02610.31090.14370.3051
1.415723.024611.36650.19140.40480.15330.1914-1.0-1.00.24780.42320.4470.447-1.0-1.00.4910.69750.15990.46830.05020.30210.06030.36180.19560.4051
1.443824.025681.29080.21030.4330.1680.2103-1.0-1.00.26430.45120.47610.4761-1.0-1.00.53680.71360.14930.48730.07890.36090.0430.38910.24330.4294
1.404425.026751.47520.17090.37490.13880.1709-1.0-1.00.21870.39260.41910.4191-1.0-1.00.48620.71670.090.39050.07620.2990.03930.35270.16270.3367
1.370326.027821.30470.21620.45680.17140.2162-1.0-1.00.26610.43440.45480.4548-1.0-1.00.53420.72720.1660.45080.09710.32810.04240.35270.24140.4153
1.329227.028891.26740.220.46810.17020.22-1.0-1.00.27430.42860.44730.4473-1.0-1.00.54380.72650.21280.44290.11710.34430.03870.34550.18780.3774
1.35928.029961.31560.20070.42720.15360.2007-1.0-1.00.27150.43840.45550.4555-1.0-1.00.53060.71110.1630.50160.08960.31350.03070.380.18980.3712
1.347129.031031.26460.21610.4550.1720.2161-1.0-1.00.2770.44920.47280.4728-1.0-1.00.53010.72160.17080.5190.12160.32710.03910.41450.21890.3819
1.30830.032101.30170.21070.44650.17180.2107-1.0-1.00.25560.41410.43870.4387-1.0-1.00.53210.71360.15310.4540.10370.32030.03340.32180.23130.3836
1.302331.033171.28090.21740.4620.17140.2174-1.0-1.00.26460.42420.44730.4473-1.0-1.00.54840.72590.16860.4270.11630.35360.05060.35640.20290.3734
1.256132.034241.26790.20820.45570.15790.2082-1.0-1.00.2630.42830.45110.4511-1.0-1.00.5410.72840.14480.45560.13080.36820.04110.34360.18350.3599
1.26833.035311.26320.23040.48410.18820.2304-1.0-1.00.28520.44690.46290.4629-1.0-1.00.55350.73270.20420.52860.10270.29170.04720.380.24410.3814
1.233734.036381.21570.23390.4970.18540.2339-1.0-1.00.28450.44950.47560.4756-1.0-1.00.57040.74140.1870.50950.13390.34480.04140.39820.23670.3842
1.219235.037451.21090.25050.50780.20130.2505-1.0-1.00.2880.46170.47950.4795-1.0-1.00.55990.72590.24310.50160.15640.35210.04960.41640.24370.4017
1.211936.038521.18970.25840.52290.21980.2584-1.0-1.00.29860.45750.47440.4744-1.0-1.00.57420.73830.26670.51590.13890.350.04480.34730.26760.4203
1.168637.039591.14410.25810.51640.20720.2581-1.0-1.00.30020.47120.49180.4918-1.0-1.00.59040.76420.24060.52380.15350.37080.04040.39450.26540.4056
1.166438.040661.15540.26180.5460.2040.2618-1.0-1.00.30780.48220.50690.5069-1.0-1.00.56740.7340.24860.5190.19220.39270.04090.45090.260.4379
1.174639.041731.18730.25140.52690.19950.2514-1.0-1.00.2920.46280.48420.4842-1.0-1.00.55270.72350.24270.51270.15910.34480.04680.42180.25570.4181
1.155940.042801.15940.26520.55170.20140.2652-1.0-1.00.29550.47430.48850.4885-1.0-1.00.56850.73770.26240.52380.18520.37550.05050.38730.25950.4181
1.166841.043871.15620.24790.52940.18530.2479-1.0-1.00.29670.47720.49940.4994-1.0-1.00.57420.72720.2170.54130.14920.34530.0580.47090.24110.4124
1.108442.044941.13020.26340.53520.20940.2634-1.0-1.00.31350.47830.49450.4945-1.0-1.00.58470.73330.23350.53970.16530.35830.04840.41270.28510.4282
1.111843.046011.12700.27540.55060.24080.2754-1.0-1.00.31030.47890.49730.4973-1.0-1.00.58670.73640.26620.54760.15870.36820.05990.38550.30570.4486
1.109644.047081.17720.26110.55650.20020.2611-1.0-1.00.28730.46060.47550.4755-1.0-1.00.57390.72280.2470.51430.16490.35160.0810.40730.23890.3814
1.074145.048151.08710.28570.57180.21860.2857-1.0-1.00.32370.49340.51040.5104-1.0-1.00.59270.750.27750.56980.18990.37860.07810.41450.29040.439
1.077446.049221.10490.26840.54610.21970.2684-1.0-1.00.31120.49560.51150.5115-1.0-1.00.59790.75740.25520.57940.1430.36770.0620.42910.2840.4237
1.051747.050291.10050.28060.58210.21840.2806-1.0-1.00.31120.48720.50480.5048-1.0-1.00.58610.75430.28620.56190.17490.40160.07370.39450.2820.4119
1.071748.051361.08780.29560.59020.24920.2956-1.0-1.00.31460.49250.50890.5089-1.0-1.00.61280.75060.28130.53810.19470.40360.08490.40910.30410.4429
1.057849.052431.09320.28160.56710.22460.2816-1.0-1.00.31370.4790.49430.4943-1.0-1.00.60690.75370.26160.48730.19470.40260.06130.39640.28350.4316
1.036750.053501.07840.29560.58130.24160.2956-1.0-1.00.33080.48640.49790.4979-1.0-1.00.61930.76420.28090.51430.18410.38540.08850.38910.30530.4367
1.023951.054571.07020.28930.56890.23870.2893-1.0-1.00.3150.49440.50970.5097-1.0-1.00.59780.7630.26560.50950.18870.39380.08170.43090.31250.4514
1.012752.055641.07710.28330.57130.2430.2833-1.0-1.00.31840.48560.50750.5075-1.0-1.00.60070.76050.26490.51750.18430.39320.07660.43640.29020.4299
0.994853.056711.10390.28210.56280.26070.2821-1.0-1.00.31430.49540.51320.5132-1.0-1.00.60020.75990.28460.55870.17290.36930.06460.45820.28840.4198
1.002654.057781.11250.28110.55690.24610.2811-1.0-1.00.31660.48460.50190.5019-1.0-1.00.61180.76910.26170.52860.17620.3760.06250.39640.29320.4395
1.003755.058851.11720.28190.57570.25320.2819-1.0-1.00.30890.46530.48110.4811-1.0-1.00.58690.73020.29460.50320.19140.3880.05650.37450.28020.4096
0.989256.059921.06490.3110.60270.26730.311-1.0-1.00.32260.49540.5150.515-1.0-1.00.6290.76790.29720.55560.1980.39010.12330.41820.30750.4435
0.9857.060991.04130.31370.62440.29120.3137-1.0-1.00.32930.50210.51710.5171-1.0-1.00.62880.75990.31930.56980.20620.3990.11010.42360.30430.4333
0.968258.062061.03010.31280.60030.28010.3128-1.0-1.00.32780.49630.5170.517-1.0-1.00.6320.77040.30380.52380.20930.42030.12040.41640.29840.4542
0.953359.063131.07550.29280.59540.250.2928-1.0-1.00.31770.49370.50840.5084-1.0-1.00.61020.74010.29060.56350.19420.3990.10020.42180.26880.4175
0.958960.064201.03640.32130.61820.27170.3213-1.0-1.00.33830.51210.52730.5273-1.0-1.00.6220.76170.33180.56830.22730.41930.12430.42360.3010.4638
0.939861.065271.06530.30890.60990.26730.3089-1.0-1.00.33250.49460.50460.5046-1.0-1.00.61660.75560.31440.52860.1770.3760.14510.41270.29150.4503
0.92462.066341.04090.3190.63350.26410.319-1.0-1.00.34370.50850.52340.5234-1.0-1.00.62230.75370.32630.54760.210.40210.12690.45820.30930.4554
0.92463.067411.05170.32120.64850.27270.3212-1.0-1.00.33820.51090.52320.5232-1.0-1.00.63920.77040.31950.58570.19670.3990.12680.42180.32410.439
0.945864.068481.03140.32130.63570.280.3213-1.0-1.00.34140.51520.52780.5278-1.0-1.00.62720.76480.31370.5730.2270.40780.12570.45450.31290.439
0.913665.069551.02370.32620.63770.2840.3262-1.0-1.00.34890.51570.52990.5299-1.0-1.00.62770.7660.32940.55870.22230.39580.130.46550.32160.4633
0.896966.070621.01900.31880.64780.28330.3188-1.0-1.00.34560.51530.5280.528-1.0-1.00.62540.77040.33380.57620.1870.38590.12580.45820.3220.4492
0.891767.071690.99000.3260.63040.2810.326-1.0-1.00.3540.52890.53820.5382-1.0-1.00.64780.77590.31180.55560.22450.41150.11490.46910.33120.4791
0.872468.072761.01810.31410.61190.27570.3141-1.0-1.00.33980.5170.53030.5303-1.0-1.00.6420.77160.31180.57780.20910.41770.0960.42550.31160.4588
0.888169.073831.02310.31190.630.25660.3119-1.0-1.00.34380.5070.51480.5148-1.0-1.00.62780.75560.30620.54760.20770.40520.09510.40730.32280.4582
0.860470.074901.02450.32660.64360.26210.3266-1.0-1.00.34480.51790.52730.5273-1.0-1.00.62510.76230.33010.58890.22230.41150.14060.42550.3150.4486
0.8671.075971.02490.32260.64260.27090.3226-1.0-1.00.34420.52010.53250.5325-1.0-1.00.63530.77160.31710.60320.22330.41770.12890.42360.30820.4463
0.845272.077041.03830.32250.63640.26290.3225-1.0-1.00.3440.51240.52230.5223-1.0-1.00.63750.77280.3220.57940.20820.3990.14430.42910.30030.4311
0.844473.078111.00180.33120.64540.29140.3312-1.0-1.00.35580.52230.5350.535-1.0-1.00.63950.7710.34710.58410.21580.41820.14330.43450.31050.4672
0.832574.079181.04120.32960.65740.2720.3296-1.0-1.00.34040.52030.53120.5312-1.0-1.00.64490.77220.34220.60320.23020.41250.13520.45090.29560.4169
0.843875.080251.00150.340.64820.30270.34-1.0-1.00.34870.52110.53330.5333-1.0-1.00.64490.76910.35550.59680.23240.41720.14270.43820.32480.4452
0.819776.081321.01960.34260.65440.29310.3426-1.0-1.00.3560.52340.53470.5347-1.0-1.00.64430.77220.350.57940.24650.41720.14340.44550.32880.4593
0.824177.082391.02150.3390.65310.30120.339-1.0-1.00.35230.51620.52790.5279-1.0-1.00.64110.7710.34860.59210.23240.40730.13750.40910.33570.4599
0.801978.083461.03560.32650.62790.27990.3265-1.0-1.00.35030.51690.52640.5264-1.0-1.00.6390.7710.33130.57620.23150.4130.11440.41270.31620.4593
0.811379.084531.00870.33770.65560.30590.3377-1.0-1.00.34870.52690.53690.5369-1.0-1.00.63960.77650.33920.58730.23360.40680.15130.45640.32490.4576
0.804380.085601.02250.33510.65260.30350.3351-1.0-1.00.34850.51770.52850.5285-1.0-1.00.65290.78270.3310.57780.22240.3990.14210.42550.32730.4576
0.790981.086671.00090.3450.65630.30860.345-1.0-1.00.35120.5220.53590.5359-1.0-1.00.66490.78770.35890.57140.23660.42190.13740.44180.32720.4565
0.803482.087741.00060.3330.64550.2950.333-1.0-1.00.35290.52570.53690.5369-1.0-1.00.63870.77410.33520.58730.23420.42080.13620.44910.32060.4531
0.774483.088810.99460.3410.65130.31820.341-1.0-1.00.36150.52730.53810.5381-1.0-1.00.64250.77780.36360.58730.24040.41980.12460.44180.33410.4638
0.776984.089880.99940.34060.65230.30090.3406-1.0-1.00.36370.53040.53840.5384-1.0-1.00.64780.77280.34720.59680.23550.41980.14440.43820.32810.4644
0.767985.090951.01700.33970.65490.28880.3397-1.0-1.00.36240.53010.54070.5407-1.0-1.00.64850.76850.34310.60790.22940.41670.16070.460.3170.4503
0.77286.092020.99530.35150.67220.30990.3515-1.0-1.00.36740.53310.5430.543-1.0-1.00.65990.77780.36920.60320.240.41870.15230.460.33610.4554
0.78387.093091.00030.34010.65450.29430.3401-1.0-1.00.36030.53090.54570.5457-1.0-1.00.6510.77590.35210.62220.22960.41610.14330.460.32460.4542
0.750888.094160.98490.35170.67340.32170.3517-1.0-1.00.3620.53330.54470.5447-1.0-1.00.65720.77470.3690.59840.24390.43180.15030.45270.3380.4661
0.755889.095230.98610.35170.67490.31730.3517-1.0-1.00.36280.53290.54080.5408-1.0-1.00.64690.77280.36610.58890.23230.41040.17470.47090.33850.461
0.739890.096300.99660.35650.67840.3150.3565-1.0-1.00.36210.53670.54840.5484-1.0-1.00.65240.77960.36550.61270.23740.41510.17690.46550.35030.4689
0.753291.097370.98830.3540.67090.31790.354-1.0-1.00.36230.53950.54750.5475-1.0-1.00.65510.7790.36690.61750.24320.41870.15450.45640.35020.4661
0.741792.098440.99320.35570.67770.31310.3557-1.0-1.00.36290.53610.54650.5465-1.0-1.00.65660.77650.37130.61430.24250.42240.1630.44910.34510.4701
0.755493.099510.99210.35980.68410.32120.3598-1.0-1.00.36430.53750.54660.5466-1.0-1.00.65910.78210.36960.59520.24820.42140.1750.46910.34690.465
0.733994.0100581.00490.35870.68380.31840.3587-1.0-1.00.36680.53470.54470.5447-1.0-1.00.6580.77530.37670.60480.24340.41770.17440.460.34110.4655
0.718495.0101650.99690.35470.67390.32140.3547-1.0-1.00.36330.53530.54490.5449-1.0-1.00.65210.77220.3690.60790.24020.4130.17070.46730.34120.4638
0.744896.0102720.99250.35560.67080.30610.3556-1.0-1.00.3650.53270.5440.544-1.0-1.00.65170.77650.37070.60160.23540.42080.17810.45270.34210.4684
0.726497.0103790.99140.35760.67930.3150.3576-1.0-1.00.36570.53320.54390.5439-1.0-1.00.65170.77530.3780.60.24640.4260.17070.45450.34140.4638
0.745598.0104860.98290.35860.67940.31320.3586-1.0-1.00.3670.53530.54730.5473-1.0-1.00.65860.77720.37810.60790.24680.42660.17420.45820.33550.4667
0.710599.0105930.98620.35770.67950.30890.3577-1.0-1.00.36490.53520.54760.5476-1.0-1.00.65910.77780.37120.60320.24810.42710.17270.46550.33760.4644
0.7089100.0107000.98650.35780.67810.31050.3578-1.0-1.00.3650.5350.54830.5483-1.0-1.00.65840.77720.36910.60630.24770.42660.17660.46550.33710.4661

Framework versions

  • Transformers 4.40.0.dev0
  • Pytorch 1.13.0+cu117
  • Datasets 2.18.0
  • Tokenizers 0.19.0

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей