Hugging Face's logo

onnx-community
/
vitpose-base-simple

Keypoint Detection
Transformers.js
ONNX
vitpose
Model card Files
xet
Community
Xenova HF Staff commited on
Commit
4d7fa11
·
verified ·
1 Parent(s): 7a84889

Upload folder using huggingface_hub

Browse files
Files changed (11) hide show
  1. config.json +73 -0
  2. onnx/model.onnx +3 -0
  3. onnx/model_bnb4.onnx +3 -0
  4. onnx/model_fp16.onnx +3 -0
  5. onnx/model_int8.onnx +3 -0
  6. onnx/model_q4.onnx +3 -0
  7. onnx/model_q4f16.onnx +3 -0
  8. onnx/model_quantized.onnx +3 -0
  9. onnx/model_uint8.onnx +3 -0
  10. preprocessor_config.json +22 -0
  11. quantize_config.json +17 -0
config.json ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_attn_implementation_autoset": true,
3
+ "_name_or_path": "nielsr/vitpose-base-simple",
4
+ "architectures": [
5
+ "ViTPoseForPoseEstimation"
6
+ ],
7
+ "backbone": null,
8
+ "backbone_config": {
9
+ "model_type": "vitpose_backbone",
10
+ "out_features": [
11
+ "stage12"
12
+ ],
13
+ "out_indices": [
14
+ 12
15
+ ]
16
+ },
17
+ "backbone_hidden_size": 768,
18
+ "backbone_kwargs": null,
19
+ "id2label": {
20
+ "0": "LABEL_0",
21
+ "1": "LABEL_1",
22
+ "2": "LABEL_2",
23
+ "3": "LABEL_3",
24
+ "4": "LABEL_4",
25
+ "5": "LABEL_5",
26
+ "6": "LABEL_6",
27
+ "7": "LABEL_7",
28
+ "8": "LABEL_8",
29
+ "9": "LABEL_9",
30
+ "10": "LABEL_10",
31
+ "11": "LABEL_11",
32
+ "12": "LABEL_12",
33
+ "13": "LABEL_13",
34
+ "14": "LABEL_14",
35
+ "15": "LABEL_15",
36
+ "16": "LABEL_16"
37
+ },
38
+ "image_size": [
39
+ 256,
40
+ 192
41
+ ],
42
+ "initializer_range": 0.02,
43
+ "label2id": {
44
+ "LABEL_0": 0,
45
+ "LABEL_1": 1,
46
+ "LABEL_10": 10,
47
+ "LABEL_11": 11,
48
+ "LABEL_12": 12,
49
+ "LABEL_13": 13,
50
+ "LABEL_14": 14,
51
+ "LABEL_15": 15,
52
+ "LABEL_16": 16,
53
+ "LABEL_2": 2,
54
+ "LABEL_3": 3,
55
+ "LABEL_4": 4,
56
+ "LABEL_5": 5,
57
+ "LABEL_6": 6,
58
+ "LABEL_7": 7,
59
+ "LABEL_8": 8,
60
+ "LABEL_9": 9
61
+ },
62
+ "model_type": "vitpose",
63
+ "patch_size": [
64
+ 16,
65
+ 16
66
+ ],
67
+ "scale_factor": 4,
68
+ "torch_dtype": "float32",
69
+ "transformers_version": "4.47.0.dev0",
70
+ "use_pretrained_backbone": false,
71
+ "use_simple_decoder": true,
72
+ "use_timm_backbone": false
73
+ }
onnx/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47af35f9acd4f6b36323c145c55e0618a5460a242eb28105ac30d34ddb1d584f
3
+ size 343893670
onnx/model_bnb4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5f6f806c316bb779b293abf33d9ea3ab46c90fb6109146812dc56bc94542689f
3
+ size 51941992
onnx/model_fp16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f8eb8a49cab006b6b74317360809f02c45832aa1098cad148d3a31112651865
3
+ size 172069880
onnx/model_int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a50dd93dc3246a90049ec5df2c92c80c0b46cdd7f832cc0c2edaf6a9f4d6ed5
3
+ size 87479939
onnx/model_q4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c8ef2bf82220abc6510a1f3bc171e44390a85dcbbec6d9baa504dfc4e4dfdd10
3
+ size 57249880
onnx/model_q4f16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7b533788cdb2f97db1c1f622e9c08f5f4eee4d08f7891f7c2c8f2d3aed42be6
3
+ size 49987083
onnx/model_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:688e70371392881176137988f3fec3e0e8da80a24ea5a82f6be3c77254db6df3
3
+ size 87479939
onnx/model_uint8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:688e70371392881176137988f3fec3e0e8da80a24ea5a82f6be3c77254db6df3
3
+ size 87479939
preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_affine_transform": true,
3
+ "do_normalize": true,
4
+ "do_rescale": true,
5
+ "image_mean": [
6
+ 0.485,
7
+ 0.456,
8
+ 0.406
9
+ ],
10
+ "image_processor_type": "VitPoseImageProcessor",
11
+ "image_std": [
12
+ 0.229,
13
+ 0.224,
14
+ 0.225
15
+ ],
16
+ "normalize_factor": 200.0,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 256,
20
+ "width": 192
21
+ }
22
+ }
quantize_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "modes": [
3
+ "fp16",
4
+ "q8",
5
+ "int8",
6
+ "uint8",
7
+ "q4",
8
+ "q4f16",
9
+ "bnb4"
10
+ ],
11
+ "per_channel": true,
12
+ "reduce_range": true,
13
+ "block_size": null,
14
+ "is_symmetric": true,
15
+ "accuracy_level": null,
16
+ "quant_type": 1
17
+ }