lithiumice commited on
Commit
7cb94e6
·
1 Parent(s): 5145ad8

add smplx misc files

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. 10_FLAME2020/FLAME_albedo_from_BFM.npz +0 -3
  2. 10_FLAME2020/FLAME_masks.pkl +0 -3
  3. 10_FLAME2020/FLAME_masks/FLAME_masks.gif +0 -3
  4. 10_FLAME2020/FLAME_masks/FLAME_masks.pkl +0 -3
  5. 10_FLAME2020/FLAME_masks/FLAME_masks.pkl.bak +0 -0
  6. 10_FLAME2020/FLAME_masks/readme +0 -2
  7. 10_FLAME2020/Readme.pdf +0 -0
  8. 10_FLAME2020/flame_dynamic_embedding.npy +0 -3
  9. 10_FLAME2020/generic_model.pkl +0 -3
  10. 10_FLAME2020/head_template.obj +0 -0
  11. 10_FLAME2020/head_template_color.obj +0 -0
  12. 10_FLAME2020/head_template_mesh.obj +0 -0
  13. 10_FLAME2020/l_eyelid.npy +0 -3
  14. 10_FLAME2020/landmark_embedding.npy +0 -3
  15. 10_FLAME2020/male_model.pkl +0 -3
  16. 10_FLAME2020/r_eyelid.npy +0 -3
  17. 10_FLAME2020/uv_mask_eyes.jpg +0 -3
  18. 10_FLAME2020/uv_template.obj +0 -0
  19. 11_PIXIE/SMPL-X__FLAME_vertex_ids.npy +0 -3
  20. 11_PIXIE/SMPLX_NEUTRAL_2020.npz +0 -3
  21. 11_PIXIE/SMPL_X_template_FLAME_uv.obj +0 -0
  22. 11_PIXIE/flame2smplx_tex_1024.npy +0 -3
  23. 11_PIXIE/pixie_model.tar +0 -3
  24. 11_PIXIE/smplx_extra_joints.yaml +0 -132
  25. 11_PIXIE/smplx_hand.obj +0 -0
  26. 11_PIXIE/smplx_tex.obj +0 -0
  27. 11_PIXIE/smplx_tex.png +0 -3
  28. 11_PIXIE/uv_face_eye_mask.png +0 -3
  29. 11_PIXIE/uv_face_mask.png +0 -3
  30. 12_vposer_v1_0/.DS_Store +0 -0
  31. 12_vposer_v1_0/._.DS_Store +0 -0
  32. 12_vposer_v1_0/._TR00_004_00_WO_accad.ini +0 -0
  33. 12_vposer_v1_0/._snapshots +0 -0
  34. 12_vposer_v1_0/._version.txt +0 -0
  35. 12_vposer_v1_0/._vposer_smpl.py +0 -0
  36. 12_vposer_v1_0/TR00_004_00_WO_accad.ini +0 -29
  37. 12_vposer_v1_0/__pycache__/vposer_smpl.cpython-37.pyc +0 -0
  38. 12_vposer_v1_0/__pycache__/vposer_smpl.cpython-39.pyc +0 -0
  39. 12_vposer_v1_0/snapshots/._TR00_E096.pt +0 -3
  40. 12_vposer_v1_0/snapshots/TR00_E096.pt +0 -3
  41. 12_vposer_v1_0/version.txt +0 -1
  42. 12_vposer_v1_0/vposer_smpl.py +0 -164
  43. 13_vposer_V02_05/V02_05.log +0 -80
  44. 13_vposer_V02_05/V02_05.yaml +0 -58
  45. 13_vposer_V02_05/snapshots/V02_05_epoch=08_val_loss=0.03.ckpt +0 -3
  46. 13_vposer_V02_05/snapshots/V02_05_epoch=13_val_loss=0.03.ckpt +0 -3
  47. 14_PyMafX/README.md +0 -2
  48. 14_PyMafX/mano_downsampling.npz +0 -3
  49. 14_PyMafX/partial_mesh/smpl_arm_vids.npz +0 -3
  50. 14_PyMafX/partial_mesh/smpl_face_vids.npz +0 -3
