Hugging Face's logo

fshala
/
segformer-finetuned-sidewalk-10k-steps

Image Segmentation
Transformers
Safetensors
segformer
vision
Generated from Trainer
Inference Endpoints
Model card Files Community
fshala commited on
Commit
a699a4f
•
1 Parent(s): 2795eff

End of training

Browse files
Files changed (5) hide show
  1. README.md +3 -1
  2. all_results.json +84 -0
  3. eval_results.json +80 -0
  4. train_results.json +7 -0
  5. trainer_state.json +117 -0
README.md CHANGED
@@ -2,6 +2,8 @@
2
  license: other
3
  base_model: nvidia/mit-b0
4
  tags:
 
 
5
  - generated_from_trainer
6
  model-index:
7
  - name: segformer-finetuned-sidewalk-10k-steps
@@ -13,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
13
 
14
  # segformer-finetuned-sidewalk-10k-steps
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
  - Loss: 2.3528
19
  - Mean Iou: 0.0705
 
2
  license: other
3
  base_model: nvidia/mit-b0
4
  tags:
5
+ - image-segmentation
6
+ - vision
7
  - generated_from_trainer
8
  model-index:
9
  - name: segformer-finetuned-sidewalk-10k-steps
 
15
 
16
  # segformer-finetuned-sidewalk-10k-steps
17
 
18
+ This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the segments/sidewalk-semantic dataset.
19
  It achieves the following results on the evaluation set:
20
  - Loss: 2.3528
21
  - Mean Iou: 0.0705
