/ Модель

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

Лицензия
Своя лицензия
Архитектура
segformer · dense
Опубликована
7 апр 2023

segformer-finetuned-lane-10k-steps

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

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

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

Бенчмарки

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

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

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

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

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

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

Efferbach/segformer-finetuned-lane-10k-steps — сегментационная модель на архитектуре SegFormer, дообученная на наборе Efferbach/lane_master на базе nvidia/segformer-b0-finetuned-cityscapes-512-1024. Она предназначена для семантической сегментации изображений дороги с выделением фона и левой и правой полос. В оценке на тестовой выборке указаны mean IoU 0,4899 и mean accuracy 0,7371; для классов Left и Right приведены IoU 0,7371 и 0,7325 соответственно.

Модель распространяется в формате PyTorch. Лицензия указана как other, поэтому конкретные условия использования следует сверить с материалами репозитория.

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

  • Выделение левой и правой полос на изображении дороги
  • Сегментация разметки полос для систем помощи водителю
  • Подготовка масок полос движения для анализа дорожных сцен

Ограничения

В README не описаны ограничения и условия оценки; на тестовой выборке указаны mean IoU 0,4899 и mean accuracy 0,7371. Лицензия обозначена как other, поэтому условия применения требуют проверки в репозитории.

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

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

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

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

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

Открыть на Hugging Face

segformer-finetuned-lane-10k-steps

This model is a fine-tuned version of nvidia/segformer-b0-finetuned-cityscapes-512-1024 on the Efferbach/lane_master dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0365
  • Mean Iou: 0.4899
  • Mean Accuracy: 0.7371
  • Overall Accuracy: 0.7371
  • Accuracy Background: nan
  • Accuracy Left: 0.7394
  • Accuracy Right: 0.7348
  • Iou Background: 0.0
  • Iou Left: 0.7371
  • Iou Right: 0.7325

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: 8
  • eval_batch_size: 8
  • seed: 1337
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: polynomial
  • training_steps: 10000

Training results

Training LossEpochStepValidation LossMean IouMean AccuracyOverall AccuracyAccuracy BackgroundAccuracy LeftAccuracy RightIou BackgroundIou LeftIou Right
0.07921.03080.07140.01480.02290.0225nan0.03730.00850.00.03620.0083
0.04372.06160.05020.16870.27750.2784nan0.24920.30580.00.23430.2718
0.03263.09240.04450.26140.44410.4479nan0.31340.57480.00.31000.4742
0.02244.012320.03700.40480.60980.6100nan0.60430.61530.00.60310.6113
0.01845.015400.03460.38200.58580.5870nan0.54210.62950.00.54000.6060
0.01596.018480.03190.43670.65670.6573nan0.63430.67910.00.63410.6760
0.01397.021560.03170.45550.68550.6860nan0.66910.70190.00.66800.6986
0.01298.024640.03210.43480.65330.6535nan0.64790.65880.00.64740.6571
0.01229.027720.02750.45410.68270.6830nan0.67100.69430.00.66970.6927
0.011110.030800.03050.46090.69280.6927nan0.69690.68870.00.69630.6865
0.01111.033880.02860.46460.69880.6991nan0.68900.70870.00.68830.7055
0.010312.036960.02980.46930.70580.7062nan0.69390.71770.00.69320.7148
0.009713.040040.02930.47170.70900.7087nan0.71840.69960.00.71760.6975
0.009314.043120.03300.45370.68350.6836nan0.67750.68940.00.67680.6843
0.00915.046200.03310.48040.72260.7226nan0.71940.72570.00.71780.7234
0.008816.049280.03150.48900.73550.7357nan0.72750.74350.00.72590.7411
0.008617.052360.03380.48130.72340.7234nan0.72240.72430.00.72160.7223
0.008518.055440.03480.47430.71290.7126nan0.72250.70330.00.72170.7012
0.008319.058520.03570.48120.72450.7244nan0.72810.72100.00.72540.7183
0.008120.061600.03340.48290.72710.7269nan0.73370.72050.00.73050.7182
0.007921.064680.03590.47730.71770.7177nan0.71840.71700.00.71740.7146
0.007722.067760.03510.48740.73320.7329nan0.74400.72230.00.74320.7190
0.007523.070840.03440.48550.72960.7292nan0.74370.71560.00.74250.7141
0.007724.073920.03620.47990.72160.7216nan0.72360.71960.00.72230.7174
0.007125.077000.03910.47750.71790.7180nan0.71730.71860.00.71610.7163
0.007726.080080.03390.48950.73670.7366nan0.74050.73290.00.73880.7297
0.006927.083160.03440.48580.73050.7305nan0.72910.73180.00.72780.7297
0.006928.086240.03610.48440.72830.7282nan0.73240.72430.00.73090.7221
0.00729.089320.03710.48370.72730.7270nan0.73600.71860.00.73450.7166
0.00730.092400.03660.48540.73050.7303nan0.73790.72310.00.73530.7208
0.006731.095480.03670.48660.73220.7321nan0.73570.72860.00.73350.7263
0.006832.098560.03640.48830.73480.7347nan0.73770.73180.00.73550.7295
0.006732.47100000.03650.48990.73710.7371nan0.73940.73480.00.73710.7325

Framework versions

  • Transformers 4.28.0.dev0
  • Pytorch 2.0.0+cu118
  • Datasets 2.11.0
  • Tokenizers 0.13.3

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей