Upload 3 files

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "nvidia/mit-b2",
+  "_name_or_path": "nvidia/mit-b3",
   "architectures": [
     "SegformerForSemanticSegmentation"
   ],
@@ -9,7 +9,7 @@
   "depths": [
     3,
     4,
-    6,
+    18,
     3
   ],
   "downsampling_rates": [
@@ -30,14 +30,6 @@
   "id2label": {
     "0": "Background",
     "1": "Hat",
-    "2": "Hair",
-    "3": "Sunglasses",
-    "4": "Upper-clothes",
-    "5": "Skirt",
-    "6": "Pants",
-    "7": "Dress",
-    "8": "Belt",
-    "9": "Left-shoe",
     "10": "Right-shoe",
     "11": "Face",
     "12": "Left-leg",
@@ -45,29 +37,37 @@
     "14": "Left-arm",
     "15": "Right-arm",
     "16": "Bag",
-    "17": "Scarf"
+    "17": "Scarf",
+    "2": "Hair",
+    "3": "Sunglasses",
+    "4": "Upper-clothes",
+    "5": "Skirt",
+    "6": "Pants",
+    "7": "Dress",
+    "8": "Belt",
+    "9": "Left-shoe"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
-    "Background": 0,
-    "Bag": 16,
-    "Belt": 8,
-    "Dress": 7,
-    "Face": 11,
-    "Hair": 2,
-    "Hat": 1,
-    "Left-arm": 14,
-    "Left-leg": 12,
-    "Left-shoe": 9,
-    "Pants": 6,
-    "Right-arm": 15,
-    "Right-leg": 13,
-    "Right-shoe": 10,
-    "Scarf": 17,
-    "Skirt": 5,
-    "Sunglasses": 3,
-    "Upper-clothes": 4
+    "Background": "0",
+    "Bag": "16",
+    "Belt": "8",
+    "Dress": "7",
+    "Face": "11",
+    "Hair": "2",
+    "Hat": "1",
+    "Left-arm": "14",
+    "Left-leg": "12",
+    "Left-shoe": "9",
+    "Pants": "6",
+    "Right-arm": "15",
+    "Right-leg": "13",
+    "Right-shoe": "10",
+    "Scarf": "17",
+    "Skirt": "5",
+    "Sunglasses": "3",
+    "Upper-clothes": "4"
   },
   "layer_norm_eps": 1e-06,
   "mlp_ratios": [
@@ -106,5 +106,5 @@
     2
   ],
   "torch_dtype": "float32",
-  "transformers_version": "4.24.0"
+  "transformers_version": "4.38.1"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8f86fd90c567afd4370b3cc3a7e81ed767a632b2832a738331af660acc0c4c68
-size 109493236
+oid sha256:f70ae566c5773fb335796ebaa8acc924ac25eb97222c2b2967d44d2fc11568e6
+size 189029000

preprocessor_config.json

@@ -1,18 +1,23 @@
 {
   "do_normalize": true,
+  "do_reduce_labels": false,
+  "do_rescale": true,
   "do_resize": true,
-  "feature_extractor_type": "SegformerFeatureExtractor",
   "image_mean": [
     0.485,
     0.456,
     0.406
   ],
+  "image_processor_type": "SegformerImageProcessor",
   "image_std": [
     0.229,
     0.224,
     0.225
   ],
-  "reduce_labels": false,
   "resample": 2,
-  "size": 512
-}
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 512,
+    "width": 512
+  }
+}