/ Модель

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

Лицензия
Своя лицензия
Архитектура
segformer · dense
Базовая модель
mit-b5
Опубликована
6 сен 2023

segformer-b5-finetuned-100by100PNG-50epochs-attempt2-100epochs-backgroundclass-2

Сегментация изображенийИзображение → mask

Дообученная SegFormer B5 для семантической сегментации ветвей и фона на изображениях.

12скачиваний за 30 дней0лайковвеса обновлены6 сен 2023

Бенчмарки

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

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

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

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

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

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

Модель JCAI2000/segformer-b5-finetuned-100by100PNG-50epochs-attempt2-100epochs-backgroundclass-2 — дообученная на датасете JCAI2000/100By100BranchPNG модель SegFormer B5 для семантической сегментации изображений. Она размечает пиксели на классы «фон» и «ветка» и подходит для анализа изображений с ветвями на фоне.

В README приведены результаты на оценочной выборке: mean IoU 0,8933, mean accuracy 0,9531 и overall accuracy 0,9662. Точность для фона составляет 0,9732, для ветки — 0,9330; IoU для этих классов — 0,9597 и 0,8270 соответственно. Модель распространяется в формате PyTorch под лицензией other; условия этой лицензии следует проверить перед использованием.

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

  • выделение ветвей на изображениях
  • сегментация фона и ветвей на изображениях размером 100×100 пикселей
  • подсчёт или анализ площади ветвей по пиксельной маске

Ограничения

Показатели приведены только для оценочной выборки указанного датасета и не гарантируют такое же качество на других изображениях. README не описывает обучающие данные и условия применения подробнее; лицензия указана как other.

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

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

Открыть на Hugging Face

segformer-b3-finetuned-100by100PNG-50epochs-attempt2-100epochs-backgroundclass

This model is a fine-tuned version of nvidia/mit-b5 on the JCAI2000/100By100BranchPNG dataset. It achieves the following results on the evaluation set:

  • Loss: 0.1497
  • Mean Iou: 0.8933
  • Mean Accuracy: 0.9531
  • Overall Accuracy: 0.9662
  • Accuracy Background: 0.9732
  • Accuracy Branch: 0.9330
  • Iou Background: 0.9597
  • Iou Branch: 0.8270

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: 6e-05
  • train_batch_size: 2
  • eval_batch_size: 2
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 100

Training results