10_FLAME2020/FLAME_albedo_from_BFM.npz DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:c6a9986024679caa881fcfe8a49687843e9338bfe0b0bb4923dccd29e01135ad
3
- size 1258291694
 
 
 
 
10_FLAME2020/FLAME_masks.pkl DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ccefbe1ac0774ff78c68caf2c627b4abc067a6555ebeb0be5d5b0812366ab492
3
- size 215062
 
 
 
 
10_FLAME2020/FLAME_masks/FLAME_masks.gif DELETED

Git LFS Details

  • SHA256: 0021ad6d1a2e748eb80d2561ff3b9489ff12ffa346083c230b57c037f2f06f52
  • Pointer size: 132 Bytes
  • Size of remote file: 1.04 MB
10_FLAME2020/FLAME_masks/FLAME_masks.pkl DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ccefbe1ac0774ff78c68caf2c627b4abc067a6555ebeb0be5d5b0812366ab492
3
- size 215062
 
 
 
 
10_FLAME2020/FLAME_masks/FLAME_masks.pkl.bak DELETED
The diff for this file is too large to render. See raw diff
 
10_FLAME2020/FLAME_masks/readme DELETED
@@ -1,2 +0,0 @@
1
- Dictionary with vertex indices for different masks for the publicly available FLAME head model (https://flame.is.tue.mpg.de/).
2
- See the gif for a visualization of all masks.
 
 
 
10_FLAME2020/Readme.pdf DELETED
Binary file (19.3 kB)
 
10_FLAME2020/flame_dynamic_embedding.npy DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:fd046a4740f5e6837e65622d0e050273bb71bbcb1ad6ab322474013a2a23de37
3
- size 42639
 
 
 
 
10_FLAME2020/generic_model.pkl DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:efcd14cc4a69f3a3d9af8ded80146b5b6b50df3bd74cf69108213b144eba725b
3
- size 53023716
 
 
 
 
10_FLAME2020/head_template.obj DELETED
The diff for this file is too large to render. See raw diff
 
10_FLAME2020/head_template_color.obj DELETED
The diff for this file is too large to render. See raw diff
 
10_FLAME2020/head_template_mesh.obj DELETED
The diff for this file is too large to render. See raw diff
 
10_FLAME2020/l_eyelid.npy DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:fa50997166c2f884fbfc577d99fe8707966763fc3d83a1870bb786df6cc9410d
3
- size 120680
 
 
 
 
10_FLAME2020/landmark_embedding.npy DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:8095348eeafce5a02f6bd8765146307f9567a3f03b316d788a2e47336d667954
3
- size 31292
 
 
 
 
10_FLAME2020/male_model.pkl DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:5a996eaca4b56b03a0c2907f7dfa3a92680d965fd190d345a7e588f6fa017d1b
3
- size 53023716
 
 
 
 
10_FLAME2020/r_eyelid.npy DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:dd78776fed765be4468889dd7d5b54b4bf61c8efbb2869b5b91409ce798aad8d
3
- size 120680
 
 
 
 
10_FLAME2020/uv_mask_eyes.jpg DELETED

Git LFS Details

  • SHA256: 627a089a1767293670629b17dbbe337893b5ec9f003722e1fc1b740106396612
  • Pointer size: 132 Bytes
  • Size of remote file: 2.93 MB
10_FLAME2020/uv_template.obj DELETED
The diff for this file is too large to render. See raw diff
 
11_PIXIE/SMPL-X__FLAME_vertex_ids.npy DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:7e70cdc3659aae699b9732e8dd4af49106310c69b90dc83d9f73e96dbf871e49
3
- size 40312
 
 
 
 
11_PIXIE/SMPLX_NEUTRAL_2020.npz DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:bdf06146e27d92022fe5dadad3b9203373f6879eca8e4d8235359ee3ec6a5a74
3
- size 167264530
 
 
 
 
11_PIXIE/SMPL_X_template_FLAME_uv.obj DELETED
The diff for this file is too large to render. See raw diff
 
11_PIXIE/flame2smplx_tex_1024.npy DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:945d164becfe0f217e1a51356e843ef3f615278d66f6bbb9a7a8f70cb356c93b
3
- size 22043098
 
 
 
 
11_PIXIE/pixie_model.tar DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:9dade173cbe63527209bfd413b9698e8f194ac9bfb9cf11012a823b4c75ed6b9
3
- size 773605748
 
 
 
 
11_PIXIE/smplx_extra_joints.yaml DELETED
@@ -1,132 +0,0 @@
1
- head_top:
2
- bc:
3
- - 0.8277337276382795
4
- - 0.1422200962169292
5
- - 0.030046176144791284
6
- face: 2581
7
- left_big_toe:
8
- bc:
9
- - 0.0
10
- - 0.0
11
- - 1.0
12
- face: 4407
13
- left_ear:
14
- bc:
15
- - 0.0
16
- - 0.0
17
- - 1.0
18
- face: 1946
19
- left_eye:
20
- bc:
21
- - 0.0
22
- - 1.0
23
- - 0.0
24
- face: 9470
25
- left_heel:
26
- bc:
27
- - 1.0
28
- - 0.0
29
- - 0.0
30
- face: 4621
31
- left_index:
32
- bc:
33
- - 0.0
34
- - 0.0
35
- - 1.0
36
- face: 3720
37
- left_middle:
38
- bc:
39
- - 0.0
40
- - 0.0
41
- - 1.0
42
- face: 3469
43
- left_pinky:
44
- bc:
45
- - 1.0
46
- - 0.0
47
- - 0.0
48
- face: 3575
49
- left_ring:
50
- bc:
51
- - 0.0
52
- - 0.0
53
- - 1.0
54
- face: 3542
55
- left_small_toe:
56
- bc:
57
- - 1.0
58
- - 0.0
59
- - 0.0
60
- face: 4329
61
- left_thumb:
62
- bc:
63
- - 0.0
64
- - 1.0
65
- - 0.0
66
- face: 3630
67
- nose:
68
- bc:
69
- - 0.0
70
- - 1.0
71
- - 0.0
72
- face: 9041
73
- right_big_toe:
74
- bc:
75
- - 1.0
76
- - 0.0
77
- - 0.0
78
- face: 8094
79
- right_ear:
80
- bc:
81
- - 1.0
82
- - 0.0
83
- - 0.0
84
- face: 351
85
- right_eye:
86
- bc:
87
- - 1.0
88
- - 0.0
89
- - 0.0
90
- face: 10093
91
- right_heel:
92
- bc:
93
- - 1.0
94
- - 0.0
95
- - 0.0
96
- face: 8247
97
- right_index:
98
- bc:
99
- - 1.0
100
- - 0.0
101
- - 0.0
102
- face: 6919
103
- right_middle:
104
- bc:
105
- - 0.0
106
- - 0.0
107
- - 1.0
108
- face: 7050
109
- right_pinky:
110
- bc:
111
- - 1.0
112
- - 0.0
113
- - 0.0
114
- face: 7284
115
- right_ring:
116
- bc:
117
- - 0.0
118
- - 0.0
119
- - 1.0
120
- face: 7168
121
- right_small_toe:
122
- bc:
123
- - 0.0
124
- - 0.0
125
- - 1.0
126
- face: 8096
127
- right_thumb:
128
- bc:
129
- - 0.0
130
- - 0.0
131
- - 1.0
132
- face: 7370
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11_PIXIE/smplx_hand.obj DELETED
The diff for this file is too large to render. See raw diff
 
11_PIXIE/smplx_tex.obj DELETED
The diff for this file is too large to render. See raw diff
 
11_PIXIE/smplx_tex.png DELETED

Git LFS Details

  • SHA256: becb85b4222f0cf8bb981b3b4adb910423ba641f1370427eab6bd0ec5c51dae8
  • Pointer size: 132 Bytes
  • Size of remote file: 1.14 MB
11_PIXIE/uv_face_eye_mask.png DELETED

Git LFS Details

  • SHA256: ca7d777c2a8c9e1ae0bafecd59d5d938fbf4691cdfd17061140a5aa60d20007a
  • Pointer size: 130 Bytes
  • Size of remote file: 11.8 kB
11_PIXIE/uv_face_mask.png DELETED

Git LFS Details

  • SHA256: 357743bf60bfba7ecd3e5db5ae19a2bc044547aa655b05f5e341bc65bd275909
  • Pointer size: 130 Bytes
  • Size of remote file: 12.1 kB
12_vposer_v1_0/.DS_Store DELETED
Binary file (6.15 kB)
 
12_vposer_v1_0/._.DS_Store DELETED
Binary file (4.1 kB)
 
12_vposer_v1_0/._TR00_004_00_WO_accad.ini DELETED
Binary file (4.1 kB)
 
12_vposer_v1_0/._snapshots DELETED
Binary file (4.1 kB)
 
12_vposer_v1_0/._version.txt DELETED
Binary file (4.1 kB)
 
12_vposer_v1_0/._vposer_smpl.py DELETED
Binary file (4.1 kB)
 
12_vposer_v1_0/TR00_004_00_WO_accad.ini DELETED
@@ -1,29 +0,0 @@
1
- [All]
2
- adam_beta1 : 0.9
3
- base_lr : 0.005
4
- batch_size : 512
5
- best_model_fname : None
6
- cuda_id : 0
7
- data_shape : [1, 21, 3]
8
- dataset_dir : None
9
- display_model_gender : male
10
- expr_code : 004_00_WO_accad
11
- fp_precision : 32
12
- ip_avoid : False
13
- kl_coef : 0.005
14
- latentD : 32
15
- log_every_epoch : 2
16
- model_type : smpl
17
- n_workers : 10
18
- num_bodies_to_display : 10
19
- num_epochs : 100
20
- num_neurons : 512
21
- reg_coef : 0.0001
22
- remove_Zrot : True
23
- seed : 4815
24
- sm_coef : 0.01
25
- test_only : False
26
- try_num : 0
27
- use_cont_repr : True
28
- verbosity : 0
29
- work_dir : None
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12_vposer_v1_0/__pycache__/vposer_smpl.cpython-37.pyc DELETED
Binary file (4.82 kB)
 
12_vposer_v1_0/__pycache__/vposer_smpl.cpython-39.pyc DELETED
Binary file (4.84 kB)
 
12_vposer_v1_0/snapshots/._TR00_E096.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:4e2615cd1d2e78cdfac7169c6182a7352d02992336dad7329d3d97f6947fb515
3
- size 4096
 
 
 
 
12_vposer_v1_0/snapshots/TR00_E096.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:0e4ad40f922606989939d3fae6eadf82d1a8e98112dffb6e39d89d6471270d5c
3
- size 2702962
 
 
 
 
12_vposer_v1_0/version.txt DELETED
@@ -1 +0,0 @@
1
- The codenames "TR00_004_00_WO_accad" and "TR00_E096" correspond to "VPoser Version 1.0".
 
 
12_vposer_v1_0/vposer_smpl.py DELETED
@@ -1,164 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- #
3
- # Copyright (C) 2019 Max-Planck-Gesellschaft zur Förderung der Wissenschaften e.V. (MPG),
4
- # acting on behalf of its Max Planck Institute for Intelligent Systems and the
5
- # Max Planck Institute for Biological Cybernetics. All rights reserved.
6
- #
7
- # Max-Planck-Gesellschaft zur Förderung der Wissenschaften e.V. (MPG) is holder of all proprietary rights
8
- # on this computer program. You can only use this computer program if you have closed a license agreement
9
- # with MPG or you get the right to use the computer program from someone who is authorized to grant you that right.
10
- # Any use of the computer program without a valid license is prohibited and liable to prosecution.
11
- # Contact: ps-license@tuebingen.mpg.de
12
- #
13
- #
14
- # If you use this code in a research publication please consider citing the following:
15
- #
16
- # Expressive Body Capture: 3D Hands, Face, and Body from a Single Image <https://arxiv.org/abs/1904.05866>
17
- # AMASS: Archive of Motion Capture as Surface Shapes <https://arxiv.org/abs/1904.03278>
18
- #
19
- #
20
- # Code Developed by:
21
- # Nima Ghorbani <https://www.linkedin.com/in/nghorbani/>
22
- # Vassilis Choutas <https://ps.is.tuebingen.mpg.de/employees/vchoutas> for ContinousRotReprDecoder
23
- #
24
- # 2018.01.02
25
-
26
- '''
27
- A human body pose prior built with Auto-Encoding Variational Bayes
28
- '''
29
-
30
- __all__ = ['VPoser']
31
-
32
- import os, sys, shutil
33
-
34
- import torch
35
-
36
- from torch import nn
37
- from torch.nn import functional as F
38
-
39
- import numpy as np
40
-
41
- import torchgeometry as tgm
42
-
43
- class ContinousRotReprDecoder(nn.Module):
44
- def __init__(self):
45
- super(ContinousRotReprDecoder, self).__init__()
46
-
47
- def forward(self, module_input):
48
- reshaped_input = module_input.view(-1, 3, 2)
49
-
50
- b1 = F.normalize(reshaped_input[:, :, 0], dim=1)
51
-
52
- dot_prod = torch.sum(b1 * reshaped_input[:, :, 1], dim=1, keepdim=True)
53
- b2 = F.normalize(reshaped_input[:, :, 1] - dot_prod * b1, dim=-1)
54
- b3 = torch.cross(b1, b2, dim=1)
55
-
56
- return torch.stack([b1, b2, b3], dim=-1)
57
-
58
-
59
- class VPoser(nn.Module):
60
- def __init__(self, num_neurons, latentD, data_shape, use_cont_repr=True):
61
- super(VPoser, self).__init__()
62
-
63
- self.latentD = latentD
64
- self.use_cont_repr = use_cont_repr
65
-
66
- n_features = np.prod(data_shape)
67
- self.num_joints = data_shape[1]
68
-
69
- self.bodyprior_enc_bn1 = nn.BatchNorm1d(n_features)
70
- self.bodyprior_enc_fc1 = nn.Linear(n_features, num_neurons)
71
- self.bodyprior_enc_bn2 = nn.BatchNorm1d(num_neurons)
72
- self.bodyprior_enc_fc2 = nn.Linear(num_neurons, num_neurons)
73
- self.bodyprior_enc_mu = nn.Linear(num_neurons, latentD)
74
- self.bodyprior_enc_logvar = nn.Linear(num_neurons, latentD)
75
- self.dropout = nn.Dropout(p=.1, inplace=False)
76
-
77
- self.bodyprior_dec_fc1 = nn.Linear(latentD, num_neurons)
78
- self.bodyprior_dec_fc2 = nn.Linear(num_neurons, num_neurons)
79
-
80
- if self.use_cont_repr:
81
- self.rot_decoder = ContinousRotReprDecoder()
82
-
83
- self.bodyprior_dec_out = nn.Linear(num_neurons, self.num_joints* 6)
84
-
85
- def encode(self, Pin):
86
- '''
87
-
88
- :param Pin: Nx(numjoints*3)
89
- :param rep_type: 'matrot'/'aa' for matrix rotations or axis-angle
90
- :return:
91
- '''
92
- Xout = Pin.view(Pin.size(0), -1) # flatten input
93
- Xout = self.bodyprior_enc_bn1(Xout)
94
-
95
- Xout = F.leaky_relu(self.bodyprior_enc_fc1(Xout), negative_slope=.2)
96
- Xout = self.bodyprior_enc_bn2(Xout)
97
- Xout = self.dropout(Xout)
98
- Xout = F.leaky_relu(self.bodyprior_enc_fc2(Xout), negative_slope=.2)
99
- return torch.distributions.normal.Normal(self.bodyprior_enc_mu(Xout), F.softplus(self.bodyprior_enc_logvar(Xout)))
100
-
101
- def decode(self, Zin, output_type='matrot'):
102
- assert output_type in ['matrot', 'aa']
103
-
104
- Xout = F.leaky_relu(self.bodyprior_dec_fc1(Zin), negative_slope=.2)
105
- Xout = self.dropout(Xout)
106
- Xout = F.leaky_relu(self.bodyprior_dec_fc2(Xout), negative_slope=.2)
107
- Xout = self.bodyprior_dec_out(Xout)
108
- if self.use_cont_repr:
109
- Xout = self.rot_decoder(Xout)
110
- else:
111
- Xout = torch.tanh(Xout)
112
-
113
- Xout = Xout.view([-1, 1, self.num_joints, 9])
114
- if output_type == 'aa': return VPoser.matrot2aa(Xout)
115
- return Xout
116
-
117
- def forward(self, Pin, input_type='matrot', output_type='matrot'):
118
- '''
119
-
120
- :param Pin: aa: Nx1xnum_jointsx3 / matrot: Nx1xnum_jointsx9
121
- :param input_type: matrot / aa for matrix rotations or axis angles
122
- :param output_type: matrot / aa
123
- :return:
124
- '''
125
- assert output_type in ['matrot', 'aa']
126
- # if input_type == 'aa': Pin = VPoser.aa2matrot(Pin)
127
- q_z = self.encode(Pin)
128
- q_z_sample = q_z.rsample()
129
- Prec = self.decode(q_z_sample)
130
- if output_type == 'aa': Prec = VPoser.matrot2aa(Prec)
131
-
132
- #return Prec, q_z.mean, q_z.sigma
133
- return {'pose':Prec, 'mean':q_z.mean, 'std':q_z.scale}
134
-
135
- def sample_poses(self, num_poses, output_type='aa', seed=None):
136
- np.random.seed(seed)
137
- dtype = self.bodyprior_dec_fc1.weight.dtype
138
- device = self.bodyprior_dec_fc1.weight.device
139
- self.eval()
140
- with torch.no_grad():
141
- Zgen = torch.tensor(np.random.normal(0., 1., size=(num_poses, self.latentD)), dtype=dtype).to(device)
142
- return self.decode(Zgen, output_type=output_type)
143
-
144
- @staticmethod
145
- def matrot2aa(pose_matrot):
146
- '''
147
- :param pose_matrot: Nx1xnum_jointsx9
148
- :return: Nx1xnum_jointsx3
149
- '''
150
- batch_size = pose_matrot.size(0)
151
- homogen_matrot = F.pad(pose_matrot.view(-1, 3, 3), [0,1])
152
- pose = tgm.rotation_matrix_to_angle_axis(homogen_matrot).view(batch_size, 1, -1, 3).contiguous()
153
- return pose
154
-
155
- @staticmethod
156
- def aa2matrot(pose):
157
- '''
158
- :param Nx1xnum_jointsx3
159
- :return: pose_matrot: Nx1xnum_jointsx9
160
- '''
161
- batch_size = pose.size(0)
162
- pose_body_matrot = tgm.angle_axis_to_rotation_matrix(pose.reshape(-1, 3))[:, :3, :3].contiguous().view(batch_size, 1, -1, 9)
163
- return pose_body_matrot
164
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13_vposer_V02_05/V02_05.log DELETED
@@ -1,80 +0,0 @@
1
- [V02_05] -- VPoser dataset already exists at /ps/scratch/human_body_prior/VPoser/data/smpl/V02_03
2
- [V02_05] -- Total Trainable Parameters Count in vp_model is 0.94 M.
3
- [V02_05] -- Epoch 0: val_loss:0.51
4
- [V02_05] -- lr is [0.001]
5
- [V02_05] -- Created a git archive backup at /ps/scratch/human_body_prior/VPoser/training_experiments/V02_05/code/vposer_2020_10_23_17_10_39.tar.gz
6
- [V02_05] -- Epoch 0: val_loss:0.03
7
- [V02_05] -- lr is [0.001]
8
- [V02_05] -- Epoch 1: val_loss:0.03
9
- [V02_05] -- lr is [0.001]
10
- [V02_05] -- Epoch 2: val_loss:0.03
11
- [V02_05] -- lr is [0.001]
12
- [V02_05] -- Epoch 3: val_loss:0.03
13
- [V02_05] -- lr is [0.001]
14
- [V02_05] -- Epoch 4: val_loss:0.03
15
- [V02_05] -- lr is [0.001]
16
- [V02_05] -- Epoch 5: val_loss:0.03
17
- [V02_05] -- lr is [0.001]
18
- [V02_05] -- Epoch 6: val_loss:0.03
19
- [V02_05] -- lr is [0.001]
20
- [V02_05] -- Epoch 7: val_loss:0.03
21
- [V02_05] -- lr is [0.001]
22
- [V02_05] -- Epoch 8: val_loss:0.03
23
- [V02_05] -- lr is [0.001]
24
- [V02_05] -- Epoch 9: val_loss:0.03
25
- [V02_05] -- lr is [0.001]
26
- [V02_05] -- Epoch 10: val_loss:0.03
27
- [V02_05] -- lr is [0.001]
28
- [V02_05] -- Epoch 11: val_loss:0.03
29
- [V02_05] -- lr is [0.001]
30
- [V02_05] -- Epoch 12: val_loss:0.03
31
- [V02_05] -- lr is [0.001]
32
- [V02_05] -- Epoch 13: val_loss:0.03
33
- [V02_05] -- lr is [0.001]
34
- [V02_05] -- Epoch 14: val_loss:0.03
35
- [V02_05] -- lr is [0.001]
36
- [V02_05] -- Epoch 15: val_loss:0.04
37
- [V02_05] -- lr is [0.0001]
38
- [V02_05] -- Epoch 16: val_loss:0.04
39
- [V02_05] -- lr is [0.0001]
40
- [V02_05] -- Epoch 17: val_loss:0.04
41
- [V02_05] -- lr is [0.0001]
42
- [V02_05] -- Epoch 18: val_loss:0.04
43
- [V02_05] -- lr is [0.0001]
44
- [V02_05] -- VPoser dataset already exists at /ps/scratch/human_body_prior/VPoser/data/smpl/V02_03
45
- [V02_05] -- Total Trainable Parameters Count in vp_model is 0.94 M.
46
- [V02_05] -- Epoch 0: val_loss:0.03
47
- [V02_05] -- lr is [0.001]
48
- [V02_05] -- Created a git archive backup at /ps/scratch/human_body_prior/VPoser/training_experiments/V02_05/code/vposer_2020_10_23_18_48_23.tar.gz
49
- [V02_05] -- Epoch 9: val_loss:0.03
50
- [V02_05] -- lr is [0.001]
51
- [V02_05] -- Epoch 10: val_loss:0.03
52
- [V02_05] -- lr is [0.001]
53
- [V02_05] -- Epoch 11: val_loss:0.03
54
- [V02_05] -- lr is [0.001]
55
- [V02_05] -- Epoch 12: val_loss:0.03
56
- [V02_05] -- lr is [0.001]
57
- [V02_05] -- Epoch 13: val_loss:0.03
58
- [V02_05] -- lr is [0.0001]
59
- [V02_05] -- Epoch 14: val_loss:0.03
60
- [V02_05] -- lr is [0.0001]
61
- [V02_05] -- Epoch 15: val_loss:0.04
62
- [V02_05] -- lr is [0.0001]
63
- [V02_05] -- Epoch 16: val_loss:0.04
64
- [V02_05] -- lr is [0.0001]
65
- [V02_05] -- Epoch 17: val_loss:0.04
66
- [V02_05] -- lr is [0.0001]
67
- [V02_05] -- Epoch 18: val_loss:0.04
68
- [V02_05] -- lr is [0.0001]
69
- [V02_05] -- Epoch 19: val_loss:0.04
70
- [V02_05] -- lr is [0.0001]
71
- [V02_05] -- Epoch 20: val_loss:0.04
72
- [V02_05] -- lr is [1e-05]
73
- [V02_05] -- Epoch 21: val_loss:0.04
74
- [V02_05] -- lr is [1e-05]
75
- [V02_05] -- Epoch 22: val_loss:0.04
76
- [V02_05] -- lr is [1e-05]
77
- [V02_05] -- Epoch 23: val_loss:0.04
78
- [V02_05] -- lr is [1e-05]
79
- [V02_05] -- Epoch 23 - Finished training at 2020_10_23_19_40_46 after 0:52:23
80
- [V02_05] -- best_model_fname: /ps/scratch/human_body_prior/VPoser/training_experiments/V02_05/snapshots/V02_05_epoch=13_val_loss=0.03.ckpt
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13_vposer_V02_05/V02_05.yaml DELETED
@@ -1,58 +0,0 @@
1
- body_model:
2
- bm_path: /ps/project/common/moshpp/smplx/locked_head/model_6_merged_exp_hands_fixed_eyes/neutral/model.npz
3
- gender: neutral
4
- data_parms:
5
- amass_dir: /ps/scratch/amass/20200508/mosh_smplx_neutral
6
- amass_splits:
7
- test:
8
- - BMLrub_test
9
- train:
10
- - CMU
11
- - BMLrub_train
12
- vald:
13
- - BMLrub_vald
14
- num_timeseq_frames: {}
15
- num_workers: 5
16
- general:
17
- dataset_basedir: /ps/scratch/human_body_prior/VPoser/data/smpl
18
- dataset_id: V02_03
19
- expr_id: V02_05
20
- rnd_seed: 100
21
- verbosity: 0
22
- work_basedir: /ps/scratch/human_body_prior/VPoser/training_experiments
23
- logging:
24
- best_model_fname: null
25
- dataset_dir: /ps/scratch/human_body_prior/VPoser/data/smpl/V02_03
26
- expr_msg: '[V02_05] batch_size = 128.'
27
- num_bodies_to_display: 25
28
- render_during_training: true
29
- work_dir: /ps/scratch/human_body_prior/VPoser/training_experiments/V02_05
30
- model_params:
31
- latentD: 32
32
- num_neurons: 512
33
- train_parms:
34
- batch_size: 128
35
- early_stopping:
36
- min_delta: 0.0
37
- mode: min
38
- monitor: val_loss
39
- patience: 10
40
- verbose: true
41
- gen_optimizer:
42
- args:
43
- lr: 0.001
44
- weight_decay: 1.0e-05
45
- type: Adam
46
- keep_extra_loss_terms_until_epoch: 15
47
- loss_weights:
48
- loss_jtr_wt: 2
49
- loss_kl_wt: 0.005
50
- loss_matrot_wt: 2
51
- loss_rec_wt: 4
52
- lr_scheduler:
53
- args:
54
- patience: 5
55
- verbose: true
56
- type: ReduceLROnPlateau
57
- num_epochs: 100
58
- restore_optimizer: false
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13_vposer_V02_05/snapshots/V02_05_epoch=08_val_loss=0.03.ckpt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:a914aef21a5c1f4e4d1a616f28def84bc2550c4e7312a689a64e9a35eb7ebee3
3
- size 88644482
 
 
 
 
13_vposer_V02_05/snapshots/V02_05_epoch=13_val_loss=0.03.ckpt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ce1f8e7804e645103a877855be45d57588940545d969cbc1ce5fd96841427b3c
3
- size 88644490
 
 
 
 
14_PyMafX/README.md DELETED
@@ -1,2 +0,0 @@
1
- # Extra data
2
- You need to get extra data by running the ```fetch_data.sh``` script.
 
 
 
14_PyMafX/mano_downsampling.npz DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:db2b23b0ede7c34039f1d8960e2e839552e0f58d4b34a8612b4065d8a47f9c80
3
- size 176509
 
 
 
 
14_PyMafX/partial_mesh/smpl_arm_vids.npz DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:0fc3729a4519abe0a98be581e6c0f72d9cd1045e66b40cb28da18abe29ce2979
3
- size 160216
 
 
 
 
14_PyMafX/partial_mesh/smpl_face_vids.npz DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:10a733ce250bfe5af188f245aa1eb404d91ab4d36d5f77fdb0d91acbfc08410e
3
- size 68152