save code

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