Training LossEpochStepValidation LossMean IouMean AccuracyOverall AccuracyAccuracy BackgroundAccuracy BranchIou BackgroundIou Branch
0.20551.05200.29250.81510.94690.93200.92420.96950.91830.7118
0.15492.11400.13280.88020.93110.96280.97960.88250.95610.8043
0.07353.16600.11780.88040.95120.96130.96660.93570.95380.8070
0.06364.21800.08440.89660.93680.96860.98540.88810.96290.8303
0.05465.261000.10990.89690.95260.96760.97560.92970.96140.8325
0.05676.321200.10120.89960.95000.96880.97880.92130.96290.8364
0.05157.371400.11370.89350.94620.96680.97770.91470.96050.8265
0.0528.421600.09870.89140.93170.96700.98580.87760.96110.8217
0.03589.471800.11670.89780.95810.96760.97260.94350.96130.8344
0.025410.532000.07670.91110.95190.97290.98400.91970.96780.8545
0.048311.582200.09530.90370.95240.97010.97950.92530.96450.8429
0.028512.632400.09040.90260.94900.97000.98110.91690.96430.8409
0.038913.682600.09020.90250.94720.97010.98210.91230.96440.8406
0.047314.742800.08520.90840.95220.97190.98230.92200.96650.8502
0.026615.793000.09830.89850.94090.96900.98390.89790.96330.8337
0.023316.843200.09650.90520.96010.97020.97560.94470.96440.8460
0.025717.893400.09410.90390.95500.97010.97810.93190.96430.8434
0.035218.953600.08550.90430.94830.97060.98240.91420.96510.8435
0.194120.03800.09460.90450.95090.97060.98090.92100.96500.8441
0.032521.054000.09720.89730.94490.96830.98070.90920.96240.8323
0.015922.114200.08280.90810.95280.97170.98170.92390.96640.8498
0.017523.164400.10610.89950.94910.96880.97930.91880.96290.8360
0.028124.214600.10900.89690.95160.96770.97610.92710.96150.8323
0.017725.264800.11220.89830.95470.96800.97500.93430.96180.8347
0.022826.325000.10880.89570.95460.96700.97360.93570.96060.8307
0.034827.375200.09330.90590.95240.97100.98080.92410.96540.8464
0.017728.425400.10530.90250.95270.96970.97870.92680.96390.8411
0.018229.475600.10390.89920.94730.96880.98020.91430.96300.8355
0.017130.535800.11170.89910.95550.96820.97500.93600.96210.8361
0.027531.586000.11420.89350.94970.96650.97540.92410.96010.8268
0.018632.636200.10650.90240.95240.96970.97880.92610.96390.8408
0.017333.686400.10810.89860.95290.96820.97640.92940.96210.8351
0.01534.746600.12430.89350.95300.96630.97330.93270.95980.8272
0.018335.796800.11200.90050.95000.96910.97920.92090.96330.8377
0.024836.847000.11850.89620.95170.96740.97570.92770.96110.8312
0.010437.897200.11360.89750.95060.96800.97710.92410.96190.8332
0.048138.957400.11270.90100.95280.96910.97780.92770.96320.8388
0.015340.07600.11010.90190.95370.96940.97770.92970.96350.8402
0.014341.057800.11050.90320.95580.96980.97710.93450.96390.8425
0.010442.118000.11220.89860.94280.96890.98270.90280.96310.8340
0.017243.168200.10970.90410.95400.97020.97880.92910.96450.8437
0.037144.218400.10640.90110.95030.96930.97940.92120.96350.8387
0.022145.268600.11500.90040.95150.96900.97830.92470.96310.8377
0.018646.328800.12280.89580.95180.96720.97540.92820.96100.8306
0.011947.379000.12050.89800.95250.96800.97620.92880.96190.8340
0.011348.429200.11330.89980.95020.96880.97870.92160.96290.8366
0.012149.479400.11450.89930.94900.96880.97920.91880.96290.8358
0.026350.539600.11680.89770.95420.96780.97500.93340.96160.8338
0.009351.589800.12130.89400.95340.96640.97330.93340.96000.8280
0.019352.6310000.12410.89710.95070.96780.97690.92460.96170.8326
0.013953.6810200.12630.89620.95460.96720.97390.93530.96090.8316
0.01254.7410400.12520.89520.95040.96710.97600.92470.96090.8296
0.00855.7910600.12190.89860.95160.96830.97720.92600.96230.8349
0.009256.8410800.12900.89950.95520.96840.97540.93490.96230.8366
0.01557.8911000.12430.89890.95450.96820.97550.93350.96210.8358
0.012658.9511200.12140.89770.95410.96780.97510.93310.96160.8337
0.021260.011400.12980.89530.95420.96690.97360.93470.96050.8301
0.019261.0511600.13410.89300.95180.96610.97370.92990.95970.8262
0.013662.1111800.13270.89700.95280.96760.97540.93020.96140.8325
0.013163.1612000.12330.89970.95490.96850.97570.93400.96240.8369
0.013564.2112200.13010.89570.95420.96700.97380.93450.96070.8307
0.022865.2612400.12740.89790.95240.96800.97620.92850.96180.8339
0.013866.3212600.13360.89650.95200.96750.97570.92830.96130.8318
0.012767.3712800.12780.89800.95190.96810.97670.92710.96200.8341
0.010768.4213000.12930.89700.95300.96760.97530.93080.96140.8327
0.027869.4713200.14130.89260.95340.96590.97250.93430.95930.8258
0.015970.5313400.13600.89530.95220.96700.97480.92960.96070.8298
0.010571.5813600.13190.89720.95370.96760.97500.93240.96140.8330
0.016872.6313800.13430.89420.95330.96650.97350.93310.96010.8283
0.015673.6814000.13570.89500.95160.96690.97510.92810.96060.8294
0.010974.7414200.14460.89050.95240.96520.97190.93280.95850.8226
0.016875.7914400.13390.89580.95330.96710.97450.93200.96080.8308
0.025276.8414600.13550.89350.95320.96620.97310.93330.95970.8272
0.010977.8914800.13880.89320.95330.96610.97290.93380.95960.8267
0.010978.9515000.14040.89240.95190.96590.97340.93050.95940.8255
0.011280.015200.14240.89210.95350.96570.97220.93490.95910.8251
0.009481.0515400.14510.89240.95240.96590.97300.93170.95930.8254
0.00782.1115600.14570.89310.95270.96610.97320.93220.95960.8266
0.011983.1615800.14240.89270.95200.96610.97350.93060.95950.8259
0.015384.2116000.15350.89090.95300.96530.97180.93420.95860.8233
0.010485.2616200.14520.89210.95290.96580.97250.93330.95920.8251
0.010186.3216400.15030.89100.95360.96530.97140.93580.95860.8235
0.00987.3716600.15080.89250.95320.96590.97250.93390.95930.8257
0.007388.4216800.14190.89490.95280.96680.97420.93150.96040.8293
0.013789.4717000.14370.89420.95260.96660.97390.93130.96010.8282
0.006190.5317200.14740.89280.95230.96600.97330.93130.95950.8260
0.013291.5817400.14080.89350.95220.96630.97380.93060.95980.8271
0.008992.6317600.14680.89330.95270.96620.97340.93200.95970.8268
0.014193.6817800.14580.89300.95290.96610.97310.93280.95960.8265
0.013294.7418000.14420.89340.95280.96620.97330.93240.95970.8270
0.010995.7918200.14450.89280.95290.96600.97300.93270.95950.8262
0.024896.8418400.13910.89380.95290.96640.97360.93220.95990.8276
0.007497.8918600.14240.89380.95310.96640.97350.93270.95990.8277
0.007298.9518800.14650.89310.95340.96610.97280.93390.95960.8266
0.0183100.019000.14970.89330.95310.96620.97320.93300.95970.8270

Framework versions

  • Transformers 4.33.0
  • Pytorch 2.0.1+cu117
  • Datasets 2.14.4
  • Tokenizers 0.13.3

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

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

GPU в ailist

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

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

Подбираем GPU…

Класс архитектуры
SegformerForSemanticSegmentation
Ревизия весов
0e60cf5

0 моделей