Hugging Face's logo

diegola123
/
mit-b0-finetuned-sidewalks

Transformers
TensorFlow
segformer
generated_from_keras_callback
Inference Endpoints
Model card Files Community
diegola123 commited on
Commit
95c231e
•
1 Parent(s): 3a88645

Training in progress epoch 1

Browse files
Files changed (2) hide show
  1. README.md +41 -40
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -15,82 +15,82 @@ probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Train Loss: 1.3626
19
- - Validation Loss: 0.9835
20
- - Validation Mean Iou: 0.1522
21
- - Validation Mean Accuracy: 0.2097
22
- - Validation Overall Accuracy: 0.7080
23
  - Validation Accuracy Unlabeled: 0.0
24
- - Validation Accuracy Flat-road: 0.7886
25
- - Validation Accuracy Flat-sidewalk: 0.8146
26
- - Validation Accuracy Flat-crosswalk: 0.1217
27
- - Validation Accuracy Flat-cyclinglane: 0.6821
28
- - Validation Accuracy Flat-parkingdriveway: 0.0153
29
  - Validation Accuracy Flat-railtrack: nan
30
- - Validation Accuracy Flat-curb: 0.0673
31
- - Validation Accuracy Human-person: 0.0
32
  - Validation Accuracy Human-rider: 0.0
33
- - Validation Accuracy Vehicle-car: 0.7948
34
  - Validation Accuracy Vehicle-truck: 0.0
35
  - Validation Accuracy Vehicle-bus: 0.0
36
  - Validation Accuracy Vehicle-tramtrain: 0.0
37
  - Validation Accuracy Vehicle-motorcycle: 0.0
38
- - Validation Accuracy Vehicle-bicycle: 0.0
39
  - Validation Accuracy Vehicle-caravan: 0.0
40
  - Validation Accuracy Vehicle-cartrailer: 0.0
41
- - Validation Accuracy Construction-building: 0.7900
42
  - Validation Accuracy Construction-door: 0.0
43
- - Validation Accuracy Construction-wall: 0.0076
44
- - Validation Accuracy Construction-fenceguardrail: 0.0344
45
  - Validation Accuracy Construction-bridge: nan
46
  - Validation Accuracy Construction-tunnel: nan
47
  - Validation Accuracy Construction-stairs: 0.0
48
- - Validation Accuracy Object-pole: 0.0313
49
  - Validation Accuracy Object-trafficsign: 0.0
50
  - Validation Accuracy Object-trafficlight: 0.0
51
- - Validation Accuracy Nature-vegetation: 0.8325
52
- - Validation Accuracy Nature-terrain: 0.7788
53
- - Validation Accuracy Sky: 0.9476
54
  - Validation Accuracy Void-ground: 0.0
55
  - Validation Accuracy Void-dynamic: 0.0
56
- - Validation Accuracy Void-static: 0.0046
57
  - Validation Accuracy Void-unclear: 0.0
58
  - Validation Iou Unlabeled: 0.0
59
- - Validation Iou Flat-road: 0.5020
60
- - Validation Iou Flat-sidewalk: 0.7070
61
- - Validation Iou Flat-crosswalk: 0.1182
62
- - Validation Iou Flat-cyclinglane: 0.4076
63
- - Validation Iou Flat-parkingdriveway: 0.0149
64
  - Validation Iou Flat-railtrack: nan
65
- - Validation Iou Flat-curb: 0.0590
66
- - Validation Iou Human-person: 0.0
67
  - Validation Iou Human-rider: 0.0
68
- - Validation Iou Vehicle-car: 0.4022
69
  - Validation Iou Vehicle-truck: 0.0
70
  - Validation Iou Vehicle-bus: 0.0
71
  - Validation Iou Vehicle-tramtrain: 0.0
72
  - Validation Iou Vehicle-motorcycle: 0.0
73
- - Validation Iou Vehicle-bicycle: 0.0
74
  - Validation Iou Vehicle-caravan: 0.0
75
  - Validation Iou Vehicle-cartrailer: 0.0
76
- - Validation Iou Construction-building: 0.5052
77
  - Validation Iou Construction-door: 0.0
78
- - Validation Iou Construction-wall: 0.0076
79
- - Validation Iou Construction-fenceguardrail: 0.0301
80
  - Validation Iou Construction-bridge: nan
81
  - Validation Iou Construction-tunnel: nan
82
  - Validation Iou Construction-stairs: 0.0
