Mohamad-Jaallouk commited on
Commit
5c10ba8
1 Parent(s): 27ca84b

End of training

Browse files
README.md ADDED
@@ -0,0 +1,130 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ base_model: nvidia/mit-b0
4
+ tags:
5
+ - vision
6
+ - image-segmentation
7
+ - generated_from_trainer
8
+ model-index:
9
+ - name: segformer-b0-finetuned-segments-sidewalk-oct-22
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # segformer-b0-finetuned-segments-sidewalk-oct-22
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
+ - eval_loss: 0.8820
21
+ - eval_mean_iou: 0.1648
22
+ - eval_mean_accuracy: 0.2025
23
+ - eval_overall_accuracy: 0.7805
24
+ - eval_accuracy_unlabeled: nan
25
+ - eval_accuracy_flat-road: 0.8435
26
+ - eval_accuracy_flat-sidewalk: 0.9378
27
+ - eval_accuracy_flat-crosswalk: 0.0
28
+ - eval_accuracy_flat-cyclinglane: 0.5809
29
+ - eval_accuracy_flat-parkingdriveway: 0.0715
30
+ - eval_accuracy_flat-railtrack: 0.0
31
+ - eval_accuracy_flat-curb: 0.0041
32
+ - eval_accuracy_human-person: 0.0
33
+ - eval_accuracy_human-rider: 0.0
34
+ - eval_accuracy_vehicle-car: 0.8730
35
+ - eval_accuracy_vehicle-truck: 0.0
36
+ - eval_accuracy_vehicle-bus: 0.0
37
+ - eval_accuracy_vehicle-tramtrain: 0.0
38
+ - eval_accuracy_vehicle-motorcycle: 0.0
39
+ - eval_accuracy_vehicle-bicycle: 0.0
40
+ - eval_accuracy_vehicle-caravan: 0.0
41
+ - eval_accuracy_vehicle-cartrailer: 0.0
42
+ - eval_accuracy_construction-building: 0.8780
43
+ - eval_accuracy_construction-door: 0.0
44
+ - eval_accuracy_construction-wall: 0.0000
45
+ - eval_accuracy_construction-fenceguardrail: 0.0
46
+ - eval_accuracy_construction-bridge: 0.0
47
+ - eval_accuracy_construction-tunnel: 0.0
48
+ - eval_accuracy_construction-stairs: 0.0
49
+ - eval_accuracy_object-pole: 0.0
50
+ - eval_accuracy_object-trafficsign: 0.0
51
+ - eval_accuracy_object-trafficlight: 0.0
52
+ - eval_accuracy_nature-vegetation: 0.9399
53
+ - eval_accuracy_nature-terrain: 0.8232
54
+ - eval_accuracy_sky: 0.9347
55
+ - eval_accuracy_void-ground: 0.0
56
+ - eval_accuracy_void-dynamic: 0.0
57
+ - eval_accuracy_void-static: 0.0
58
+ - eval_accuracy_void-unclear: 0.0
59
+ - eval_iou_unlabeled: nan
60
+ - eval_iou_flat-road: 0.5426
61
+ - eval_iou_flat-sidewalk: 0.8046
62
+ - eval_iou_flat-crosswalk: 0.0
63
+ - eval_iou_flat-cyclinglane: 0.5502
64
+ - eval_iou_flat-parkingdriveway: 0.0678
65
+ - eval_iou_flat-railtrack: 0.0
66
+ - eval_iou_flat-curb: 0.0041
67
+ - eval_iou_human-person: 0.0
68
+ - eval_iou_human-rider: 0.0
69
+ - eval_iou_vehicle-car: 0.6930
70
+ - eval_iou_vehicle-truck: 0.0
71
+ - eval_iou_vehicle-bus: 0.0
72
+ - eval_iou_vehicle-tramtrain: 0.0
73
+ - eval_iou_vehicle-motorcycle: 0.0
74
+ - eval_iou_vehicle-bicycle: 0.0
75
+ - eval_iou_vehicle-caravan: 0.0
76
+ - eval_iou_vehicle-cartrailer: 0.0
77
+ - eval_iou_construction-building: 0.6055
78
+ - eval_iou_construction-door: 0.0
79
+ - eval_iou_construction-wall: 0.0000
80
+ - eval_iou_construction-fenceguardrail: 0.0
81
+ - eval_iou_construction-bridge: 0.0
82
+ - eval_iou_construction-tunnel: 0.0
83
+ - eval_iou_construction-stairs: 0.0
84
+ - eval_iou_object-pole: 0.0
85
+ - eval_iou_object-trafficsign: 0.0
86
+ - eval_iou_object-trafficlight: 0.0
87
+ - eval_iou_nature-vegetation: 0.7900
88
+ - eval_iou_nature-terrain: 0.7063
89
+ - eval_iou_sky: 0.8381
90
+ - eval_iou_void-ground: 0.0
91
+ - eval_iou_void-dynamic: 0.0
92
+ - eval_iou_void-static: 0.0
93
+ - eval_iou_void-unclear: 0.0
94
+ - eval_runtime: 21.9758
95
+ - eval_samples_per_second: 9.101
96
+ - eval_steps_per_second: 0.592
97
+ - epoch: 0.4
98
+ - step: 20
99
+
100
+ ## Model description
101
+
102
+ More information needed
103
+
104
+ ## Intended uses & limitations
105
+
106
+ More information needed
107
+
108
+ ## Training and evaluation data
109
+
110
+ More information needed
111
+
112
+ ## Training procedure
113
+
114
+ ### Training hyperparameters
115
+
116
+ The following hyperparameters were used during training:
117
+ - learning_rate: 6e-05
118
+ - train_batch_size: 16
119
+ - eval_batch_size: 16
120
+ - seed: 42
121
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
122
+ - lr_scheduler_type: linear
123
+ - num_epochs: 50
124
+
125
+ ### Framework versions
126
+
127
+ - Transformers 4.35.2
128
+ - Pytorch 2.1.0+cu121
129
+ - Datasets 2.16.1
130
+ - Tokenizers 0.15.1
config.json ADDED
@@ -0,0 +1,144 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b0",
3
+ "architectures": [
4
+ "SegformerForSemanticSegmentation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 256,
9
+ "depths": [
10
+ 2,
11
+ 2,
12
+ 2,
13
+ 2
14
+ ],
15
+ "downsampling_rates": [
16
+ 1,
17
+ 4,
18
+ 8,
19
+ 16
20
+ ],
21
+ "drop_path_rate": 0.1,
22
+ "hidden_act": "gelu",
23
+ "hidden_dropout_prob": 0.0,
24
+ "hidden_sizes": [
25
+ 32,
26
+ 64,
27
+ 160,
28
+ 256
29
+ ],
30
+ "id2label": {
31
+ "0": "unlabeled",
32
+ "1": "flat-road",
33
+ "2": "flat-sidewalk",
34
+ "3": "flat-crosswalk",
35
+ "4": "flat-cyclinglane",
36
+ "5": "flat-parkingdriveway",
37
+ "6": "flat-railtrack",
38
+ "7": "flat-curb",
39
+ "8": "human-person",
40
+ "9": "human-rider",
41
+ "10": "vehicle-car",
42
+ "11": "vehicle-truck",
43
+ "12": "vehicle-bus",
44
+ "13": "vehicle-tramtrain",
45
+ "14": "vehicle-motorcycle",
46
+ "15": "vehicle-bicycle",
47
+ "16": "vehicle-caravan",
48
+ "17": "vehicle-cartrailer",
49
+ "18": "construction-building",
50
+ "19": "construction-door",
51
+ "20": "construction-wall",
52
+ "21": "construction-fenceguardrail",
53
+ "22": "construction-bridge",
54
+ "23": "construction-tunnel",
55
+ "24": "construction-stairs",
56
+ "25": "object-pole",
57
+ "26": "object-trafficsign",
58
+ "27": "object-trafficlight",
59
+ "28": "nature-vegetation",
60
+ "29": "nature-terrain",
61
+ "30": "sky",
62
+ "31": "void-ground",
63
+ "32": "void-dynamic",
64
+ "33": "void-static",
65
+ "34": "void-unclear"
66
+ },
67
+ "image_size": 224,
68
+ "initializer_range": 0.02,
69
+ "label2id": {
70
+ "construction-bridge": 22,
71
+ "construction-building": 18,
72
+ "construction-door": 19,
73
+ "construction-fenceguardrail": 21,
74
+ "construction-stairs": 24,
75
+ "construction-tunnel": 23,
76
+ "construction-wall": 20,
77
+ "flat-crosswalk": 3,
78
+ "flat-curb": 7,
79
+ "flat-cyclinglane": 4,
80
+ "flat-parkingdriveway": 5,
81
+ "flat-railtrack": 6,
82
+ "flat-road": 1,
83
+ "flat-sidewalk": 2,
84
+ "human-person": 8,
85
+ "human-rider": 9,
86
+ "nature-terrain": 29,
87
+ "nature-vegetation": 28,
88
+ "object-pole": 25,
89
+ "object-trafficlight": 27,
90
+ "object-trafficsign": 26,
91
+ "sky": 30,
92
+ "unlabeled": 0,
93
+ "vehicle-bicycle": 15,
94
+ "vehicle-bus": 12,
95
+ "vehicle-car": 10,
96
+ "vehicle-caravan": 16,
97
+ "vehicle-cartrailer": 17,
98
+ "vehicle-motorcycle": 14,
99
+ "vehicle-tramtrain": 13,
100
+ "vehicle-truck": 11,
101
+ "void-dynamic": 32,
102
+ "void-ground": 31,
103
+ "void-static": 33,
104
+ "void-unclear": 34
105
+ },
106
+ "layer_norm_eps": 1e-06,
107
+ "mlp_ratios": [
108
+ 4,
109
+ 4,
110
+ 4,
111
+ 4
112
+ ],
113
+ "model_type": "segformer",
114
+ "num_attention_heads": [
115
+ 1,
116
+ 2,
117
+ 5,
118
+ 8
119
+ ],
120
+ "num_channels": 3,
121
+ "num_encoder_blocks": 4,
122
+ "patch_sizes": [
123
+ 7,
124
+ 3,
125
+ 3,
126
+ 3
127
+ ],
128
+ "reshape_last_stage": true,
129
+ "semantic_loss_ignore_index": 255,
130
+ "sr_ratios": [
131
+ 8,
132
+ 4,
133
+ 2,
134
+ 1
135
+ ],
136
+ "strides": [
137
+ 4,
138
+ 2,
139
+ 2,
140
+ 2
141
+ ],
142
+ "torch_dtype": "float32",
143
+ "transformers_version": "4.35.2"
144
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba5bf50124d65ca871c67e56d7c9d2c855103d498aa9b11dd7f8ac817f4a046e
3
+ size 14918708
runs/Feb01_11-21-32_a7def9ba7b1c/events.out.tfevents.1706786622.a7def9ba7b1c.6258.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:27162a39a5c0964ae6e467a5bb5a230072fc3dbee4dcbf496490bae9f3902310
3
+ size 195807
runs/Feb01_11-38-47_a7def9ba7b1c/events.out.tfevents.1706787535.a7def9ba7b1c.6258.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b879f9a312ba9a5544c3c35a63e849d3aed8f6871ca536dec333f397a87c7c0a
3
+ size 14791
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf13f6176abc5a08a1621f4deb68ea9c897decf5b68b5e464b249710a70d9fb4
3
+ size 4664