body-pose-animation/example_pose.py
2021-02-07 17:40:13 +01:00

40 lines
675 B
Python

# library imports
import numpy as np
# local imports
from renderer import DefaultRenderer
from model import SMPLyModel
from utils.general import load_config
# this a simple pose playground with a async renderer for quick prototyping
# load and select sample
config = load_config()
model = SMPLyModel.model_from_conf(config)
# try changing angles
pose = model.body_pose
# left elbow
pose[:, 56] = -np.pi / 2
# right elbow
pose[:, 53] = np.pi / 2
# left knee
pose[:, 12] = np.pi / 2
# right knee
pose[:, 9] = np.pi / 2
# right knee
pose[:, 24] = np.pi / 2
model_out = model(
pose=pose
)
r = DefaultRenderer()
r.setup(
model,
model_out=model_out
)
r.start()