Hugging Face's logo

MariaK
/
mit-b0-finetuned-sidewalks

Transformers
TensorFlow
segformer
generated_from_keras_callback
Inference Endpoints
Model card Files Community
MariaK commited on
Commit
171abfe
•
1 Parent(s): ebf1956

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
@@ -14,82 +14,82 @@ probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 1.4362
18
- - Validation Loss: 0.9804
19
- - Validation Mean Iou: 0.1752
20
- - Validation Mean Accuracy: 0.2219
21
- - Validation Overall Accuracy: 0.7360
22
  - Validation Accuracy Unlabeled: 0.0
23
- - Validation Accuracy Flat-road: 0.7417
24
- - Validation Accuracy Flat-sidewalk: 0.9512
25
- - Validation Accuracy Flat-crosswalk: 0.0213
26
- - Validation Accuracy Flat-cyclinglane: 0.3662
27
- - Validation Accuracy Flat-parkingdriveway: 0.1475
28
  - Validation Accuracy Flat-railtrack: nan
29
- - Validation Accuracy Flat-curb: 0.1397
30
- - Validation Accuracy Human-person: 0.0055
31
  - Validation Accuracy Human-rider: 0.0
32
- - Validation Accuracy Vehicle-car: 0.8653
33
  - Validation Accuracy Vehicle-truck: 0.0
34
  - Validation Accuracy Vehicle-bus: 0.0
35
  - Validation Accuracy Vehicle-tramtrain: nan
36
  - Validation Accuracy Vehicle-motorcycle: 0.0
37
- - Validation Accuracy Vehicle-bicycle: 0.0002
38
  - Validation Accuracy Vehicle-caravan: 0.0
39
  - Validation Accuracy Vehicle-cartrailer: 0.0
40
- - Validation Accuracy Construction-building: 0.7778
41
  - Validation Accuracy Construction-door: 0.0
42
- - Validation Accuracy Construction-wall: 0.3370
43
- - Validation Accuracy Construction-fenceguardrail: 0.0429
44
  - Validation Accuracy Construction-bridge: 0.0
45
  - Validation Accuracy Construction-tunnel: nan
46
  - Validation Accuracy Construction-stairs: 0.0
47
- - Validation Accuracy Object-pole: 0.0177
48
  - Validation Accuracy Object-trafficsign: 0.0
49
  - Validation Accuracy Object-trafficlight: 0.0
50
- - Validation Accuracy Nature-vegetation: 0.9324
51
- - Validation Accuracy Nature-terrain: 0.7967
52
- - Validation Accuracy Sky: 0.9157
53
  - Validation Accuracy Void-ground: 0.0
54
  - Validation Accuracy Void-dynamic: 0.0
55
- - Validation Accuracy Void-static: 0.0409
56
  - Validation Accuracy Void-unclear: 0.0
57
  - Validation Iou Unlabeled: 0.0
58
- - Validation Iou Flat-road: 0.5263
59
- - Validation Iou Flat-sidewalk: 0.7377
60
- - Validation Iou Flat-crosswalk: 0.0213
61
- - Validation Iou Flat-cyclinglane: 0.3517
62
- - Validation Iou Flat-parkingdriveway: 0.1232
63
  - Validation Iou Flat-railtrack: nan
64
- - Validation Iou Flat-curb: 0.1053
65
- - Validation Iou Human-person: 0.0055
66
  - Validation Iou Human-rider: 0.0
67
- - Validation Iou Vehicle-car: 0.6423
68
  - Validation Iou Vehicle-truck: 0.0
69
  - Validation Iou Vehicle-bus: 0.0
70
  - Validation Iou Vehicle-tramtrain: nan
71
  - Validation Iou Vehicle-motorcycle: 0.0
72
- - Validation Iou Vehicle-bicycle: 0.0002
73
  - Validation Iou Vehicle-caravan: 0.0
74
  - Validation Iou Vehicle-cartrailer: 0.0
75
- - Validation Iou Construction-building: 0.6012
76
  - Validation Iou Construction-door: 0.0
77
- - Validation Iou Construction-wall: 0.2315
78
- - Validation Iou Construction-fenceguardrail: 0.0424
79
  - Validation Iou Construction-bridge: 0.0
80
  - Validation Iou Construction-tunnel: nan
