diff --git a/node_wrappers/pose_keypoint_postprocess.py b/node_wrappers/pose_keypoint_postprocess.py index 2bebc38..c02db74 100644 --- a/node_wrappers/pose_keypoint_postprocess.py +++ b/node_wrappers/pose_keypoint_postprocess.py @@ -330,7 +330,7 @@ class RenderPeopleKpsMask: } } - RETURN_TYPES = ("MASK",) + RETURN_TYPES = ("MASK","INT",) FUNCTION = "render" CATEGORY = "ControlNet Preprocessors/Pose Keypoint Postprocess" @@ -339,7 +339,7 @@ class RenderPeopleKpsMask: kps = kps[0] poses, _, height, width = decode_json_as_poses(kps) - mask = draw_poses_mask( + mask, neck_y = draw_poses_mask( poses, height, width, @@ -347,7 +347,7 @@ class RenderPeopleKpsMask: line_len, cloth_short, ) - return (mask,) + return (mask, int(neck_y), ) class RenderPeopleKps: @classmethod