Upload MaskFormerForInstanceSegmentation

config.json

@@ -6,20 +6,20 @@
     "depths": [
       2,
       2,
-      18,
+      6,
       2
     ],
     "drop_path_rate": 0.3,
-    "embed_dim": 128,
-    "hidden_size": 1024,
-    "image_size": 384,
+    "embed_dim": 96,
+    "hidden_size": 768,
+    "image_size": 224,
     "in_channels": 3,
     "model_type": "swin",
     "num_heads": [
-      4,
-      8,
-      16,
-      32
+      3,
+      6,
+      12,
+      24
     ],
     "out_features": [
       "stage4"
@@ -28,8 +28,8 @@
       4
     ],
     "path_norm": true,
-    "pretrain_img_size": 384,
-    "window_size": 12
+    "pretrain_img_size": 224,
+    "window_size": 7
   },
   "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:6e38c5b74fe3afaa08621bfa29dcd9760f8874e158d0c9f7b35e53cd0e528fa4
-size 411354720
+oid sha256:b5becd6065e0439f0e4404a717556bc5545b9ab1c5e01cf4130dcb557ca70d03
+size 167276904