/ Модель

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

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

segformer-b5-finetuned-IDD-L2_v2

Сегментация изображенийИзображение → mask3 профессии

Дообученная SegFormer B5 модель для пиксельной сегментации объектов и элементов дорожных сцен на изображениях IDD.

1 668скачиваний за 30 дней0лайковвеса обновлены8 апр 2025

Бенчмарки

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

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

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

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

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

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

izzako/segformer-b5-finetuned-IDD-L2_v2 — модель семантической сегментации изображений, дообученная на наборе IDD 20K Semantic Segmentation Dataset на базе nvidia/mit-b5. Она размечает пиксели уличных сцен, включая дороги, парковки, тротуары, железнодорожные пути, людей, транспорт, здания, растительность и небо. Модель подходит для анализа дорожных сцен и восприятия городской инфраструктуры по изображениям.

В README приведены результаты на оценочной выборке: mean IoU 0,718, mean accuracy 0,822 и общая точность 0,908. Качество различается по классам: например, IoU дороги составляет 0,939, а ограждений — 0,457. Файлы модели представлены в формате safetensors. Лицензия указана как other, поэтому условия использования следует сверить с карточкой модели.

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

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

Ограничения

Качество зависит от класса: в оценке IoU ограждений составляет 0,457, а железнодорожных путей — 0,503. Результаты приведены для оценочной выборки IDD 20K и не гарантируют такое же качество на других данных.

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

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

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

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

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

Открыть на Hugging Face

segformer-b5-finetuned-IDD-L2_v2

This model is a fine-tuned version of nvidia/mit-b5 on the IDD 20K Semantic Segmentation Dataset dataset. It achieves the following results on the evaluation set:

  • Loss: 0.5563
  • Mean Iou: 0.7180
  • Mean Accuracy: 0.8224
  • Overall Accuracy: 0.9083
  • Accuracy Road: 0.9716
  • Accuracy Parking: 0.7949
  • Accuracy Sidewalk: 0.8240
  • Accuracy Rail track: 0.6408
  • Accuracy Person: 0.8057
  • Accuracy Rider: 0.8434
  • Accuracy Motorcycle: 0.8762
  • Accuracy Autorickshaw: 0.9451
  • Accuracy Truck: 0.9122
  • Accuracy Curb: 0.8112
  • Accuracy Fence: 0.5699
  • Accuracy Billboard: 0.7605
  • Accuracy Pole: 0.6010
  • Accuracy Building: 0.8678
  • Accuracy Vegetation: 0.9495
  • Accuracy Sky: 0.9841
  • Iou Road: 0.9391
  • Iou Parking: 0.6620
  • Iou Sidewalk: 0.6707
  • Iou Rail track: 0.5025
  • Iou Person: 0.6726
  • Iou Rider: 0.7228
  • Iou Motorcycle: 0.7637
  • Iou Autorickshaw: 0.8882
  • Iou Truck: 0.8506
  • Iou Curb: 0.6721
  • Iou Fence: 0.4571
  • Iou Billboard: 0.6238
  • Iou Pole: 0.4831
  • Iou Building: 0.7293
  • Iou Vegetation: 0.8792
  • Iou Sky: 0.9707

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

Training results

