lithiumice
commited on
Commit
·
7cb94e6
1
Parent(s):
5145ad8
add smplx misc files
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- 10_FLAME2020/FLAME_albedo_from_BFM.npz +0 -3
- 10_FLAME2020/FLAME_masks.pkl +0 -3
- 10_FLAME2020/FLAME_masks/FLAME_masks.gif +0 -3
- 10_FLAME2020/FLAME_masks/FLAME_masks.pkl +0 -3
- 10_FLAME2020/FLAME_masks/FLAME_masks.pkl.bak +0 -0
- 10_FLAME2020/FLAME_masks/readme +0 -2
- 10_FLAME2020/Readme.pdf +0 -0
- 10_FLAME2020/flame_dynamic_embedding.npy +0 -3
- 10_FLAME2020/generic_model.pkl +0 -3
- 10_FLAME2020/head_template.obj +0 -0
- 10_FLAME2020/head_template_color.obj +0 -0
- 10_FLAME2020/head_template_mesh.obj +0 -0
- 10_FLAME2020/l_eyelid.npy +0 -3
- 10_FLAME2020/landmark_embedding.npy +0 -3
- 10_FLAME2020/male_model.pkl +0 -3
- 10_FLAME2020/r_eyelid.npy +0 -3
- 10_FLAME2020/uv_mask_eyes.jpg +0 -3
- 10_FLAME2020/uv_template.obj +0 -0
- 11_PIXIE/SMPL-X__FLAME_vertex_ids.npy +0 -3
- 11_PIXIE/SMPLX_NEUTRAL_2020.npz +0 -3
- 11_PIXIE/SMPL_X_template_FLAME_uv.obj +0 -0
- 11_PIXIE/flame2smplx_tex_1024.npy +0 -3
- 11_PIXIE/pixie_model.tar +0 -3
- 11_PIXIE/smplx_extra_joints.yaml +0 -132
- 11_PIXIE/smplx_hand.obj +0 -0
- 11_PIXIE/smplx_tex.obj +0 -0
- 11_PIXIE/smplx_tex.png +0 -3
- 11_PIXIE/uv_face_eye_mask.png +0 -3
- 11_PIXIE/uv_face_mask.png +0 -3
- 12_vposer_v1_0/.DS_Store +0 -0
- 12_vposer_v1_0/._.DS_Store +0 -0
- 12_vposer_v1_0/._TR00_004_00_WO_accad.ini +0 -0
- 12_vposer_v1_0/._snapshots +0 -0
- 12_vposer_v1_0/._version.txt +0 -0
- 12_vposer_v1_0/._vposer_smpl.py +0 -0
- 12_vposer_v1_0/TR00_004_00_WO_accad.ini +0 -29
- 12_vposer_v1_0/__pycache__/vposer_smpl.cpython-37.pyc +0 -0
- 12_vposer_v1_0/__pycache__/vposer_smpl.cpython-39.pyc +0 -0
- 12_vposer_v1_0/snapshots/._TR00_E096.pt +0 -3
- 12_vposer_v1_0/snapshots/TR00_E096.pt +0 -3
- 12_vposer_v1_0/version.txt +0 -1
- 12_vposer_v1_0/vposer_smpl.py +0 -164
- 13_vposer_V02_05/V02_05.log +0 -80
- 13_vposer_V02_05/V02_05.yaml +0 -58
- 13_vposer_V02_05/snapshots/V02_05_epoch=08_val_loss=0.03.ckpt +0 -3
- 13_vposer_V02_05/snapshots/V02_05_epoch=13_val_loss=0.03.ckpt +0 -3
- 14_PyMafX/README.md +0 -2
- 14_PyMafX/mano_downsampling.npz +0 -3
- 14_PyMafX/partial_mesh/smpl_arm_vids.npz +0 -3
- 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
|
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
|
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
|
11_PIXIE/uv_face_eye_mask.png
DELETED
Git LFS Details
|
11_PIXIE/uv_face_mask.png
DELETED
Git LFS Details
|
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
|
|
|
|
|
|
|
|