From 3e3979b808f60722d56e28ce01b99f56d04af8db Mon Sep 17 00:00:00 2001 From: Hacker 17082006 Date: Wed, 3 Apr 2024 20:52:30 +0700 Subject: [PATCH] 'i' --- motiondiff_modules/mogen/smpl/render_mesh.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/motiondiff_modules/mogen/smpl/render_mesh.py b/motiondiff_modules/mogen/smpl/render_mesh.py index 7d566b0..4d8f254 100644 --- a/motiondiff_modules/mogen/smpl/render_mesh.py +++ b/motiondiff_modules/mogen/smpl/render_mesh.py @@ -284,7 +284,7 @@ def render_from_smpl(thetas, yfov, move_x, move_y, move_z, x_rot, y_rot, z_rot, if depth_only: depth = r.render(scene, flags=RenderFlags.DEPTH_ONLY) - color = np.zeros([frame_width, frame_height, 3]) + color = np.zeros([frame_height, frame_width, 4]) else: color, depth = r.render(scene, flags=RenderFlags.RGBA) @@ -419,8 +419,8 @@ def render_from_smpl_multiple_subjects(verts_frames, cam_t_frames, focal_length, cam_t_subjects = cam_t_frames[i] mesh_nodes = [] if subjects is None: - vid.append(np.zeros_like([frame_height, frame_width, 3], dtype=np.uint8)) - vid_depth.append(np.zeros_like([frame_height, frame_width, 3], dtype=np.float32)) + vid.append(np.zeros([frame_height, frame_width, 4], dtype=np.uint8)) + vid_depth.append(np.zeros([frame_height, frame_width], dtype=np.float32)) continue for subject_vertices, cam_t in zip(subjects, cam_t_subjects): mesh = vertices_to_trimesh(subject_vertices, cam_t, faces) @@ -431,7 +431,7 @@ def render_from_smpl_multiple_subjects(verts_frames, cam_t_frames, focal_length, if depth_only: depth = r.render(scene, flags=RenderFlags.DEPTH_ONLY) - color = np.zeros([frame_width, frame_height, 3]) + color = np.zeros([frame_height, frame_width, 4]) else: color, depth = r.render(scene, flags=RenderFlags.RGBA) @@ -441,5 +441,6 @@ def render_from_smpl_multiple_subjects(verts_frames, cam_t_frames, focal_length, pbar.update(1) r = None - + print([frame.shape for frame in vid]) + print([frame.shape for frame in vid_depth]) return np.stack(vid, axis=0), np.stack(vid_depth, axis=0) \ No newline at end of file