83
- - Validation Iou Object-pole: 0.0302
84
  - Validation Iou Object-trafficsign: 0.0
85
  - Validation Iou Object-trafficlight: 0.0
86
- - Validation Iou Nature-vegetation: 0.7228
87
- - Validation Iou Nature-terrain: 0.5421
88
- - Validation Iou Sky: 0.8162
89
  - Validation Iou Void-ground: 0.0
90
  - Validation Iou Void-dynamic: 0.0
91
- - Validation Iou Void-static: 0.0045
92
  - Validation Iou Void-unclear: 0.0
93
- - Epoch: 0
94
 
95
  ## Model description
96
 
@@ -117,6 +117,7 @@ The following hyperparameters were used during training:
117
  | Train Loss | Validation Loss | Validation Mean Iou | Validation Mean Accuracy | Validation Overall Accuracy | Validation Accuracy Unlabeled | Validation Accuracy Flat-road | Validation Accuracy Flat-sidewalk | Validation Accuracy Flat-crosswalk | Validation Accuracy Flat-cyclinglane | Validation Accuracy Flat-parkingdriveway | Validation Accuracy Flat-railtrack | Validation Accuracy Flat-curb | Validation Accuracy Human-person | Validation Accuracy Human-rider | Validation Accuracy Vehicle-car | Validation Accuracy Vehicle-truck | Validation Accuracy Vehicle-bus | Validation Accuracy Vehicle-tramtrain | Validation Accuracy Vehicle-motorcycle | Validation Accuracy Vehicle-bicycle | Validation Accuracy Vehicle-caravan | Validation Accuracy Vehicle-cartrailer | Validation Accuracy Construction-building | Validation Accuracy Construction-door | Validation Accuracy Construction-wall | Validation Accuracy Construction-fenceguardrail | Validation Accuracy Construction-bridge | Validation Accuracy Construction-tunnel | Validation Accuracy Construction-stairs | Validation Accuracy Object-pole | Validation Accuracy Object-trafficsign | Validation Accuracy Object-trafficlight | Validation Accuracy Nature-vegetation | Validation Accuracy Nature-terrain | Validation Accuracy Sky | Validation Accuracy Void-ground | Validation Accuracy Void-dynamic | Validation Accuracy Void-static | Validation Accuracy Void-unclear | Validation Iou Unlabeled | Validation Iou Flat-road | Validation Iou Flat-sidewalk | Validation Iou Flat-crosswalk | Validation Iou Flat-cyclinglane | Validation Iou Flat-parkingdriveway | Validation Iou Flat-railtrack | Validation Iou Flat-curb | Validation Iou Human-person | Validation Iou Human-rider | Validation Iou Vehicle-car | Validation Iou Vehicle-truck | Validation Iou Vehicle-bus | Validation Iou Vehicle-tramtrain | Validation Iou Vehicle-motorcycle | Validation Iou Vehicle-bicycle | Validation Iou Vehicle-caravan | Validation Iou Vehicle-cartrailer | Validation Iou Construction-building | Validation Iou Construction-door | Validation Iou Construction-wall | Validation Iou Construction-fenceguardrail | Validation Iou Construction-bridge | Validation Iou Construction-tunnel | Validation Iou Construction-stairs | Validation Iou Object-pole | Validation Iou Object-trafficsign | Validation Iou Object-trafficlight | Validation Iou Nature-vegetation | Validation Iou Nature-terrain | Validation Iou Sky | Validation Iou Void-ground | Validation Iou Void-dynamic | Validation Iou Void-static | Validation Iou Void-unclear | Epoch |
118
  |:----------:|:---------------:|:-------------------:|:------------------------:|:---------------------------:|:-----------------------------:|:-----------------------------:|:---------------------------------:|:----------------------------------:|:------------------------------------:|:----------------------------------------:|:----------------------------------:|:-----------------------------:|:--------------------------------:|:-------------------------------:|:-------------------------------:|:---------------------------------:|:-------------------------------:|:-------------------------------------:|:--------------------------------------:|:-----------------------------------:|:-----------------------------------:|:--------------------------------------:|:-----------------------------------------:|:-------------------------------------:|:-------------------------------------:|:-----------------------------------------------:|:---------------------------------------:|:---------------------------------------:|:---------------------------------------:|:-------------------------------:|:--------------------------------------:|:---------------------------------------:|:-------------------------------------:|:----------------------------------:|:-----------------------:|:-------------------------------:|:--------------------------------:|:-------------------------------:|:--------------------------------:|:------------------------:|:------------------------:|:----------------------------:|:-----------------------------:|:-------------------------------:|:-----------------------------------:|:-----------------------------:|:------------------------:|:---------------------------:|:--------------------------:|:--------------------------:|:----------------------------:|:--------------------------:|:--------------------------------:|:---------------------------------:|:------------------------------:|:------------------------------:|:---------------------------------:|:------------------------------------:|:--------------------------------:|:--------------------------------:|:------------------------------------------:|:----------------------------------:|:----------------------------------:|:----------------------------------:|:--------------------------:|:---------------------------------:|:----------------------------------:|:--------------------------------:|:-----------------------------:|:------------------:|:--------------------------:|:---------------------------:|:--------------------------:|:---------------------------:|:-----:|
