Upload folder using huggingface_hub

config.json

@@ -0,0 +1,55 @@
+{
+  "_commit_hash": "03f5646d1ed954c462f3837123fba723dfd1b3d5",
+  "_name_or_path": "hustvl/vitmatte-small-composition-1k",
+  "architectures": [
+    "VitMatteForImageMatting"
+  ],
+  "backbone_config": {
+    "hidden_size": 384,
+    "image_size": 512,
+    "model_type": "vitdet",
+    "num_attention_heads": 6,
+    "num_channels": 4,
+    "out_features": [
+      "stage12"
+    ],
+    "out_indices": [
+      12
+    ],
+    "residual_block_indices": [
+      2,
+      5,
+      8,
+      11
+    ],
+    "use_relative_position_embeddings": true,
+    "window_block_indices": [
+      0,
+      1,
+      3,
+      4,
+      6,
+      7,
+      9,
+      10
+    ],
+    "window_size": 14
+  },
+  "batch_norm_eps": 1e-05,
+  "convstream_hidden_sizes": [
+    48,
+    96,
+    192
+  ],
+  "fusion_hidden_sizes": [
+    256,
+    128,
+    64,
+    32
+  ],
+  "hidden_size": 384,
+  "initializer_range": 0.02,
+  "model_type": "vitmatte",
+  "torch_dtype": "float32",
+  "transformers_version": null
+}

onnx/model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bf28d2e0be2c073286e88d60ad649d7123da2749a2d99133fd1098d5887e0225
+size 103885865

onnx/model_quantized.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8216103d998864e122333e17d7d346feb1d6558d96bf08695a9838dba4f114e0
+size 27499369

preprocessor_config.json

@@ -0,0 +1,18 @@
+{
+  "do_normalize": true,
+  "do_pad": true,
+  "do_rescale": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "VitMatteImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "rescale_factor": 0.00392156862745098,
+  "size_divisibility": 32
+}

quantize_config.json

@@ -0,0 +1,42 @@
+{
+    "per_channel": true,
+    "reduce_range": true,
+    "per_model_config": {
+        "model": {
+            "op_types": [
+                "ReduceMean",
+                "Div",
+                "Resize",
+                "Softmax",
+                "Sub",
+                "Constant",
+                "Relu",
+                "ConstantOfShape",
+                "Sqrt",
+                "Erf",
+                "Split",
+                "Range",
+                "Add",
+                "Shape",
+                "MatMul",
+                "Unsqueeze",
+                "Einsum",
+                "Reshape",
+                "Mul",
+                "Concat",
+                "Slice",
+                "Conv",
+                "Pad",
+                "Identity",
+                "Sigmoid",
+                "Gather",
+                "Pow",
+                "Mod",
+                "Transpose",
+                "Squeeze",
+                "Cast"
+            ],
+            "weight_type": "QUInt8"
+        }
+    }
+}