save code

This commit is contained in:
xsl
2025-08-13 19:55:03 +08:00
parent 1f53c3d42b
commit 88ea7bd6ba
+3 -3
View File
@@ -330,7 +330,7 @@ class RenderPeopleKpsMask:
} }
} }
RETURN_TYPES = ("MASK",) RETURN_TYPES = ("MASK","INT",)
FUNCTION = "render" FUNCTION = "render"
CATEGORY = "ControlNet Preprocessors/Pose Keypoint Postprocess" CATEGORY = "ControlNet Preprocessors/Pose Keypoint Postprocess"
@@ -339,7 +339,7 @@ class RenderPeopleKpsMask:
kps = kps[0] kps = kps[0]
poses, _, height, width = decode_json_as_poses(kps) poses, _, height, width = decode_json_as_poses(kps)
mask = draw_poses_mask( mask, neck_y = draw_poses_mask(
poses, poses,
height, height,
width, width,
@@ -347,7 +347,7 @@ class RenderPeopleKpsMask:
line_len, line_len,
cloth_short, cloth_short,
) )
return (mask,) return (mask, int(neck_y), )
class RenderPeopleKps: class RenderPeopleKps:
@classmethod @classmethod