File size: 1,828 Bytes
4409449 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
import os
import argparse
import numpy as np
from scripts.hybrik_loc2rot import HybrIKJointsToRotmat
from scripts.pyrender import SMPLRender
import cv2
from scipy.spatial.transform import Rotation as RRR
parser = argparse.ArgumentParser(
description='Render a SMPL video by a j3ds npy file.')
parser.add_argument('--input', type=str, default='', help='the npy file path')
parser.add_argument('--render',
type=int,
default=1,
help='render the video if 1')
args = parser.parse_args()
input_path = args.input
output_npy_path = args.input.replace('.npy', '_pose.npy')
data = np.load(input_path)
data = data - data[0, 0]
pose_generator = HybrIKJointsToRotmat()
pose = pose_generator(data)
pose = np.concatenate(
[pose, np.stack([np.stack([np.eye(3)] * pose.shape[0], 0)] * 2, 1)], 1)
np.save(output_npy_path, pose)
shape = [768, 768]
if args.render:
render = SMPLRender()
output_mp4_path = args.input.replace('.npy', '_smpl.mp4')
os.environ['PYOPENGL_PLATFORM'] = 'egl'
size = (shape[1], shape[0])
fps = 30.0
fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V')
videoWriter = cv2.VideoWriter(output_mp4_path, fourcc, fps, size)
r = RRR.from_rotvec(np.array([np.pi, 0.0, 0.0]))
pose[:, 0] = np.matmul(r.as_matrix().reshape(1, 3, 3), pose[:, 0])
for i in range(data.shape[0]):
img = np.zeros([shape[0], shape[1], 3])
aroot = data[[i], 0] + np.array([[0.0, 0.0, 30.0]])
aroot[:, 1] = -aroot[:, 1]
params = dict(pred_shape=np.zeros([1, 10]),
pred_root=aroot,
pred_pose=pose[[i]])
renderImg = render.render(img.copy(), params)
renderImg = (renderImg * 255).astype(np.uint8)
videoWriter.write(renderImg)
videoWriter.release()
|