Upload folder using huggingface_hub

config.json

@@ -0,0 +1,53 @@
+{
+  "_commit_hash": "bf486d01a7d9e3dbcc8400f7942835caf0eaf76e",
+  "_name_or_path": "hustvl/vitmatte-base-composition-1k",
+  "architectures": [
+    "VitMatteForImageMatting"
+  ],
+  "backbone_config": {
+    "image_size": 512,
+    "model_type": "vitdet",
+    "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": 768,
+  "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:f6978437f5068849bcbf49b1f4e37b90aaca5155744e9fde4e6c689f70c2b9ee
+size 387371620

onnx/model_quantized.onnx

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

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