119
  | 1.3626 | 0.9835 | 0.1522 | 0.2097 | 0.7080 | 0.0 | 0.7886 | 0.8146 | 0.1217 | 0.6821 | 0.0153 | nan | 0.0673 | 0.0 | 0.0 | 0.7948 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.7900 | 0.0 | 0.0076 | 0.0344 | nan | nan | 0.0 | 0.0313 | 0.0 | 0.0 | 0.8325 | 0.7788 | 0.9476 | 0.0 | 0.0 | 0.0046 | 0.0 | 0.0 | 0.5020 | 0.7070 | 0.1182 | 0.4076 | 0.0149 | nan | 0.0590 | 0.0 | 0.0 | 0.4022 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.5052 | 0.0 | 0.0076 | 0.0301 | nan | nan | 0.0 | 0.0302 | 0.0 | 0.0 | 0.7228 | 0.5421 | 0.8162 | 0.0 | 0.0 | 0.0045 | 0.0 | 0 |
 
120
 
121
 
122
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Train Loss: 0.9675
19
+ - Validation Loss: 0.9235
20
+ - Validation Mean Iou: 0.1731
21
+ - Validation Mean Accuracy: 0.2326
22
+ - Validation Overall Accuracy: 0.7231
23
  - Validation Accuracy Unlabeled: 0.0
24
+ - Validation Accuracy Flat-road: 0.8544
25
+ - Validation Accuracy Flat-sidewalk: 0.7814
26
+ - Validation Accuracy Flat-crosswalk: 0.6123
27
+ - Validation Accuracy Flat-cyclinglane: 0.6708
28
+ - Validation Accuracy Flat-parkingdriveway: 0.0943
29
  - Validation Accuracy Flat-railtrack: nan
30
+ - Validation Accuracy Flat-curb: 0.2225
31
+ - Validation Accuracy Human-person: 0.0021
32
  - Validation Accuracy Human-rider: 0.0
33
+ - Validation Accuracy Vehicle-car: 0.4943
34
  - Validation Accuracy Vehicle-truck: 0.0
35
  - Validation Accuracy Vehicle-bus: 0.0
36
  - Validation Accuracy Vehicle-tramtrain: 0.0
37
  - Validation Accuracy Vehicle-motorcycle: 0.0
38
+ - Validation Accuracy Vehicle-bicycle: 0.0007
39
  - Validation Accuracy Vehicle-caravan: 0.0
40
  - Validation Accuracy Vehicle-cartrailer: 0.0
41
+ - Validation Accuracy Construction-building: 0.8354
42
  - Validation Accuracy Construction-door: 0.0
43
+ - Validation Accuracy Construction-wall: 0.0824
44
+ - Validation Accuracy Construction-fenceguardrail: 0.0179
45
  - Validation Accuracy Construction-bridge: nan
46
  - Validation Accuracy Construction-tunnel: nan
47
  - Validation Accuracy Construction-stairs: 0.0
48
+ - Validation Accuracy Object-pole: 0.0980
49
  - Validation Accuracy Object-trafficsign: 0.0
50
  - Validation Accuracy Object-trafficlight: 0.0
51
+ - Validation Accuracy Nature-vegetation: 0.8368
52
+ - Validation Accuracy Nature-terrain: 0.8757
53
+ - Validation Accuracy Sky: 0.9565
54
  - Validation Accuracy Void-ground: 0.0
55
  - Validation Accuracy Void-dynamic: 0.0
56
+ - Validation Accuracy Void-static: 0.0083
57
  - Validation Accuracy Void-unclear: 0.0
58
  - Validation Iou Unlabeled: 0.0
