From b2ad8129e1ef5b73c47e4f33cf0eb59be3bbc77e Mon Sep 17 00:00:00 2001 From: Hacker 17082006 Date: Wed, 3 Apr 2024 22:06:30 +0700 Subject: [PATCH] Change default detector to person_yolov8m from ADetailer for more generalility --- motiondiff_modules/mogen/smpl/render_mesh.py | 6 ++---- nodes/human4d_nodes.py | 2 +- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/motiondiff_modules/mogen/smpl/render_mesh.py b/motiondiff_modules/mogen/smpl/render_mesh.py index 4d8f254..534a715 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_height, frame_width, 4]) + color = np.ones([frame_height, frame_width, 4]) else: color, depth = r.render(scene, flags=RenderFlags.RGBA) @@ -419,7 +419,7 @@ 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([frame_height, frame_width, 4], dtype=np.uint8)) + vid.append(np.ones([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): @@ -441,6 +441,4 @@ 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 diff --git a/nodes/human4d_nodes.py b/nodes/human4d_nodes.py index 2991cbb..a4d01ac 100644 --- a/nodes/human4d_nodes.py +++ b/nodes/human4d_nodes.py @@ -32,7 +32,7 @@ class Humans4DLoader: def INPUT_TYPES(s): return { "required": { - "detector": (["person_yolov8m-seg.pt", "person_yolov8s-seg.pt", "yolov8x.pt", "yolov9c.pt", "yolov9e.pt"], {"default": "yolov8x.pt"}), + "detector": (["person_yolov8m-seg.pt", "person_yolov8s-seg.pt", "yolov8x.pt", "yolov9c.pt", "yolov9e.pt"], {"default": "person_yolov8m-seg.pt"}), "fp16": ("BOOLEAN", {"default": False}) } }