Change default detector to person_yolov8m from ADetailer for more generalility

This commit is contained in:
Hacker 17082006
2024-04-03 22:06:30 +07:00
parent 3e3979b808
commit b2ad8129e1
2 changed files with 3 additions and 5 deletions
+2 -4
View File
@@ -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)
+1 -1
View File
@@ -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})
}
}