59
+ - Validation Iou Flat-road: 0.5788
60
+ - Validation Iou Flat-sidewalk: 0.7239
61
+ - Validation Iou Flat-crosswalk: 0.3811
62
+ - Validation Iou Flat-cyclinglane: 0.4478
63
+ - Validation Iou Flat-parkingdriveway: 0.0839
64
  - Validation Iou Flat-railtrack: nan
65
+ - Validation Iou Flat-curb: 0.1667
66
+ - Validation Iou Human-person: 0.0021
67
  - Validation Iou Human-rider: 0.0
68
+ - Validation Iou Vehicle-car: 0.4314
69
  - Validation Iou Vehicle-truck: 0.0
70
  - Validation Iou Vehicle-bus: 0.0
71
  - Validation Iou Vehicle-tramtrain: 0.0
72
  - Validation Iou Vehicle-motorcycle: 0.0
73
+ - Validation Iou Vehicle-bicycle: 0.0007
74
  - Validation Iou Vehicle-caravan: 0.0
75
  - Validation Iou Vehicle-cartrailer: 0.0
76
+ - Validation Iou Construction-building: 0.5237
77
  - Validation Iou Construction-door: 0.0
78
+ - Validation Iou Construction-wall: 0.0734
79
+ - Validation Iou Construction-fenceguardrail: 0.0177
80
  - Validation Iou Construction-bridge: nan
81
  - Validation Iou Construction-tunnel: nan
82
  - Validation Iou Construction-stairs: 0.0
83
+ - Validation Iou Object-pole: 0.0854
84
  - Validation Iou Object-trafficsign: 0.0
85
  - Validation Iou Object-trafficlight: 0.0
86
+ - Validation Iou Nature-vegetation: 0.6620
87
+ - Validation Iou Nature-terrain: 0.5031
88
+ - Validation Iou Sky: 0.8488
89
  - Validation Iou Void-ground: 0.0
90
  - Validation Iou Void-dynamic: 0.0
91
+ - Validation Iou Void-static: 0.0080
92
  - Validation Iou Void-unclear: 0.0
93
+ - Epoch: 1
94
 
95
  ## Model description
96
 
 
117
  | Train Loss | Validation Loss | Validation Mean Iou | Validation Mean Accuracy | Validation Overall Accuracy | Validation Accuracy Unlabeled | Validation Accuracy Flat-road | Validation Accuracy Flat-sidewalk | Validation Accuracy Flat-crosswalk | Validation Accuracy Flat-cyclinglane | Validation Accuracy Flat-parkingdriveway | Validation Accuracy Flat-railtrack | Validation Accuracy Flat-curb | Validation Accuracy Human-person | Validation Accuracy Human-rider | Validation Accuracy Vehicle-car | Validation Accuracy Vehicle-truck | Validation Accuracy Vehicle-bus | Validation Accuracy Vehicle-tramtrain | Validation Accuracy Vehicle-motorcycle | Validation Accuracy Vehicle-bicycle | Validation Accuracy Vehicle-caravan | Validation Accuracy Vehicle-cartrailer | Validation Accuracy Construction-building | Validation Accuracy Construction-door | Validation Accuracy Construction-wall | Validation Accuracy Construction-fenceguardrail | Validation Accuracy Construction-bridge | Validation Accuracy Construction-tunnel | Validation Accuracy Construction-stairs | Validation Accuracy Object-pole | Validation Accuracy Object-trafficsign | Validation Accuracy Object-trafficlight | Validation Accuracy Nature-vegetation | Validation Accuracy Nature-terrain | Validation Accuracy Sky | Validation Accuracy Void-ground | Validation Accuracy Void-dynamic | Validation Accuracy Void-static | Validation Accuracy Void-unclear | Validation Iou Unlabeled | Validation Iou Flat-road | Validation Iou Flat-sidewalk | Validation Iou Flat-crosswalk | Validation Iou Flat-cyclinglane | Validation Iou Flat-parkingdriveway | Validation Iou Flat-railtrack | Validation Iou Flat-curb | Validation Iou Human-person | Validation Iou Human-rider | Validation Iou Vehicle-car | Validation Iou Vehicle-truck | Validation Iou Vehicle-bus | Validation Iou Vehicle-tramtrain | Validation Iou Vehicle-motorcycle | Validation Iou Vehicle-bicycle | Validation Iou Vehicle-caravan | Validation Iou Vehicle-cartrailer | Validation Iou Construction-building | Validation Iou Construction-door | Validation Iou Construction-wall | Validation Iou Construction-fenceguardrail | Validation Iou Construction-bridge | Validation Iou Construction-tunnel | Validation Iou Construction-stairs | Validation Iou Object-pole | Validation Iou Object-trafficsign | Validation Iou Object-trafficlight | Validation Iou Nature-vegetation | Validation Iou Nature-terrain | Validation Iou Sky | Validation Iou Void-ground | Validation Iou Void-dynamic | Validation Iou Void-static | Validation Iou Void-unclear | Epoch |
