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.0916
-- Mean Iou: 0.1657
-- Mean Accuracy: 0.2139
-- Overall Accuracy: 0.7523
-- Accuracy Unlabeled: nan
-- Accuracy Flat-road: 0.7670
-- Accuracy Flat-sidewalk: 0.9044
-- Accuracy Flat-crosswalk: 0.0
-- Accuracy Flat-cyclinglane: 0.7817
-- Accuracy Flat-parkingdriveway: 0.0531
-- Accuracy Flat-railtrack: nan
-- Accuracy Flat-curb: 0.0008
-- Accuracy Human-person: 0.0
-- Accuracy Human-rider: 0.0
-- Accuracy Vehicle-car: 0.8784
-- 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.8946
-- Accuracy Construction-door: 0.0
-- Accuracy Construction-wall: 0.0018
-- 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.9132
-- Accuracy Nature-terrain: 0.8337
-- Accuracy Sky: 0.8152
-- 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.5357
-- Iou Flat-sidewalk: 0.7678
-- Iou Flat-crosswalk: 0.0
-- Iou Flat-cyclinglane: 0.6199
-- Iou Flat-parkingdriveway: 0.0504
-- Iou Flat-railtrack: nan
-- Iou Flat-curb: 0.0008
-- Iou Human-person: 0.0
-- Iou Human-rider: 0.0
-- Iou Vehicle-car: 0.6256
-- 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.5403
-- Iou Construction-door: 0.0
-- Iou Construction-wall: 0.0018
-- 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.7516
-- Iou Nature-terrain: 0.6290
-- Iou Sky: 0.7785
-- 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: 15.0
+- num_epochs: 20.0
 
 ### Training results
 

model.safetensors

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