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:6a3e8b551dde67266bcf7df2fa22aa421c29f1d4726570bf97bb38f5d570be56
-size 167328504
+oid sha256:1df51b7ee99c53093a23e4a5294a19d03ab269241cb7c15950da44859699a99f
+size 411388776