Training LossEpochStepValidation LossMean IouMean AccuracyOverall AccuracyAccuracy RoadAccuracy ParkingAccuracy SidewalkAccuracy Rail trackAccuracy PersonAccuracy RiderAccuracy MotorcycleAccuracy AutorickshawAccuracy TruckAccuracy CurbAccuracy FenceAccuracy BillboardAccuracy PoleAccuracy BuildingAccuracy VegetationAccuracy SkyIou RoadIou ParkingIou SidewalkIou Rail trackIou PersonIou RiderIou MotorcycleIou AutorickshawIou TruckIou CurbIou FenceIou BillboardIou PoleIou BuildingIou VegetationIou Sky
0.30961.02020.33100.64760.76630.88410.95380.79980.78380.57550.69120.73550.84450.95180.86030.71730.39970.69380.44560.88680.95310.96900.92560.63170.58760.43780.57050.59110.67220.80700.77580.62550.33510.55070.36630.66930.85510.9600
0.27862.04040.33690.65600.79170.87740.89680.85710.80090.50090.72210.81050.82670.91020.92160.82040.60460.69270.52440.86500.93540.97730.88340.55560.59410.42150.60540.63400.69260.84030.77570.62290.36710.56810.41300.69670.86100.9641
0.25413.06060.30130.67960.79300.89580.97240.75790.81320.54240.74830.80200.82350.92960.91070.80670.53090.74150.53590.84200.95020.98060.93360.63840.61940.46230.61850.65050.71020.85340.80990.64460.39830.58390.42360.69750.86410.9660
0.23044.08080.30550.68600.80160.89470.94930.82190.76670.58090.79480.77800.83930.93030.90330.78690.59680.70700.58820.87110.92590.98510.92500.63100.64350.48150.62640.65890.71880.86090.80950.65190.40440.58400.43760.70660.86970.9671
0.2145.010100.31380.68450.79210.89670.95260.84810.84310.48570.75390.75700.87720.94630.88890.72980.45820.74750.58900.86610.94490.98500.92890.64380.63860.43510.63790.65940.71820.85780.82390.64020.38700.59650.45370.69550.86950.9666
0.20296.012120.31230.69140.80100.89880.96970.76120.81110.64530.79880.86210.81120.93770.90280.75670.52110.70470.53900.84940.96380.98220.93390.63780.67460.47540.62710.66600.71430.86820.83050.63440.42640.59470.43350.71680.86140.9679
0.18377.014140.32010.69630.80090.90080.96140.83040.79100.56960.75380.82120.86420.93570.90920.79000.54750.69140.51370.90040.95090.98410.93370.65340.66000.46210.64890.68810.72890.87150.83250.67170.42650.58420.43060.70740.87250.9683
0.19048.016160.30750.69460.80920.89970.96930.75580.77070.69960.78020.83790.85720.94280.89650.77140.54360.74670.63160.82160.93980.98270.93500.63710.63890.49580.64340.66650.72720.86460.82590.65330.41530.59830.46380.70580.87450.9687
0.1669.018180.31270.70180.81100.90300.97250.76340.82580.63320.77660.82860.83360.92740.91010.81990.56390.75240.57880.85690.95060.98280.93420.64270.64290.49860.65630.69660.73590.87240.82620.65800.42840.60590.46540.72320.87380.9687
0.15710.020200.32670.70240.81680.90200.96040.80610.77910.67430.80650.83190.85850.94010.90240.79240.59710.73630.59820.86370.94120.98110.93410.65250.66350.50210.65060.69020.73530.86380.82350.66490.42500.60450.46860.71660.87540.9686
0.154311.022220.32600.70090.81310.90290.97530.74830.84290.63130.79170.83800.83680.94110.89590.79040.58710.75330.59130.85230.95090.98240.93410.63770.63770.49660.65830.69510.73250.86030.82350.66200.43030.60880.47090.72130.87630.9689
0.147412.024240.33940.70540.81790.90280.95850.84390.83030.67320.78000.82310.87550.94530.89850.78960.58140.75450.50470.89940.94520.98420.93510.65930.67070.50760.66320.69950.74270.87370.84030.65950.43710.60680.43170.71510.87520.9693
0.136513.026260.33470.71210.82200.90630.97060.78580.82100.67340.80030.84900.86370.94610.91080.78850.60000.75370.59160.86480.95100.98150.93870.66040.66680.50290.67020.70670.74800.87930.84030.67130.44490.61620.47390.72800.87740.9689
0.130114.028280.34120.71480.81590.90730.97340.77250.80920.63430.79020.82360.86920.94240.90580.80160.57090.74030.62100.86330.95240.98380.93840.65490.67180.50430.66950.71210.75440.88320.84440.67440.45190.61650.48690.72520.87870.9696
0.121615.030300.36230.71370.81870.90710.96860.80450.82660.67000.80500.81770.86010.94660.91720.80570.53250.74870.58870.87700.94760.98300.93900.66540.67070.50550.66970.71210.75630.88130.84680.67020.43110.62100.47560.72600.87950.9699
0.119816.032320.36600.71540.82300.90730.97030.80290.83570.63540.80380.82890.86230.94840.91660.80850.60430.76020.59040.87190.94440.98390.93850.66130.67170.49950.66950.71300.75420.88250.85110.67570.45540.62050.47790.72790.87820.9699
0.126417.034340.36880.70930.81610.90370.96310.78180.79600.70130.81660.82030.86500.94030.90870.78330.55800.72230.59170.87960.95150.97890.93390.64010.67200.49890.66480.70750.75010.88230.84810.66130.44080.61310.47040.71950.87670.9687
0.13718.036360.35830.70790.81310.90480.96990.79000.81680.60480.81070.83550.86090.93400.91440.79440.49690.77380.62190.85780.94310.98400.93660.65360.66580.50100.66720.70360.74780.87920.83520.66000.42370.60800.47890.71740.87820.9695
0.124319.038380.36710.70500.81400.90450.97280.76090.82160.63570.79750.84560.88010.93280.89300.78460.56960.72380.59720.87480.95390.98080.93700.64700.63670.49000.65770.70190.74490.86650.82850.66000.45450.60790.47410.72830.87650.9693
0.113920.040400.37730.71140.81690.90630.97450.77520.80030.63770.79770.85020.86480.94070.90170.82820.56760.73870.60230.86130.94650.98370.93880.65840.65930.50510.66490.71080.75080.87750.83880.66280.45010.61500.47350.72730.87900.9698
0.10721.042420.38940.71560.82080.90750.96980.81610.81830.64710.79600.82170.88520.94570.90720.81570.54330.76910.60850.86230.94320.98300.93990.66850.67030.50960.67520.71440.75420.88160.84730.66700.44460.62100.48150.72630.87910.9701
0.103122.044440.40170.71510.82330.90730.96910.80620.83720.64090.81170.83510.87760.94590.91210.80860.57990.75380.60180.86250.94520.98530.93870.66340.66690.50310.67140.71450.75720.88390.84690.66650.45400.61710.48090.72640.88000.9702
0.099923.046460.41160.71570.82020.90770.97430.78440.83150.64430.79600.82790.88400.95170.89900.81630.57120.73900.60630.86840.94760.98210.93950.66270.66830.51130.67590.71770.75750.87900.84170.66450.45210.61980.48380.72800.87970.9703
0.096324.048480.42640.71530.81770.90740.97540.77810.81630.63710.81270.83060.87850.94770.91280.77850.55630.76050.59150.87200.95290.98270.93890.65550.67800.50420.66890.71790.75900.88420.84670.66650.45060.62480.47720.72440.87840.9703
0.09625.050500.42910.71570.82080.90780.97270.79870.83350.62170.81030.83480.87730.94580.90830.80450.57170.75630.59770.86680.95080.98250.93920.66240.66590.50070.66800.71680.75860.88550.84970.66740.45710.62240.47990.72740.87940.9702
0.092626.052520.43600.71690.82300.90810.97210.79660.83040.62490.81040.83520.87460.95300.90800.81150.58510.77360.59620.86160.94950.98480.93960.66430.66790.50000.67110.72080.76260.88360.84810.66720.45830.62770.48080.72880.87940.9704
0.090327.054540.44530.71660.82260.90770.97260.78530.82580.62820.80040.83740.88240.94570.91910.81910.59070.76000.58910.87710.94700.98200.93920.66020.67050.50290.67620.71910.76190.88670.84720.66570.45590.62530.47760.72770.87970.9702
0.09328.056560.44380.71520.81950.90720.96960.79800.82270.64210.79450.85610.87050.94870.90110.80250.55750.74600.59750.86940.95040.98470.93720.65810.66500.50610.67460.72090.75490.88220.84610.66400.45360.62010.48140.72900.87860.9706
0.088329.058580.45140.71730.82310.90790.97140.79870.81990.64750.82390.83080.87220.94800.91290.80730.57820.76590.59170.86850.95160.98070.93950.66520.67290.50590.67000.72220.76210.88440.84890.66740.46040.62190.47850.72820.87900.9700
0.085730.060600.44780.71750.82250.90810.96910.80460.82080.64610.80210.84010.88440.94560.91170.80350.55460.77300.61060.85960.95100.98360.93870.66480.66880.50430.67300.72090.76290.88740.85090.66930.45180.62290.48670.72930.87880.9703
0.085231.062620.47040.71650.82070.90780.97350.78780.82140.65760.80430.84230.86660.94840.90230.81100.56320.75240.59690.87480.94480.98440.93870.66080.67390.50540.67150.72180.76400.88290.84120.67000.45130.62320.47960.73000.87990.9706
0.081332.064640.46420.71730.82390.90800.96980.79890.82830.64220.80680.83900.88130.94790.90810.81470.57510.77720.59070.86780.94990.98420.93880.66390.67570.50690.67080.72100.76030.88520.84820.66830.45880.62070.47860.72960.87970.9708
0.080733.066660.48110.71770.82180.90790.97100.80030.82160.65090.79690.84680.86850.94830.90880.81220.57170.75430.59950.86620.94770.98430.93870.66270.67630.50870.67440.72370.76320.88470.84760.66560.45740.62020.48060.72950.87990.9707
0.080434.068680.48240.71760.82200.90790.96990.79930.82690.63920.80770.83210.87930.94790.91140.80280.57860.75350.59340.87840.94690.98490.93840.66210.67080.50420.67360.72080.76310.88790.85150.66960.45970.62270.47890.72800.87940.9706
0.076935.070700.49600.71780.82140.90810.96940.80530.82230.64160.79900.84030.87790.94760.91320.81000.55110.75740.60430.87150.94810.98290.93860.66320.67490.50310.67270.72180.76540.88780.84950.67250.44900.62290.48400.72850.87980.9706
0.077636.072720.49180.71680.82190.90770.96970.80430.82470.65360.80090.83460.87870.95170.90770.80460.55430.77300.59200.86800.94730.98470.93870.66310.67320.50200.67550.72270.76510.88440.84830.66690.44850.62350.47880.72770.87950.9704
0.078237.074740.49380.71710.82370.90810.97310.79050.83430.64440.80660.85080.87590.94710.90910.81690.57330.76360.59730.86420.94790.98430.93920.66210.66780.50460.67310.72210.76230.88660.85010.67020.45210.62320.48120.73000.87910.9706
0.075838.076760.51680.71510.82020.90690.97600.74770.82820.66160.80800.84630.87420.94560.91390.81580.54880.76290.59670.86540.94810.98450.93590.63950.66830.50660.67460.72300.76360.88830.84980.66770.44390.62220.47940.72850.87970.9706
0.074539.078780.51000.71730.82030.90810.97250.79480.81940.64880.80330.84170.87190.94570.91110.81400.54650.75580.59420.87230.94710.98500.93880.66220.67520.50430.67320.72390.76490.88690.85010.66830.44720.62250.48040.72950.87930.9708
0.071940.080800.51470.71810.82330.90830.97060.80030.82530.64940.80330.84910.87410.94530.91250.80690.57250.76520.59520.86850.94930.98470.93910.66360.67370.50470.67610.72370.76270.88590.85010.67250.45350.62230.48180.72910.87960.9706
0.073141.082820.52210.71730.82170.90830.97260.79390.82990.63560.81090.84190.86850.94480.91320.80950.55810.76490.60380.86740.94780.98460.93930.66250.67100.50080.67140.72370.76550.88590.84750.66900.45230.62340.48460.72960.87980.9708
0.072342.084840.52450.71760.82300.90840.97110.79900.82750.64170.81000.83960.88310.94650.91260.80830.56160.76490.60500.86400.94970.98390.93960.66450.67090.50260.67070.72240.76240.88640.84930.67090.45310.62370.48450.73040.87940.9706
0.070643.086860.53270.71760.82260.90810.97100.79810.82700.64150.81210.84090.87740.94660.91090.80680.56850.75620.60780.86380.94980.98330.93890.66180.67140.50290.67220.72220.76370.88580.84820.67000.45760.62250.48570.72970.87880.9706
0.070944.088880.54220.71790.82290.90820.97210.79640.82610.63650.80630.84400.87370.94640.91250.80770.58230.76220.59880.86870.94800.98430.93940.66300.67120.50260.67330.72310.76390.88760.85030.67040.45730.62450.48140.72850.87940.9707
0.068345.090900.54110.71820.82230.90830.97220.79470.82540.64320.80200.84360.87300.94570.91370.80800.57350.75920.60240.86880.94830.98360.93900.66130.67340.50180.67290.72370.76360.88710.84980.67180.45860.62450.48370.72920.87960.9707
0.068646.092920.54690.71820.82280.90830.97150.79730.82470.63660.80790.84750.87200.94560.91390.80860.57930.75540.60590.86610.94950.98340.93880.66140.67040.50120.67290.72410.76260.88750.85010.67240.46040.62320.48530.73010.87950.9707
0.067347.094940.55040.71780.82200.90820.97350.78570.82430.63510.80830.84160.87410.94730.91240.81120.57240.76290.60560.86600.94750.98470.93890.65850.67070.50210.67260.72330.76420.88700.85040.67180.45710.62350.48440.72980.87950.9707
0.066748.096960.55450.71790.82210.90830.97220.79250.82400.63890.80650.84200.87380.94610.91320.81140.56710.76280.60230.86710.94910.98430.93890.66130.67110.50240.67290.72290.76340.88760.85010.67220.45640.62320.48440.72970.87940.9707
0.067749.098980.55650.71790.82240.90830.97190.79500.82790.64040.80290.84180.87590.94770.91140.81060.57070.76050.60010.86900.94800.98440.93900.66220.66990.50270.67320.72310.76390.88720.85020.67170.45760.62380.48260.72930.87950.9707
0.067450.0101000.55630.71800.82240.90830.97160.79490.82400.64080.80570.84340.87620.94510.91220.81120.56990.76050.60100.86780.94950.98410.93910.66200.67070.50250.67260.72280.76370.88820.85060.67210.45710.62380.48310.72930.87920.9707

Framework versions

  • Transformers 4.42.4
  • Pytorch 2.3.1+cu121
  • Datasets 2.19.1
  • Tokenizers 0.19.1

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей