my-fine-tuned-model
This model is a fine-tuned version of nvidia/segformer-b1-finetuned-ade-512-512 on the segments/sidewalk-semantic dataset. It achieves the following results on the evaluation set:
- Loss: 1.6073
- Mean Iou: 0.1725
- Mean Accuracy: 0.2142
- Overall Accuracy: 0.7712
- Accuracy Unlabeled: nan
- Accuracy Flat-road: 0.8994
- Accuracy Flat-sidewalk: 0.9458
- Accuracy Flat-crosswalk: 0.0
- Accuracy Flat-cyclinglane: 0.5523
- Accuracy Flat-parkingdriveway: 0.0046
- Accuracy Flat-railtrack: nan
- Accuracy Flat-curb: 0.0
- Accuracy Human-person: 0.0
- Accuracy Human-rider: 0.0
- Accuracy Vehicle-car: 0.8918
- 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.8587
- Accuracy Construction-door: 0.0
- Accuracy Construction-wall: 0.0000
- Accuracy Construction-fenceguardrail: 0.0
- Accuracy Construction-bridge: 0.0
- Accuracy Construction-tunnel: nan
- Accuracy Construction-stairs: 0.0
- Accuracy Object-pole: 0.0
- Accuracy Object-trafficsign: 0.0
- Accuracy Object-trafficlight: 0.0
- Accuracy Nature-vegetation: 0.9487
- Accuracy Nature-terrain: 0.6545
- Accuracy Sky: 0.8834
- Accuracy Void-ground: 0.0
- Accuracy Void-dynamic: 0.0
- Accuracy Void-static: 0.0
- Accuracy Void-unclear: 0.0
- Iou Unlabeled: nan
- Iou Flat-road: 0.6288
- Iou Flat-sidewalk: 0.7990
- Iou Flat-crosswalk: 0.0
- Iou Flat-cyclinglane: 0.4898
- Iou Flat-parkingdriveway: 0.0046
- Iou Flat-railtrack: nan
- Iou Flat-curb: 0.0
- Iou Human-person: 0.0
- Iou Human-rider: 0.0
- Iou Vehicle-car: 0.7296
- 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.5826
- Iou Construction-door: 0.0
- Iou Construction-wall: 0.0000
- Iou Construction-fenceguardrail: 0.0
- Iou Construction-bridge: 0.0
- Iou Construction-tunnel: nan
- Iou Construction-stairs: 0.0
- Iou Object-pole: 0.0
- Iou Object-trafficsign: 0.0
- Iou Object-trafficlight: 0.0
- Iou Nature-vegetation: 0.7290
- Iou Nature-terrain: 0.5528
- Iou Sky: 0.8323
- Iou Void-ground: 0.0
- Iou Void-dynamic: 0.0
- Iou Void-static: 0.0
- 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: 6e-05
- train_batch_size: 8
- eval_batch_size: 8
- seed: 42
- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- num_epochs: 2
Training results
Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Unlabeled | Accuracy Flat-road | Accuracy Flat-sidewalk | Accuracy Flat-crosswalk | Accuracy Flat-cyclinglane | Accuracy Flat-parkingdriveway | Accuracy Flat-railtrack | Accuracy Flat-curb | Accuracy Human-person | Accuracy Human-rider | Accuracy Vehicle-car | Accuracy Vehicle-truck | Accuracy Vehicle-bus | Accuracy Vehicle-tramtrain | Accuracy Vehicle-motorcycle | Accuracy Vehicle-bicycle | Accuracy Vehicle-caravan | Accuracy Vehicle-cartrailer | Accuracy Construction-building | Accuracy Construction-door | Accuracy Construction-wall | Accuracy Construction-fenceguardrail | Accuracy Construction-bridge | Accuracy Construction-tunnel | Accuracy Construction-stairs | Accuracy Object-pole | Accuracy Object-trafficsign | Accuracy Object-trafficlight | Accuracy Nature-vegetation | Accuracy Nature-terrain | Accuracy Sky | Accuracy Void-ground | Accuracy Void-dynamic | Accuracy Void-static | Accuracy Void-unclear | Iou Unlabeled | Iou Flat-road | Iou Flat-sidewalk | Iou Flat-crosswalk | Iou Flat-cyclinglane | Iou Flat-parkingdriveway | Iou Flat-railtrack | Iou Flat-curb | Iou Human-person | Iou Human-rider | Iou Vehicle-car | Iou Vehicle-truck | Iou Vehicle-bus | Iou Vehicle-tramtrain | Iou Vehicle-motorcycle | Iou Vehicle-bicycle | Iou Vehicle-caravan | Iou Vehicle-cartrailer | Iou Construction-building | Iou Construction-door | Iou Construction-wall | Iou Construction-fenceguardrail | Iou Construction-bridge | Iou Construction-tunnel | Iou Construction-stairs | Iou Object-pole | Iou Object-trafficsign | Iou Object-trafficlight | Iou Nature-vegetation | Iou Nature-terrain | Iou Sky | Iou Void-ground | Iou Void-dynamic | Iou Void-static | Iou Void-unclear |
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
2.9867 | 0.2 | 20 | 2.7695 | 0.0608 | 0.1215 | 0.5176 | nan | 0.1535 | 0.9545 | 0.0 | 0.4276 | 0.0007 | nan | 0.0011 | 0.0313 | 0.0316 | 0.4054 | 0.0 | 0.0 | nan | 0.0 | 0.0002 | 0.0123 | 0.0 | 0.0474 | 0.4432 | 0.0 | 0.2061 | 0.0 | nan | 0.0126 | 0.0026 | 0.0 | 0.0834 | 0.9202 | 0.0051 | 0.0216 | 0.0 | 0.0054 | 0.0016 | 0.0 | 0.0 | 0.1444 | 0.6958 | 0.0 | 0.1556 | 0.0006 | 0.0 | 0.0010 | 0.0307 | 0.0041 | 0.4029 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0002 | 0.0019 | 0.0 | 0.0469 | 0.0110 | 0.0 | 0.0341 | 0.0 | 0.0 | 0.0023 | 0.0022 | 0.0 | 0.0004 | 0.5623 | 0.0050 | 0.0191 | 0.0 | 0.0045 | 0.0016 | 0.0 |
2.5329 | 0.4 | 40 | 2.4300 | 0.1145 | 0.1755 | 0.6737 | nan | 0.6841 | 0.9241 | 0.0 | 0.4931 | 0.0019 | nan | 0.0008 | 0.0042 | 0.0 | 0.8445 | 0.0003 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.4625 | 0.0759 | 0.0002 | 0.1086 | 0.0 | nan | 0.0307 | 0.0 | 0.0 | 0.0 | 0.9880 | 0.0139 | 0.8085 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.5074 | 0.7430 | 0.0 | 0.3076 | 0.0019 | 0.0 | 0.0008 | 0.0042 | 0.0 | 0.7114 | 0.0002 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.4046 | 0.0092 | 0.0002 | 0.0473 | 0.0 | 0.0 | 0.0080 | 0.0 | 0.0 | 0.0 | 0.5680 | 0.0137 | 0.6814 | 0.0 | 0.0 | 0.0 | 0.0 |
2.3589 | 0.6 | 60 | 2.1477 | 0.1380 | 0.1895 | 0.7251 | nan | 0.7858 | 0.9356 | 0.0 | 0.5233 | 0.0064 | nan | 0.0010 | 0.0005 | 0.0 | 0.8708 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.7632 | 0.0005 | 0.0000 | 0.0098 | 0.0 | nan | 0.0008 | 0.0 | 0.0 | 0.0 | 0.9782 | 0.1240 | 0.8748 | 0.0 | 0.0 | 0.0 | 0.0 | nan | 0.5672 | 0.7591 | 0.0 | 0.3992 | 0.0064 | 0.0 | 0.0010 | 0.0005 | 0.0 | 0.7148 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.5655 | 0.0004 | 0.0000 | 0.0086 | 0.0 | 0.0 | 0.0006 | 0.0 | 0.0 | 0.0 | 0.6252 | 0.1144 | 0.7905 | 0.0 | 0.0 | 0.0 | 0.0 |
2.0019 | 0.8 | 80 | 1.9809 | 0.1509 | 0.1995 | 0.7427 | nan | 0.8556 | 0.9209 | 0.0 | 0.5549 | 0.0048 | nan | 0.0000 | 0.0 | 0.0 | 0.8962 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.8344 | 0.0 | 0.0 | 0.0002 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.9686 | 0.2662 | 0.8840 | 0.0 | 0.0 | 0.0 | 0.0 | nan | 0.5881 | 0.7793 | 0.0 | 0.4554 | 0.0048 | nan | 0.0000 | 0.0 | 0.0 | 0.7039 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.5763 | 0.0 | 0.0 | 0.0002 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.6579 | 0.2418 | 0.8194 | 0.0 | 0.0 | 0.0 | 0.0 |
1.9575 | 1.0 | 100 | 1.8123 | 0.1525 | 0.2015 | 0.7465 | nan | 0.8756 | 0.9349 | 0.0 | 0.4741 | 0.0042 | nan | 0.0 | 0.0 | 0.0 | 0.9205 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.7650 | 0.0 | 0.0 | 0.0002 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.9646 | 0.4191 | 0.8883 | 0.0 | 0.0 | 0.0 | 0.0 | nan | 0.5946 | 0.7767 | 0.0 | 0.4219 | 0.0042 | nan | 0.0 | 0.0 | 0.0 | 0.6575 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.5673 | 0.0 | 0.0 | 0.0002 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.6760 | 0.3757 | 0.8070 | 0.0 | 0.0 | 0.0 | 0.0 |
1.9557 | 1.2 | 120 | 1.7501 | 0.1667 | 0.2075 | 0.7580 | nan | 0.8464 | 0.9461 | 0.0 | 0.5353 | 0.0052 | nan | 0.0 | 0.0 | 0.0 | 0.8843 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.8335 | 0.0 | 0.0 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.9601 | 0.5422 | 0.8780 | 0.0 | 0.0 | 0.0 | 0.0 | nan | 0.6091 | 0.7752 | 0.0 | 0.4861 | 0.0052 | nan | 0.0 | 0.0 | 0.0 | 0.7197 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.5675 | 0.0 | 0.0 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.7028 | 0.4764 | 0.8254 | 0.0 | 0.0 | 0.0 | 0.0 |
2.0378 | 1.4 | 140 | 1.6594 | 0.1692 | 0.2095 | 0.7627 | nan | 0.8560 | 0.9518 | 0.0 | 0.5336 | 0.0036 | nan | 0.0 | 0.0 | 0.0 | 0.8866 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.8332 | 0.0 | 0.0000 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.9562 | 0.6099 | 0.8640 | 0.0 | 0.0 | 0.0 | 0.0 | nan | 0.6235 | 0.7756 | 0.0 | 0.4882 | 0.0036 | nan | 0.0 | 0.0 | 0.0 | 0.7252 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.5733 | 0.0 | 0.0000 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.7120 | 0.5260 | 0.8185 | 0.0 | 0.0 | 0.0 | 0.0 |
1.8032 | 1.6 | 160 | 1.6499 | 0.1715 | 0.2141 | 0.7687 | nan | 0.8843 | 0.9420 | 0.0 | 0.5999 | 0.0053 | nan | 0.0 | 0.0 | 0.0 | 0.9009 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.8384 | 0.0 | 0.0000 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.9560 | 0.6342 | 0.8765 | 0.0 | 0.0 | 0.0 | 0.0 | nan | 0.6322 | 0.7961 | 0.0 | 0.5077 | 0.0053 | nan | 0.0 | 0.0 | 0.0 | 0.7154 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.5802 | 0.0 | 0.0000 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.7138 | 0.5379 | 0.8264 | 0.0 | 0.0 | 0.0 | 0.0 |
1.7118 | 1.8 | 180 | 1.6190 | 0.1711 | 0.2138 | 0.7682 | nan | 0.9177 | 0.9317 | 0.0 | 0.5441 | 0.0058 | nan | 0.0 | 0.0 | 0.0 | 0.8963 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.8505 | 0.0 | 0.0000 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.9522 | 0.6567 | 0.8742 | 0.0 | 0.0 | 0.0 | 0.0 | nan | 0.5990 | 0.8121 | 0.0 | 0.4792 | 0.0058 | nan | 0.0 | 0.0 | 0.0 | 0.7249 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.5825 | 0.0 | 0.0000 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.7221 | 0.5512 | 0.8274 | 0.0 | 0.0 | 0.0 | 0.0 |
1.5752 | 2.0 | 200 | 1.6073 | 0.1725 | 0.2142 | 0.7712 | nan | 0.8994 | 0.9458 | 0.0 | 0.5523 | 0.0046 | nan | 0.0 | 0.0 | 0.0 | 0.8918 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.8587 | 0.0 | 0.0000 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.9487 | 0.6545 | 0.8834 | 0.0 | 0.0 | 0.0 | 0.0 | nan | 0.6288 | 0.7990 | 0.0 | 0.4898 | 0.0046 | nan | 0.0 | 0.0 | 0.0 | 0.7296 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.5826 | 0.0 | 0.0000 | 0.0 | 0.0 | nan | 0.0 | 0.0 | 0.0 | 0.0 | 0.7290 | 0.5528 | 0.8323 | 0.0 | 0.0 | 0.0 | 0.0 |
Framework versions
- Transformers 4.48.0
- Pytorch 2.1.1+cu118
- Datasets 3.2.0
- Tokenizers 0.21.0
- Downloads last month
- 13
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
the model is not deployed on the HF Inference API.
Model tree for Totsanat/my-fine-tuned-model
Base model
nvidia/segformer-b1-finetuned-ade-512-512