Model save

README.md

@@ -2,8 +2,6 @@
 license: other
 base_model: nvidia/mit-b0
 tags:
-- image-segmentation
-- vision
 - generated_from_trainer
 model-index:
 - name: Segments
@@ -15,82 +13,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # Segments
 
-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:
-- Loss: 1.0296
-- Mean Iou: 0.1549
-- Mean Accuracy: 0.2005
-- Overall Accuracy: 0.7433
-- Accuracy Unlabeled: nan
-- Accuracy Flat-road: 0.7243
-- Accuracy Flat-sidewalk: 0.9040
-- Accuracy Flat-crosswalk: 0.0017
-- Accuracy Flat-cyclinglane: 0.6786
-- Accuracy Flat-parkingdriveway: 0.1296
-- Accuracy Flat-railtrack: 0.0
-- Accuracy Flat-curb: 0.0053
-- Accuracy Human-person: 0.0
-- Accuracy Human-rider: 0.0
-- Accuracy Vehicle-car: 0.8724
-- Accuracy Vehicle-truck: 0.0
-- Accuracy Vehicle-bus: 0.0
-- Accuracy Vehicle-tramtrain: 0.0
-- Accuracy Vehicle-motorcycle: 0.0
-- Accuracy Vehicle-bicycle: 0.0
-- Accuracy Vehicle-caravan: 0.0
-- Accuracy Vehicle-cartrailer: 0.0
-- Accuracy Construction-building: 0.9045
-- Accuracy Construction-door: 0.0
-- Accuracy Construction-wall: 0.0126
-- Accuracy Construction-fenceguardrail: 0.0
-- Accuracy Construction-bridge: 0.0
-- Accuracy Construction-tunnel: 0.0
-- Accuracy Construction-stairs: 0.0
-- Accuracy Object-pole: 0.0128
-- Accuracy Object-trafficsign: 0.0
-- Accuracy Object-trafficlight: 0.0
-- Accuracy Nature-vegetation: 0.9126
-- Accuracy Nature-terrain: 0.7863
-- Accuracy Sky: 0.8733
-- 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.5257
-- Iou Flat-sidewalk: 0.7726
-- Iou Flat-crosswalk: 0.0017
-- Iou Flat-cyclinglane: 0.5195
-- Iou Flat-parkingdriveway: 0.0990
-- Iou Flat-railtrack: 0.0
-- Iou Flat-curb: 0.0051
-- Iou Human-person: 0.0
-- Iou Human-rider: 0.0
-- Iou Vehicle-car: 0.6387
-- Iou Vehicle-truck: 0.0
-- Iou Vehicle-bus: 0.0
-- Iou Vehicle-tramtrain: 0.0
-- Iou Vehicle-motorcycle: 0.0
-- Iou Vehicle-bicycle: 0.0
-- Iou Vehicle-caravan: 0.0
-- Iou Vehicle-cartrailer: 0.0
-- Iou Construction-building: 0.5134
-- Iou Construction-door: 0.0
-- Iou Construction-wall: 0.0123
-- Iou Construction-fenceguardrail: 0.0
-- Iou Construction-bridge: 0.0
-- Iou Construction-tunnel: 0.0
-- Iou Construction-stairs: 0.0
-- Iou Object-pole: 0.0125
-- Iou Object-trafficsign: 0.0
-- Iou Object-trafficlight: 0.0
-- Iou Nature-vegetation: 0.7452
-- Iou Nature-terrain: 0.6074
-- Iou Sky: 0.8149
-- Iou Void-ground: 0.0
-- Iou Void-dynamic: 0.0
-- Iou Void-static: 0.0
-- Iou Void-unclear: 0.0
+This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
 
 ## Model description
 
@@ -117,7 +40,7 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 16
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 20.0
+- num_epochs: 1.0
 
 ### Training results
 

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c1991f97e9f07cefc8de8ccafb51a37a0bb6099018615724a0da76dc677245a5
+oid sha256:14b7ad548fe2ad065277ffb0a7983086d151359d58d1432993c262f8d4de573d
 size 14918708

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dabb161a7c89471ed388d64d037bae3f7ed86fbe43de6c776c0c07dd7ec888bf
+oid sha256:3964be0fb553241843b2f7a9afdff4453bfa4129425cd7bdcc119d5400430f91
 size 4027