Upload folder using huggingface_hub

config.json

@@ -0,0 +1,55 @@
+{
+  "_commit_hash": "25c4d0d9954a8205951fab1265a85ee71f44a4b9",
+  "_name_or_path": "hustvl/vitmatte-small-distinctions-646",
+  "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:21306ed8c990f563abaa0d8b58936ab29a2288c7f4a92a9d44db3d08f8dcc6d7
+size 103866744

onnx/model_quantized.onnx

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

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": [
+                "ConstantOfShape",
+                "Relu",
+                "Constant",
+                "Conv",
+                "Shape",
+                "Add",
+                "Mul",
+                "Erf",
+                "Unsqueeze",
+                "MatMul",
+                "ReduceMean",
+                "Sub",
+                "Squeeze",
+                "Mod",
+                "Cast",
+                "Resize",
+                "Concat",
+                "Slice",
+                "Gather",
+                "Einsum",
+                "Div",
+                "Split",
+                "Pad",
+                "Sigmoid",
+                "Transpose",
+                "Softmax",
+                "Sqrt",
+                "Reshape",
+                "Pow",
+                "Range",
+                "Identity"
+            ],
+            "weight_type": "QUInt8"
+        }
+    }
+}