Upload MaskFormerForInstanceSegmentation

config.json

@@ -7,21 +7,21 @@
     "depths": [
       2,
       2,
-      18,
+      6,
       2
     ],
     "drop_path_rate": 0.3,
-    "embed_dim": 128,
+    "embed_dim": 96,
     "encoder_stride": 32,
-    "hidden_size": 1024,
-    "image_size": 384,
+    "hidden_size": 768,
+    "image_size": 224,
     "in_channels": 3,
     "model_type": "maskformer-swin",
     "num_heads": [
-      4,
-      8,
-      16,
-      32
+      3,
+      6,
+      12,
+      24
     ],
     "out_features": [
       "stage1",
@@ -36,8 +36,8 @@
       4
     ],
     "path_norm": true,
-    "pretrain_img_size": 384,
-    "window_size": 12
+    "pretrain_img_size": 224,
+    "window_size": 7
   },
   "backbone_kwargs": null,
   "ce_weight": 1.0,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7057977befb40fae48a1e9179b8785d7852ccb7933e84c4f9838d925d4816507
-size 411253576
+oid sha256:4fe4b48be9d861df8f5a953322f25ae2ca3ab417a97e48238654f3d3b974897a
+size 167175760