Upload MaskFormerForInstanceSegmentation

config.json

@@ -6,20 +6,20 @@
     "depths": [
       2,
       2,
-      6,
+      18,
       2
     ],
     "drop_path_rate": 0.3,
-    "embed_dim": 96,
-    "hidden_size": 768,
-    "image_size": 224,
+    "embed_dim": 128,
+    "hidden_size": 1024,
+    "image_size": 384,
     "in_channels": 3,
     "model_type": "swin",
     "num_heads": [
-      3,
-      6,
-      12,
-      24
+      4,
+      8,
+      16,
+      32
     ],
     "out_features": [
       "stage4"
@@ -28,8 +28,8 @@
       4
     ],
     "path_norm": true,
-    "pretrain_img_size": 224,
-    "window_size": 7
+    "pretrain_img_size": 384,
+    "window_size": 12
   },
   "ce_weight": 1.0,
   "cross_entropy_weight": 1.0,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bef44c56ed52790bbc0655e6b49aeefd86828f0d804672200fd45ea5d889d8ac
-size 167328504
+oid sha256:6e38c5b74fe3afaa08621bfa29dcd9760f8874e158d0c9f7b35e53cd0e528fa4
+size 411354720