81
  - Validation Iou Construction-stairs: 0.0
82
- - Validation Iou Object-pole: 0.0163
83
  - Validation Iou Object-trafficsign: 0.0
84
  - Validation Iou Object-trafficlight: 0.0
85
- - Validation Iou Nature-vegetation: 0.7258
86
- - Validation Iou Nature-terrain: 0.6752
87
- - Validation Iou Sky: 0.7692
88
  - Validation Iou Void-ground: 0.0
89
  - Validation Iou Void-dynamic: 0.0
90
- - Validation Iou Void-static: 0.0321
91
  - Validation Iou Void-unclear: 0.0
92
- - Epoch: 0
93
 
94
  ## Model description
95
 
@@ -116,6 +116,7 @@ The following hyperparameters were used during training:
116
  | 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 |
117
  |:----------:|:---------------:|:-------------------:|:------------------------:|:---------------------------:|:-----------------------------:|:-----------------------------:|:---------------------------------:|:----------------------------------:|:------------------------------------:|:----------------------------------------:|:----------------------------------:|:-----------------------------:|:--------------------------------:|:-------------------------------:|:-------------------------------:|:---------------------------------:|:-------------------------------:|:-------------------------------------:|:--------------------------------------:|:-----------------------------------:|:-----------------------------------:|:--------------------------------------:|:-----------------------------------------:|:-------------------------------------:|:-------------------------------------:|:-----------------------------------------------:|:---------------------------------------:|:---------------------------------------:|:---------------------------------------:|:-------------------------------:|:--------------------------------------:|:---------------------------------------:|:-------------------------------------:|:----------------------------------:|:-----------------------:|:-------------------------------:|:--------------------------------:|:-------------------------------:|:--------------------------------:|:------------------------:|:------------------------:|:----------------------------:|:-----------------------------:|:-------------------------------:|:-----------------------------------:|:-----------------------------:|:------------------------:|:---------------------------:|:--------------------------:|:--------------------------:|:----------------------------:|:--------------------------:|:--------------------------------:|:---------------------------------:|:------------------------------:|:------------------------------:|:---------------------------------:|:------------------------------------:|:--------------------------------:|:--------------------------------:|:------------------------------------------:|:----------------------------------:|:----------------------------------:|:----------------------------------:|:--------------------------:|:---------------------------------:|:----------------------------------:|:--------------------------------:|:-----------------------------:|:------------------:|:--------------------------:|:---------------------------:|:--------------------------:|:---------------------------:|:-----:|
118
  | 1.4362 | 0.9804 | 0.1752 | 0.2219 | 0.7360 | 0.0 | 0.7417 | 0.9512 | 0.0213 | 0.3662 | 0.1475 | nan | 0.1397 | 0.0055 | 0.0 | 0.8653 | 0.0 | 0.0 | nan | 0.0 | 0.0002 | 0.0 | 0.0 | 0.7778 | 0.0 | 0.3370 | 0.0429 | 0.0 | nan | 0.0 | 0.0177 | 0.0 | 0.0 | 0.9324 | 0.7967 | 0.9157 | 0.0 | 0.0 | 0.0409 | 0.0 | 0.0 | 0.5263 | 0.7377 | 0.0213 | 0.3517 | 0.1232 | nan | 0.1053 | 0.0055 | 0.0 | 0.6423 | 0.0 | 0.0 | nan | 0.0 | 0.0002 | 0.0 | 0.0 | 0.6012 | 0.0 | 0.2315 | 0.0424 | 0.0 | nan | 0.0 | 0.0163 | 0.0 | 0.0 | 0.7258 | 0.6752 | 0.7692 | 0.0 | 0.0 | 0.0321 | 0.0 | 0 |
 
119
 
120
 
121
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Train Loss: 0.8550
18
+ - Validation Loss: 0.8639
19
+ - Validation Mean Iou: 0.2220
20
+ - Validation Mean Accuracy: 0.2670
21
+ - Validation Overall Accuracy: 0.7725
22
  - Validation Accuracy Unlabeled: 0.0
23
+ - Validation Accuracy Flat-road: 0.6015
24
+ - Validation Accuracy Flat-sidewalk: 0.9708
25
+ - Validation Accuracy Flat-crosswalk: 0.3807
26
+ - Validation Accuracy Flat-cyclinglane: 0.7538
27
+ - Validation Accuracy Flat-parkingdriveway: 0.1524
28
  - Validation Accuracy Flat-railtrack: nan
