Hugging Face's logo

LukaMarki
/
repogen-vitpose-small

PyTorch
vitpose
Model card Files
xet
Community
LukaMarki commited on
Commit
2079290
·
verified ·
1 Parent(s): c372ac5

Upload 2 files

Browse files
Files changed (2) hide show
  1. config.json +143 -0
  2. preprocessor_config.json +22 -0
config.json ADDED
@@ -0,0 +1,143 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "VitPoseForPoseEstimation"
4
+ ],
5
+ "backbone": null,
6
+ "backbone_config": {
7
+ "hidden_size": 384,
8
+ "model_type": "vitpose_backbone",
9
+ "num_experts": 6,
10
+ "out_features": [
11
+ "stage12"
12
+ ],
13
+ "out_indices": [
14
+ 12
15
+ ],
16
+ "part_features": 96
17
+ },
18
+ "backbone_kwargs": null,
19
+ "edges": [
20
+ [
21
+ 15,
22
+ 13
23
+ ],
24
+ [
25
+ 13,
26
+ 11
27
+ ],
28
+ [
29
+ 16,
30
+ 14
31
+ ],
32
+ [
33
+ 14,
34
+ 12
35
+ ],
36
+ [
37
+ 11,
38
+ 12
39
+ ],
40
+ [
41
+ 5,
42
+ 11
43
+ ],
44
+ [
45
+ 6,
46
+ 12
47
+ ],
48
+ [
49
+ 5,
50
+ 6
51
+ ],
52
+ [
53
+ 5,
54
+ 7
55
+ ],
56
+ [
57
+ 6,
58
+ 8
59
+ ],
60
+ [
61
+ 7,
62
+ 9
63
+ ],
64
+ [
65
+ 8,
66
+ 10
67
+ ],
68
+ [
69
+ 1,
70
+ 2
71
+ ],
72
+ [
73
+ 0,
74
+ 1
75
+ ],
76
+ [
77
+ 0,
78
+ 2
79
+ ],
80
+ [
81
+ 1,
82
+ 3
83
+ ],
84
+ [
85
+ 2,
86
+ 4
87
+ ],
88
+ [
89
+ 3,
90
+ 5
91
+ ],
92
+ [
93
+ 4,
94
+ 6
95
+ ]
96
+ ],
97
+ "id2label": {
98
+ "0": "Nose",
99
+ "1": "L_Eye",
100
+ "2": "R_Eye",
101
+ "3": "L_Ear",
102
+ "4": "R_Ear",
103
+ "5": "L_Shoulder",
104
+ "6": "R_Shoulder",
105
+ "7": "L_Elbow",
106
+ "8": "R_Elbow",
107
+ "9": "L_Wrist",
108
+ "10": "R_Wrist",
109
+ "11": "L_Hip",
110
+ "12": "R_Hip",
111
+ "13": "L_Knee",
112
+ "14": "R_Knee",
113
+ "15": "L_Ankle",
114
+ "16": "R_Ankle"
115
+ },
116
+ "initializer_range": 0.02,
117
+ "label2id": {
118
+ "L_Ankle": 15,
119
+ "L_Ear": 3,
120
+ "L_Elbow": 7,
121
+ "L_Eye": 1,
122
+ "L_Hip": 11,
123
+ "L_Knee": 13,
124
+ "L_Shoulder": 5,
125
+ "L_Wrist": 9,
126
+ "Nose": 0,
127
+ "R_Ankle": 16,
128
+ "R_Ear": 4,
129
+ "R_Elbow": 8,
130
+ "R_Eye": 2,
131
+ "R_Hip": 12,
132
+ "R_Knee": 14,
133
+ "R_Shoulder": 6,
134
+ "R_Wrist": 10
135
+ },
136
+ "model_type": "vitpose",
137
+ "scale_factor": 4,
138
+ "torch_dtype": "float32",
139
+ "transformers_version": "4.49.0.dev0",
140
+ "use_pretrained_backbone": false,
141
+ "use_simple_decoder": false,
142
+ "use_timm_backbone": false
143
+ }
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
+ }