19 lines
428 B
Python
19 lines
428 B
Python
import os
|
|
|
|
os.environ["CUDA_VISIBLE_DEVICES"] = '6'
|
|
|
|
import numpy as np
|
|
from smpl_tf import SMPL
|
|
from utils import plot_mesh
|
|
|
|
if __name__ == '__main__':
|
|
smpl_tf : SMPL = SMPL('model_files')
|
|
|
|
betas = np.zeros((1, 11), dtype=np.float32)
|
|
poses = np.zeros((1, 72), dtype=np.float32)
|
|
R = smpl_tf.run_rodrigues(poses)
|
|
posed, _, _ = smpl_tf.run(R, betas)
|
|
|
|
plot_mesh(posed[0].numpy(), smpl_tf.f).show()
|
|
|
|
pass |