From 88ea7bd6ba34063959125b5028775bc80f462bf4 Mon Sep 17 00:00:00 2001 From: xsl Date: Wed, 13 Aug 2025 19:55:03 +0800 Subject: [PATCH] save code --- node_wrappers/pose_keypoint_postprocess.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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