118
  |:----------:|:---------------:|:-------------------:|:------------------------:|:---------------------------:|:-----------------------------:|:-----------------------------:|:---------------------------------:|:----------------------------------:|:------------------------------------:|:----------------------------------------:|:----------------------------------:|:-----------------------------:|:--------------------------------:|:-------------------------------:|:-------------------------------:|:---------------------------------:|:-------------------------------:|:-------------------------------------:|:--------------------------------------:|:-----------------------------------:|:-----------------------------------:|:--------------------------------------:|:-----------------------------------------:|:-------------------------------------:|:-------------------------------------:|:-----------------------------------------------:|:---------------------------------------:|:---------------------------------------:|:---------------------------------------:|:-------------------------------:|:--------------------------------------:|:---------------------------------------:|:-------------------------------------:|:----------------------------------:|:-----------------------:|:-------------------------------:|:--------------------------------:|:-------------------------------:|:--------------------------------:|:------------------------:|:------------------------:|:----------------------------:|:-----------------------------:|:-------------------------------:|:-----------------------------------:|:-----------------------------:|:------------------------:|:---------------------------:|:--------------------------:|:--------------------------:|:----------------------------:|:--------------------------:|:--------------------------------:|:---------------------------------:|:------------------------------:|:------------------------------:|:---------------------------------:|:------------------------------------:|:--------------------------------:|:--------------------------------:|:------------------------------------------:|:----------------------------------:|:----------------------------------:|:----------------------------------:|:--------------------------:|:---------------------------------:|:----------------------------------:|:--------------------------------:|:-----------------------------:|:------------------:|:--------------------------:|:---------------------------:|:--------------------------:|:---------------------------:|:-----:|
119
  | 1.3626 | 0.9835 | 0.1522 | 0.2097 | 0.7080 | 0.0 | 0.7886 | 0.8146 | 0.1217 | 0.6821 | 0.0153 | nan | 0.0673 | 0.0 | 0.0 | 0.7948 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.7900 | 0.0 | 0.0076 | 0.0344 | nan | nan | 0.0 | 0.0313 | 0.0 | 0.0 | 0.8325 | 0.7788 | 0.9476 | 0.0 | 0.0 | 0.0046 | 0.0 | 0.0 | 0.5020 | 0.7070 | 0.1182 | 0.4076 | 0.0149 | nan | 0.0590 | 0.0 | 0.0 | 0.4022 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.5052 | 0.0 | 0.0076 | 0.0301 | nan | nan | 0.0 | 0.0302 | 0.0 | 0.0 | 0.7228 | 0.5421 | 0.8162 | 0.0 | 0.0 | 0.0045 | 0.0 | 0 |
120
+ | 0.9675 | 0.9235 | 0.1731 | 0.2326 | 0.7231 | 0.0 | 0.8544 | 0.7814 | 0.6123 | 0.6708 | 0.0943 | nan | 0.2225 | 0.0021 | 0.0 | 0.4943 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0007 | 0.0 | 0.0 | 0.8354 | 0.0 | 0.0824 | 0.0179 | nan | nan | 0.0 | 0.0980 | 0.0 | 0.0 | 0.8368 | 0.8757 | 0.9565 | 0.0 | 0.0 | 0.0083 | 0.0 | 0.0 | 0.5788 | 0.7239 | 0.3811 | 0.4478 | 0.0839 | nan | 0.1667 | 0.0021 | 0.0 | 0.4314 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0007 | 0.0 | 0.0 | 0.5237 | 0.0 | 0.0734 | 0.0177 | nan | nan | 0.0 | 0.0854 | 0.0 | 0.0 | 0.6620 | 0.5031 | 0.8488 | 0.0 | 0.0 | 0.0080 | 0.0 | 1 |
121
 
122
 
123
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8ffe0229d2e70366754b07d9f830b4bba37aadb305fe1bf42b330aaeb8ce489e
3
  size 15167476
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c87fa6bdc2bcbb17c967a098f7d573be76f0e27ba4ac6bd4acb3506d2ad9e0f
3
  size 15167476