Upload SegformerForSemanticSegmentation

config.json

@@ -4,12 +4,12 @@
   ],
   "attention_probs_dropout_prob": 0.0,
   "classifier_dropout_prob": 0.1,
-  "decoder_hidden_size": 768,
+  "decoder_hidden_size": 256,
   "depths": [
-    3,
-    4,
-    6,
-    3
+    2,
+    2,
+    2,
+    2
   ],
   "downsampling_rates": [
     1,
@@ -27,22 +27,30 @@
     512
   ],
   "id2label": {
-    "0": "class_200",
-    "1": "class_300",
-    "2": "class_500",
-    "3": "class_550",
-    "4": "class_800",
-    "5": "class_7100"
+    "0": "class_100",
+    "1": "class_200",
+    "2": "class_300",
+    "3": "class_500",
+    "4": "class_550",
+    "5": "class_600",
+    "6": "class_700",
+    "7": "class_800",
+    "8": "class_7100",
+    "9": "class_10000"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
-    "class_200": 0,
-    "class_300": 1,
-    "class_500": 2,
-    "class_550": 3,
-    "class_7100": 5,
-    "class_800": 4
+    "class_100": 0,
+    "class_10000": 9,
+    "class_200": 1,
+    "class_300": 2,
+    "class_500": 3,
+    "class_550": 4,
+    "class_600": 5,
+    "class_700": 6,
+    "class_7100": 8,
+    "class_800": 7
   },
   "layer_norm_eps": 1e-06,
   "mlp_ratios": [

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8cd13c55b61a107368501ab3a8b3745d064c213020e9897b6d8b773c9c23d44c
-size 109456320
+oid sha256:fbf653b6f3e09a9e2d11ba12c19dec8c7b1e1bcee8bda7f4c48f5a51a0ee4e54
+size 54745600