29
+ - Validation Accuracy Flat-curb: 0.1957
30
+ - Validation Accuracy Human-person: 0.2585
31
  - Validation Accuracy Human-rider: 0.0
32
+ - Validation Accuracy Vehicle-car: 0.8971
33
  - Validation Accuracy Vehicle-truck: 0.0
34
  - Validation Accuracy Vehicle-bus: 0.0
35
  - Validation Accuracy Vehicle-tramtrain: nan
36
  - Validation Accuracy Vehicle-motorcycle: 0.0
37
+ - Validation Accuracy Vehicle-bicycle: 0.0716
38
  - Validation Accuracy Vehicle-caravan: 0.0
39
  - Validation Accuracy Vehicle-cartrailer: 0.0
40
+ - Validation Accuracy Construction-building: 0.8784
41
  - Validation Accuracy Construction-door: 0.0
42
+ - Validation Accuracy Construction-wall: 0.4315
43
+ - Validation Accuracy Construction-fenceguardrail: 0.1948
44
  - Validation Accuracy Construction-bridge: 0.0
45
  - Validation Accuracy Construction-tunnel: nan
46
  - Validation Accuracy Construction-stairs: 0.0
47
+ - Validation Accuracy Object-pole: 0.1201
48
  - Validation Accuracy Object-trafficsign: 0.0
49
  - Validation Accuracy Object-trafficlight: 0.0
50
+ - Validation Accuracy Nature-vegetation: 0.8952
51
+ - Validation Accuracy Nature-terrain: 0.8231
52
+ - Validation Accuracy Sky: 0.8496
53
  - Validation Accuracy Void-ground: 0.0
54
  - Validation Accuracy Void-dynamic: 0.0
55
+ - Validation Accuracy Void-static: 0.0692
56
  - Validation Accuracy Void-unclear: 0.0
57
  - Validation Iou Unlabeled: 0.0
58
+ - Validation Iou Flat-road: 0.5568
59
+ - Validation Iou Flat-sidewalk: 0.7479
60
+ - Validation Iou Flat-crosswalk: 0.3509
61
+ - Validation Iou Flat-cyclinglane: 0.6355
62
+ - Validation Iou Flat-parkingdriveway: 0.1298
63
  - Validation Iou Flat-railtrack: nan
64
+ - Validation Iou Flat-curb: 0.1326
65
+ - Validation Iou Human-person: 0.2455
66
  - Validation Iou Human-rider: 0.0
67
+ - Validation Iou Vehicle-car: 0.6973
68
  - Validation Iou Vehicle-truck: 0.0
69
  - Validation Iou Vehicle-bus: 0.0
70
  - Validation Iou Vehicle-tramtrain: nan
71
  - Validation Iou Vehicle-motorcycle: 0.0
72
+ - Validation Iou Vehicle-bicycle: 0.0610
73
  - Validation Iou Vehicle-caravan: 0.0
74
  - Validation Iou Vehicle-cartrailer: 0.0
75
+ - Validation Iou Construction-building: 0.6479
76
  - Validation Iou Construction-door: 0.0
77
+ - Validation Iou Construction-wall: 0.3003
78
+ - Validation Iou Construction-fenceguardrail: 0.1727
79
  - Validation Iou Construction-bridge: 0.0
80
  - Validation Iou Construction-tunnel: nan
81
  - Validation Iou Construction-stairs: 0.0
82
+ - Validation Iou Object-pole: 0.0927
83
  - Validation Iou Object-trafficsign: 0.0
84
  - Validation Iou Object-trafficlight: 0.0
85
+ - Validation Iou Nature-vegetation: 0.7758
86
+ - Validation Iou Nature-terrain: 0.7000
87
+ - Validation Iou Sky: 0.8002
88
  - Validation Iou Void-ground: 0.0
89
  - Validation Iou Void-dynamic: 0.0
90
+ - Validation Iou Void-static: 0.0573
91
  - Validation Iou Void-unclear: 0.0
92
+ - Epoch: 1
93
 
94
  ## Model description
95
 
 
116
  | 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 |
