save code
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user