Hi, thanks for your amazing work!
I successfully ran evaluator.py and calculated FIDs with tools/calculate_scores.py.
I want to visualize the evaluation results by using function named visualize which is defined as below.
|
# visualize(result_motion, smpl) |
|
def visualize(motion, smpl_model): |
|
smpl_poses, smpl_trans = recover_to_axis_angles(motion) |
|
smpl_poses = np.squeeze(smpl_poses, axis=0) # (seq_len, 24, 3) |
|
smpl_trans = np.squeeze(smpl_trans, axis=0) # (seq_len, 3) |
|
keypoints3d = smpl_model.forward( |
|
global_orient=torch.from_numpy(smpl_poses[:, 0:1]).float(), |
|
body_pose=torch.from_numpy(smpl_poses[:, 1:]).float(), |
|
transl=torch.from_numpy(smpl_trans).float(), |
|
).joints.detach().numpy() # (seq_len, 24, 3) |
|
|
|
bbox_center = ( |
|
keypoints3d.reshape(-1, 3).max(axis=0) |
|
+ keypoints3d.reshape(-1, 3).min(axis=0) |
|
) / 2.0 |
|
bbox_size = ( |
|
keypoints3d.reshape(-1, 3).max(axis=0) |
|
- keypoints3d.reshape(-1, 3).min(axis=0) |
|
) |
|
world = vedo.Box(bbox_center, bbox_size[0], bbox_size[1], bbox_size[2]).wireframe() |
|
vedo.show(world, axes=True, viewup="y", interactive=0) |
|
for kpts in keypoints3d: |
|
pts = vedo.Points(kpts).c("red") |
|
plotter = vedo.show(world, pts) |
|
if plotter.escaped: break # if ESC |
|
time.sleep(0.01) |
|
vedo.interactive().close() |
However, my code says Aborted (core dumped) when I uncomment L211 of tools/calculate_scores.py and doesn't work.
Can you help me out with this problem?