117
  |:----------:|:---------------:|:-------------------:|:------------------------:|:---------------------------:|:-----------------------------:|:-----------------------------:|:---------------------------------:|:----------------------------------:|:------------------------------------:|:----------------------------------------:|:----------------------------------:|:-----------------------------:|:--------------------------------:|:-------------------------------:|:-------------------------------:|:---------------------------------:|:-------------------------------:|:-------------------------------------:|:--------------------------------------:|:-----------------------------------:|:-----------------------------------:|:--------------------------------------:|:-----------------------------------------:|:-------------------------------------:|:-------------------------------------:|:-----------------------------------------------:|:---------------------------------------:|:---------------------------------------:|:---------------------------------------:|:-------------------------------:|:--------------------------------------:|:---------------------------------------:|:-------------------------------------:|:----------------------------------:|:-----------------------:|:-------------------------------:|:--------------------------------:|:-------------------------------:|:--------------------------------:|:------------------------:|:------------------------:|:----------------------------:|:-----------------------------:|:-------------------------------:|:-----------------------------------:|:-----------------------------:|:------------------------:|:---------------------------:|:--------------------------:|:--------------------------:|:----------------------------:|:--------------------------:|:--------------------------------:|:---------------------------------:|:------------------------------:|:------------------------------:|:---------------------------------:|:------------------------------------:|:--------------------------------:|:--------------------------------:|:------------------------------------------:|:----------------------------------:|:----------------------------------:|:----------------------------------:|:--------------------------:|:---------------------------------:|:----------------------------------:|:--------------------------------:|:-----------------------------:|:------------------:|:--------------------------:|:---------------------------:|:--------------------------:|:---------------------------:|:-----:|
118
  | 1.4362 | 0.9804 | 0.1752 | 0.2219 | 0.7360 | 0.0 | 0.7417 | 0.9512 | 0.0213 | 0.3662 | 0.1475 | nan | 0.1397 | 0.0055 | 0.0 | 0.8653 | 0.0 | 0.0 | nan | 0.0 | 0.0002 | 0.0 | 0.0 | 0.7778 | 0.0 | 0.3370 | 0.0429 | 0.0 | nan | 0.0 | 0.0177 | 0.0 | 0.0 | 0.9324 | 0.7967 | 0.9157 | 0.0 | 0.0 | 0.0409 | 0.0 | 0.0 | 0.5263 | 0.7377 | 0.0213 | 0.3517 | 0.1232 | nan | 0.1053 | 0.0055 | 0.0 | 0.6423 | 0.0 | 0.0 | nan | 0.0 | 0.0002 | 0.0 | 0.0 | 0.6012 | 0.0 | 0.2315 | 0.0424 | 0.0 | nan | 0.0 | 0.0163 | 0.0 | 0.0 | 0.7258 | 0.6752 | 0.7692 | 0.0 | 0.0 | 0.0321 | 0.0 | 0 |
119
+ | 0.8550 | 0.8639 | 0.2220 | 0.2670 | 0.7725 | 0.0 | 0.6015 | 0.9708 | 0.3807 | 0.7538 | 0.1524 | nan | 0.1957 | 0.2585 | 0.0 | 0.8971 | 0.0 | 0.0 | nan | 0.0 | 0.0716 | 0.0 | 0.0 | 0.8784 | 0.0 | 0.4315 | 0.1948 | 0.0 | nan | 0.0 | 0.1201 | 0.0 | 0.0 | 0.8952 | 0.8231 | 0.8496 | 0.0 | 0.0 | 0.0692 | 0.0 | 0.0 | 0.5568 | 0.7479 | 0.3509 | 0.6355 | 0.1298 | nan | 0.1326 | 0.2455 | 0.0 | 0.6973 | 0.0 | 0.0 | nan | 0.0 | 0.0610 | 0.0 | 0.0 | 0.6479 | 0.0 | 0.3003 | 0.1727 | 0.0 | nan | 0.0 | 0.0927 | 0.0 | 0.0 | 0.7758 | 0.7000 | 0.8002 | 0.0 | 0.0 | 0.0573 | 0.0 | 1 |
120
 
121
 
122
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2bdf624fd0787762fbf84d767d93e4281b4bfb5b86339ddeddcabb3e6a57ff6f
3
  size 15167588
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:163a3d593e5ed6e9655c316c0a8ee477f8b3d682d87df1bbab408e09b0a940d9
3
  size 15167588