How can I visualize results of evaluation?

[eunhee-dev]

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.

mint/tools/calculate_scores.py

Line 211 in b8f8bdf

# visualize(result_motion, smpl)

mint/tools/calculate_scores.py

Lines 54 to 79 in b8f8bdf

	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?