--- library_name: transformers license: other base_model: nvidia/mit-b0 tags: - vision - image-segmentation - generated_from_trainer model-index: - name: segformer-b0-finetuned-segments-sidewalk-oct-22 results: [] --- # segformer-b0-finetuned-segments-sidewalk-oct-22 This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the segments/sidewalk-semantic dataset. It achieves the following results on the evaluation set: - eval_loss: 0.9975 - eval_mean_iou: 0.1753 - eval_mean_accuracy: 0.2215 - eval_overall_accuracy: 0.7617 - eval_accuracy_unlabeled: nan - eval_accuracy_flat-road: 0.8209 - eval_accuracy_flat-sidewalk: 0.9513 - eval_accuracy_flat-crosswalk: 0.0 - eval_accuracy_flat-cyclinglane: 0.5766 - eval_accuracy_flat-parkingdriveway: 0.0616 - eval_accuracy_flat-railtrack: nan - eval_accuracy_flat-curb: 0.0067 - eval_accuracy_human-person: 0.0 - eval_accuracy_human-rider: 0.0 - eval_accuracy_vehicle-car: 0.8717 - eval_accuracy_vehicle-truck: 0.0 - eval_accuracy_vehicle-bus: 0.0 - eval_accuracy_vehicle-tramtrain: nan - eval_accuracy_vehicle-motorcycle: 0.0 - eval_accuracy_vehicle-bicycle: 0.0 - eval_accuracy_vehicle-caravan: 0.0 - eval_accuracy_vehicle-cartrailer: 0.0 - eval_accuracy_construction-building: 0.8790 - eval_accuracy_construction-door: 0.0 - eval_accuracy_construction-wall: 0.0087 - eval_accuracy_construction-fenceguardrail: 0.0 - eval_accuracy_construction-bridge: 0.0 - eval_accuracy_construction-tunnel: nan - eval_accuracy_construction-stairs: 0.0 - eval_accuracy_object-pole: 0.0 - eval_accuracy_object-trafficsign: 0.0 - eval_accuracy_object-trafficlight: 0.0 - eval_accuracy_nature-vegetation: 0.9327 - eval_accuracy_nature-terrain: 0.8573 - eval_accuracy_sky: 0.9003 - eval_accuracy_void-ground: 0.0 - eval_accuracy_void-dynamic: 0.0 - eval_accuracy_void-static: 0.0 - eval_accuracy_void-unclear: 0.0 - eval_iou_unlabeled: nan - eval_iou_flat-road: 0.5348 - eval_iou_flat-sidewalk: 0.7883 - eval_iou_flat-crosswalk: 0.0 - eval_iou_flat-cyclinglane: 0.5319 - eval_iou_flat-parkingdriveway: 0.0574 - eval_iou_flat-railtrack: nan - eval_iou_flat-curb: 0.0066 - eval_iou_human-person: 0.0 - eval_iou_human-rider: 0.0 - eval_iou_vehicle-car: 0.6518 - eval_iou_vehicle-truck: 0.0 - eval_iou_vehicle-bus: 0.0 - eval_iou_vehicle-tramtrain: nan - eval_iou_vehicle-motorcycle: 0.0 - eval_iou_vehicle-bicycle: 0.0 - eval_iou_vehicle-caravan: 0.0 - eval_iou_vehicle-cartrailer: 0.0 - eval_iou_construction-building: 0.5480 - eval_iou_construction-door: 0.0 - eval_iou_construction-wall: 0.0086 - eval_iou_construction-fenceguardrail: 0.0 - eval_iou_construction-bridge: 0.0 - eval_iou_construction-tunnel: nan - eval_iou_construction-stairs: 0.0 - eval_iou_object-pole: 0.0 - eval_iou_object-trafficsign: 0.0 - eval_iou_object-trafficlight: 0.0 - eval_iou_nature-vegetation: 0.7901 - eval_iou_nature-terrain: 0.7000 - eval_iou_sky: 0.8158 - eval_iou_void-ground: 0.0 - eval_iou_void-dynamic: 0.0 - eval_iou_void-static: 0.0 - eval_iou_void-unclear: 0.0 - eval_runtime: 33.7434 - eval_samples_per_second: 5.927 - eval_steps_per_second: 2.964 - epoch: 1.0 - step: 400 ## 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: 50 ### Framework versions - Transformers 4.44.2 - Pytorch 2.4.1+cu121 - Datasets 3.1.0 - Tokenizers 0.19.1