nielsr's picture
nielsr HF staff
update model card README.md
11ec7bf
|
raw
history blame
3.4 kB
metadata
license: apache-2.0
tags:
  - image-segmentation
  - vision
  - generated_from_trainer
model-index:
  - name: segformer-finetuned-sidewalk-50-epochs
    results: []

segformer-finetuned-sidewalk-50-epochs

This model is a fine-tuned version of nvidia/mit-b0 on the segments/sidewalk-semantic dataset. It achieves the following results on the evaluation set:

  • Loss: 0.7794
  • Mean Iou: 0.2323
  • Mean Accuracy: 0.2967
  • Overall Accuracy: 0.7880
  • Accuracy Unlabeled: nan
  • Accuracy Flat-road: 0.7710
  • Accuracy Flat-sidewalk: 0.9216
  • Accuracy Flat-crosswalk: 0.2824
  • Accuracy Flat-cyclinglane: 0.7632
  • Accuracy Flat-parkingdriveway: 0.2202
  • Accuracy Flat-railtrack: nan
  • Accuracy Flat-curb: 0.5054
  • Accuracy Human-person: 0.5200
  • Accuracy Human-rider: 0.0
  • Accuracy Vehicle-car: 0.9194
  • Accuracy Vehicle-truck: 0.0
  • Accuracy Vehicle-bus: 0.0
  • Accuracy Vehicle-tramtrain: nan
  • Accuracy Vehicle-motorcycle: 0.0
  • Accuracy Vehicle-bicycle: 0.0
  • Accuracy Vehicle-caravan: 0.0
  • Accuracy Vehicle-cartrailer: 0.0
  • Accuracy Construction-building: 0.8614
  • Accuracy Construction-door: 0.0
  • Accuracy Construction-wall: 0.4325
  • Accuracy Construction-fenceguardrail: 0.0652
  • Accuracy Construction-bridge: 0.0
  • Accuracy Construction-tunnel: nan
  • Accuracy Construction-stairs: 0.0
  • Accuracy Object-pole: 0.2563
  • Accuracy Object-trafficsign: 0.0
  • Accuracy Object-trafficlight: 0.0
  • Accuracy Nature-vegetation: 0.9089
  • Accuracy Nature-terrain: 0.8215
  • Accuracy Sky: 0.8732
  • Accuracy Void-ground: 0.0
  • Accuracy Void-dynamic: 0.0
  • Accuracy Void-static: 0.0739
  • Accuracy Void-unclear: 0.0
  • Iou Unlabeled: nan
  • Iou Flat-road: 0.5975
  • Iou Flat-sidewalk: 0.8299
  • Iou Flat-crosswalk: 0.2624
  • Iou Flat-cyclinglane: 0.5689
  • Iou Flat-parkingdriveway: 0.1733
  • Iou Flat-railtrack: nan
  • Iou Flat-curb: 0.3084
  • Iou Human-person: 0.3708
  • Iou Human-rider: 0.0
  • Iou Vehicle-car: 0.6164
  • Iou Vehicle-truck: 0.0
  • Iou Vehicle-bus: 0.0
  • Iou Vehicle-tramtrain: nan
  • Iou Vehicle-motorcycle: 0.0
  • Iou Vehicle-bicycle: 0.0
  • Iou Vehicle-caravan: 0.0
  • Iou Vehicle-cartrailer: 0.0
  • Iou Construction-building: 0.6305
  • Iou Construction-door: 0.0
  • Iou Construction-wall: 0.3016
  • Iou Construction-fenceguardrail: 0.0580
  • Iou Construction-bridge: 0.0
  • Iou Construction-tunnel: nan
  • Iou Construction-stairs: 0.0
  • Iou Object-pole: 0.1310
  • Iou Object-trafficsign: 0.0
  • Iou Object-trafficlight: 0.0
  • Iou Nature-vegetation: 0.7954
  • Iou Nature-terrain: 0.6628
  • Iou Sky: 0.8233
  • Iou Void-ground: 0.0
  • Iou Void-dynamic: 0.0
  • Iou Void-static: 0.0705
  • Iou Void-unclear: 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: 16
  • eval_batch_size: 16
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 50.0

Training results

Framework versions

  • Transformers 4.19.0.dev0
  • Pytorch 1.11.0+cu113
  • Datasets 2.0.0
  • Tokenizers 0.11.6