Hugging Face's logo

Vrjb
/
SegFormer_b3

Image Segmentation
Transformers
Safetensors
segformer
vision
Generated from Trainer
Model card Files Community
SegFormer_b3
File size: 2,635 Bytes 
f127ed6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
 
{
  "_name_or_path": "nvidia/segformer-b3-finetuned-cityscapes-1024-1024",
  "architectures": [
    "SegformerForSemanticSegmentation"
  ],
  "attention_probs_dropout_prob": 0.0,
  "classifier_dropout_prob": 0.1,
  "decoder_hidden_size": 768,
  "depths": [
    3,
    4,
    18,
    3
  ],
  "downsampling_rates": [
    1,
    4,
    8,
    16
  ],
  "drop_path_rate": 0.1,
  "hidden_act": "gelu",
  "hidden_dropout_prob": 0.0,
  "hidden_sizes": [
    64,
    128,
    320,
    512
  ],
  "id2label": {
    "0": "unlabeled",
    "1": "ego vehicle",
    "2": "rectification border",
    "3": "out of roi",
    "4": "static",
    "5": "dynamic",
    "6": "ground",
    "7": "road",
    "8": "sidewalk",
    "9": "parking",
    "10": "rail track",
    "11": "building",
    "12": "wall",
    "13": "fence",
    "14": "guard rail",
    "15": "bridge",
    "16": "tunnel",
    "17": "pole",
    "18": "polegroup",
    "19": "traffic light",
    "20": "traffic sign",
    "21": "vegetation",
    "22": "terrain",
    "23": "sky",
    "24": "person",
    "25": "rider",
    "26": "car",
    "27": "truck",
    "28": "bus",
    "29": "caravan",
    "30": "trailer",
    "31": "train",
    "32": "motorcycle",
    "33": "bicycle",
    "34": "license plate"
  },
  "image_size": 224,
  "initializer_range": 0.02,
  "label2id": {
    "bicycle": 33,
    "bridge": 15,
    "building": 11,
    "bus": 28,
    "car": 26,
    "caravan": 29,
    "dynamic": 5,
    "ego vehicle": 1,
    "fence": 13,
    "ground": 6,
    "guard rail": 14,
    "license plate": 34,
    "motorcycle": 32,
    "out of roi": 3,
    "parking": 9,
    "person": 24,
    "pole": 17,
    "polegroup": 18,
    "rail track": 10,
    "rectification border": 2,
    "rider": 25,
    "road": 7,
    "sidewalk": 8,
    "sky": 23,
    "static": 4,
    "terrain": 22,
    "traffic light": 19,
    "traffic sign": 20,
    "trailer": 30,
    "train": 31,
    "truck": 27,
    "tunnel": 16,
    "unlabeled": 0,
    "vegetation": 21,
    "wall": 12
  },
  "layer_norm_eps": 1e-06,
  "mlp_ratios": [
    4,
    4,
    4,
    4
  ],
  "model_type": "segformer",
  "num_attention_heads": [
    1,
    2,
    5,
    8
  ],
  "num_channels": 3,
  "num_encoder_blocks": 4,
  "patch_sizes": [
    7,
    3,
    3,
    3
  ],
  "reshape_last_stage": true,
  "semantic_loss_ignore_index": 255,
  "sr_ratios": [
    8,
    4,
    2,
    1
  ],
  "strides": [
    4,
    2,
    2,
    2
  ],
  "torch_dtype": "float32",
  "transformers_version": "4.47.1"
}