all_results.json ADDED
@@ -0,0 +1,84 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 0.93,
3
+ "eval_accuracy_construction-bridge": 0.0,
4
+ "eval_accuracy_construction-building": 0.9444419202471694,
5
+ "eval_accuracy_construction-door": 0.0,
6
+ "eval_accuracy_construction-fenceguardrail": 3.923604404095036e-05,
7
+ "eval_accuracy_construction-stairs": 0.0,
8
+ "eval_accuracy_construction-tunnel": NaN,
9
+ "eval_accuracy_construction-wall": 0.001183617722412426,
10
+ "eval_accuracy_flat-crosswalk": 0.0,
11
+ "eval_accuracy_flat-curb": 0.0,
12
+ "eval_accuracy_flat-cyclinglane": 0.013581607990224233,
13
+ "eval_accuracy_flat-parkingdriveway": 0.007780595173933017,
14
+ "eval_accuracy_flat-railtrack": 0.0,
15
+ "eval_accuracy_flat-road": 0.5861226950873202,
16
+ "eval_accuracy_flat-sidewalk": 0.865327509305379,
17
+ "eval_accuracy_human-person": 0.0,
18
+ "eval_accuracy_human-rider": 0.0,
19
+ "eval_accuracy_nature-terrain": 0.28296336923929294,
20
+ "eval_accuracy_nature-vegetation": 0.8840515291641757,
21
+ "eval_accuracy_object-pole": 0.0,
22
+ "eval_accuracy_object-trafficlight": 0.0,
23
+ "eval_accuracy_object-trafficsign": 0.0,
24
+ "eval_accuracy_sky": 0.02156511530897428,
25
+ "eval_accuracy_unlabeled": NaN,
26
+ "eval_accuracy_vehicle-bicycle": 0.0,
27
+ "eval_accuracy_vehicle-bus": 0.0,
28
+ "eval_accuracy_vehicle-car": 0.0736926804716787,
29
+ "eval_accuracy_vehicle-caravan": 0.0,
30
+ "eval_accuracy_vehicle-cartrailer": 0.0,
31
+ "eval_accuracy_vehicle-motorcycle": 0.0,
32
+ "eval_accuracy_vehicle-tramtrain": 0.0,
33
+ "eval_accuracy_vehicle-truck": 0.0,
34
+ "eval_accuracy_void-dynamic": 0.0,
35
+ "eval_accuracy_void-ground": 0.0,
36
+ "eval_accuracy_void-static": 0.0,
37
+ "eval_accuracy_void-unclear": 0.0,
38
+ "eval_iou_construction-bridge": 0.0,
39
+ "eval_iou_construction-building": 0.24548082236952903,
40
+ "eval_iou_construction-door": 0.0,
41
+ "eval_iou_construction-fenceguardrail": 3.920906272243603e-05,
42
+ "eval_iou_construction-stairs": 0.0,
43
+ "eval_iou_construction-tunnel": NaN,
44
+ "eval_iou_construction-wall": 0.0011254059577880163,
45
+ "eval_iou_flat-crosswalk": 0.0,
46
+ "eval_iou_flat-curb": 0.0,
47
+ "eval_iou_flat-cyclinglane": 0.013460014020449992,
48
+ "eval_iou_flat-parkingdriveway": 0.00755640428976123,
49
+ "eval_iou_flat-railtrack": 0.0,
50
+ "eval_iou_flat-road": 0.3993429122612635,
51
+ "eval_iou_flat-sidewalk": 0.667230911539625,
52
+ "eval_iou_human-person": 0.0,
53
+ "eval_iou_human-rider": 0.0,
54
+ "eval_iou_nature-terrain": 0.2515121800324801,
55
+ "eval_iou_nature-vegetation": 0.6483838911605302,
56
+ "eval_iou_object-pole": 0.0,
57
+ "eval_iou_object-trafficlight": 0.0,
58
+ "eval_iou_object-trafficsign": 0.0,
59
+ "eval_iou_sky": 0.021554642017551418,
60
+ "eval_iou_unlabeled": NaN,
61
+ "eval_iou_vehicle-bicycle": 0.0,
62
+ "eval_iou_vehicle-bus": 0.0,
63
+ "eval_iou_vehicle-car": 0.07115443062659547,
64
+ "eval_iou_vehicle-caravan": 0.0,
65
+ "eval_iou_vehicle-cartrailer": 0.0,
66
+ "eval_iou_vehicle-motorcycle": 0.0,
67
+ "eval_iou_vehicle-tramtrain": 0.0,
68
+ "eval_iou_vehicle-truck": 0.0,
69
+ "eval_iou_void-dynamic": 0.0,
70
+ "eval_iou_void-ground": 0.0,
71
+ "eval_iou_void-static": 0.0,
72
+ "eval_iou_void-unclear": 0.0,
73
+ "eval_loss": 2.352823495864868,
74
+ "eval_mean_accuracy": 0.11153787502286669,
75
+ "eval_mean_iou": 0.07051032797994838,
76
+ "eval_overall_accuracy": 0.5632273082669075,
77
+ "eval_runtime": 236.8835,
78
+ "eval_samples_per_second": 0.633,
79
+ "eval_steps_per_second": 0.08,
80
+ "train_loss": 2.811392517089844,
81
+ "train_runtime": 880.5399,
82
+ "train_samples_per_second": 0.909,
83
+ "train_steps_per_second": 0.114
84
+ }
eval_results.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 0.93,
3
+ "eval_accuracy_construction-bridge": 0.0,
4
+ "eval_accuracy_construction-building": 0.9444419202471694,
5
+ "eval_accuracy_construction-door": 0.0,
6
+ "eval_accuracy_construction-fenceguardrail": 3.923604404095036e-05,
7
+ "eval_accuracy_construction-stairs": 0.0,
8
+ "eval_accuracy_construction-tunnel": NaN,
9
+ "eval_accuracy_construction-wall": 0.001183617722412426,
10
+ "eval_accuracy_flat-crosswalk": 0.0,
11
+ "eval_accuracy_flat-curb": 0.0,
12
+ "eval_accuracy_flat-cyclinglane": 0.013581607990224233,
13
+ "eval_accuracy_flat-parkingdriveway": 0.007780595173933017,
14
+ "eval_accuracy_flat-railtrack": 0.0,
15
+ "eval_accuracy_flat-road": 0.5861226950873202,
16
+ "eval_accuracy_flat-sidewalk": 0.865327509305379,
17
+ "eval_accuracy_human-person": 0.0,
18
+ "eval_accuracy_human-rider": 0.0,
19
+ "eval_accuracy_nature-terrain": 0.28296336923929294,
20
+ "eval_accuracy_nature-vegetation": 0.8840515291641757,
21
+ "eval_accuracy_object-pole": 0.0,
22
+ "eval_accuracy_object-trafficlight": 0.0,
23
+ "eval_accuracy_object-trafficsign": 0.0,
24
+ "eval_accuracy_sky": 0.02156511530897428,
25
+ "eval_accuracy_unlabeled": NaN,
26
+ "eval_accuracy_vehicle-bicycle": 0.0,
27
+ "eval_accuracy_vehicle-bus": 0.0,
28
+ "eval_accuracy_vehicle-car": 0.0736926804716787,
29
+ "eval_accuracy_vehicle-caravan": 0.0,
30
+ "eval_accuracy_vehicle-cartrailer": 0.0,
31
+ "eval_accuracy_vehicle-motorcycle": 0.0,
32
+ "eval_accuracy_vehicle-tramtrain": 0.0,
33
+ "eval_accuracy_vehicle-truck": 0.0,
34
+ "eval_accuracy_void-dynamic": 0.0,
35
+ "eval_accuracy_void-ground": 0.0,
36
+ "eval_accuracy_void-static": 0.0,
37
+ "eval_accuracy_void-unclear": 0.0,
38
+ "eval_iou_construction-bridge": 0.0,
39
+ "eval_iou_construction-building": 0.24548082236952903,
40
+ "eval_iou_construction-door": 0.0,
41
+ "eval_iou_construction-fenceguardrail": 3.920906272243603e-05,
42
+ "eval_iou_construction-stairs": 0.0,
43
+ "eval_iou_construction-tunnel": NaN,
44
+ "eval_iou_construction-wall": 0.0011254059577880163,
45
+ "eval_iou_flat-crosswalk": 0.0,
46
+ "eval_iou_flat-curb": 0.0,
47
+ "eval_iou_flat-cyclinglane": 0.013460014020449992,
48
+ "eval_iou_flat-parkingdriveway": 0.00755640428976123,
49
+ "eval_iou_flat-railtrack": 0.0,
50
+ "eval_iou_flat-road": 0.3993429122612635,
51
+ "eval_iou_flat-sidewalk": 0.667230911539625,
52
+ "eval_iou_human-person": 0.0,
53
+ "eval_iou_human-rider": 0.0,
54
+ "eval_iou_nature-terrain": 0.2515121800324801,
55
+ "eval_iou_nature-vegetation": 0.6483838911605302,
56
+ "eval_iou_object-pole": 0.0,
57
+ "eval_iou_object-trafficlight": 0.0,
58
+ "eval_iou_object-trafficsign": 0.0,
59
+ "eval_iou_sky": 0.021554642017551418,
60
+ "eval_iou_unlabeled": NaN,
61
+ "eval_iou_vehicle-bicycle": 0.0,
62
+ "eval_iou_vehicle-bus": 0.0,
63
+ "eval_iou_vehicle-car": 0.07115443062659547,
64
+ "eval_iou_vehicle-caravan": 0.0,
65
+ "eval_iou_vehicle-cartrailer": 0.0,
66
+ "eval_iou_vehicle-motorcycle": 0.0,
67
+ "eval_iou_vehicle-tramtrain": 0.0,
68
+ "eval_iou_vehicle-truck": 0.0,
69
+ "eval_iou_void-dynamic": 0.0,
70
+ "eval_iou_void-ground": 0.0,
71
+ "eval_iou_void-static": 0.0,
72
+ "eval_iou_void-unclear": 0.0,
73
+ "eval_loss": 2.352823495864868,
74
+ "eval_mean_accuracy": 0.11153787502286669,
75
+ "eval_mean_iou": 0.07051032797994838,
76
+ "eval_overall_accuracy": 0.5632273082669075,
77
+ "eval_runtime": 236.8835,
78
+ "eval_samples_per_second": 0.633,
79
+ "eval_steps_per_second": 0.08
80
+ }
train_results.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 0.93,
3
+ "train_loss": 2.811392517089844,
4
+ "train_runtime": 880.5399,
5
+ "train_samples_per_second": 0.909,
6
+ "train_steps_per_second": 0.114
7
+ }
trainer_state.json ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 0.9345794392523364,
5
+ "eval_steps": 500,
6
+ "global_step": 100,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.93,
13
+ "learning_rate": 1e-07,
14
+ "loss": 2.8114,
15
+ "step": 100
16
+ },
17
+ {
18
+ "epoch": 0.93,
19
+ "eval_accuracy_construction-bridge": 0.0,
20
+ "eval_accuracy_construction-building": 0.9444419202471694,
21
+ "eval_accuracy_construction-door": 0.0,
22
+ "eval_accuracy_construction-fenceguardrail": 3.923604404095036e-05,
23
+ "eval_accuracy_construction-stairs": 0.0,
24
+ "eval_accuracy_construction-tunnel": NaN,
25
+ "eval_accuracy_construction-wall": 0.001183617722412426,
26
+ "eval_accuracy_flat-crosswalk": 0.0,
27
+ "eval_accuracy_flat-curb": 0.0,
28
+ "eval_accuracy_flat-cyclinglane": 0.013581607990224233,
29
+ "eval_accuracy_flat-parkingdriveway": 0.007780595173933017,
30
+ "eval_accuracy_flat-railtrack": 0.0,
31
+ "eval_accuracy_flat-road": 0.5861226950873202,
32
+ "eval_accuracy_flat-sidewalk": 0.865327509305379,
33
+ "eval_accuracy_human-person": 0.0,
34
+ "eval_accuracy_human-rider": 0.0,
35
+ "eval_accuracy_nature-terrain": 0.28296336923929294,
36
+ "eval_accuracy_nature-vegetation": 0.8840515291641757,
37
+ "eval_accuracy_object-pole": 0.0,
38
+ "eval_accuracy_object-trafficlight": 0.0,
39
+ "eval_accuracy_object-trafficsign": 0.0,
40
+ "eval_accuracy_sky": 0.02156511530897428,
41
+ "eval_accuracy_unlabeled": NaN,
42
+ "eval_accuracy_vehicle-bicycle": 0.0,
43
+ "eval_accuracy_vehicle-bus": 0.0,
44
+ "eval_accuracy_vehicle-car": 0.0736926804716787,
45
+ "eval_accuracy_vehicle-caravan": 0.0,
46
+ "eval_accuracy_vehicle-cartrailer": 0.0,
47
+ "eval_accuracy_vehicle-motorcycle": 0.0,
48
+ "eval_accuracy_vehicle-tramtrain": 0.0,
49
+ "eval_accuracy_vehicle-truck": 0.0,
50
+ "eval_accuracy_void-dynamic": 0.0,
51
+ "eval_accuracy_void-ground": 0.0,
52
+ "eval_accuracy_void-static": 0.0,
53
+ "eval_accuracy_void-unclear": 0.0,
54
+ "eval_iou_construction-bridge": 0.0,
55
+ "eval_iou_construction-building": 0.24548082236952903,
56
+ "eval_iou_construction-door": 0.0,
57
+ "eval_iou_construction-fenceguardrail": 3.920906272243603e-05,
58
+ "eval_iou_construction-stairs": 0.0,
59
+ "eval_iou_construction-tunnel": NaN,
60
+ "eval_iou_construction-wall": 0.0011254059577880163,
61
+ "eval_iou_flat-crosswalk": 0.0,
62
+ "eval_iou_flat-curb": 0.0,
63
+ "eval_iou_flat-cyclinglane": 0.013460014020449992,
64
+ "eval_iou_flat-parkingdriveway": 0.00755640428976123,
65
+ "eval_iou_flat-railtrack": 0.0,
66
+ "eval_iou_flat-road": 0.3993429122612635,
67
+ "eval_iou_flat-sidewalk": 0.667230911539625,
68
+ "eval_iou_human-person": 0.0,
69
+ "eval_iou_human-rider": 0.0,
70
+ "eval_iou_nature-terrain": 0.2515121800324801,
71
+ "eval_iou_nature-vegetation": 0.6483838911605302,
72
+ "eval_iou_object-pole": 0.0,
73
+ "eval_iou_object-trafficlight": 0.0,
74
+ "eval_iou_object-trafficsign": 0.0,
75
+ "eval_iou_sky": 0.021554642017551418,
76
+ "eval_iou_unlabeled": NaN,
77
+ "eval_iou_vehicle-bicycle": 0.0,
78
+ "eval_iou_vehicle-bus": 0.0,
79
+ "eval_iou_vehicle-car": 0.07115443062659547,
80
+ "eval_iou_vehicle-caravan": 0.0,
81
+ "eval_iou_vehicle-cartrailer": 0.0,
82
+ "eval_iou_vehicle-motorcycle": 0.0,
83
+ "eval_iou_vehicle-tramtrain": 0.0,
84
+ "eval_iou_vehicle-truck": 0.0,
85
+ "eval_iou_void-dynamic": 0.0,
86
+ "eval_iou_void-ground": 0.0,
87
+ "eval_iou_void-static": 0.0,
88
+ "eval_iou_void-unclear": 0.0,
89
+ "eval_loss": 2.352823495864868,
90
+ "eval_mean_accuracy": 0.11153787502286669,
91
+ "eval_mean_iou": 0.07051032797994838,
92
+ "eval_overall_accuracy": 0.5632273082669075,
93
+ "eval_runtime": 236.1459,
94
+ "eval_samples_per_second": 0.635,
95
+ "eval_steps_per_second": 0.08,
96
+ "step": 100
97
+ },
98
+ {
99
+ "epoch": 0.93,
100
+ "step": 100,
101
+ "total_flos": 1.40543791202304e+16,
102
+ "train_loss": 2.811392517089844,
103
+ "train_runtime": 880.5399,
104
+ "train_samples_per_second": 0.909,
105
+ "train_steps_per_second": 0.114
106
+ }
107
+ ],
108
+ "logging_steps": 100,
109
+ "max_steps": 100,
110
+ "num_input_tokens_seen": 0,
111
+ "num_train_epochs": 1,
112
+ "save_steps": 500,
113
+ "total_flos": 1.40543791202304e+16,
114
+ "train_batch_size": 8,
115
+ "trial_name": null,
116
+ "trial_params": null
117
+ }