xsl/comfyui_controlnet_aux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

save code

Browse Source
This commit is contained in:
xsl
2025-08-13 10:12:45 +08:00
parent a41a40639f
commit 1f53c3d42b
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/custom_controlnet_aux/dwpose/__init__.py
+6 -5
View File
@@ -316,11 +316,11 @@ def draw_pose_mask(mask: torch.Tensor, keypoints, H, W, cloth_len, line_len, clo
add_point(points, keypoints, 12, H, W, unit_len*offsetX, 0)
add_point(points, keypoints, 9, H, W, -unit_len*offsetX, 0)
if keypoints[10] == None or keypoints[13] == None:
add_point(points, keypoints, 12, H, W, unit_len*offsetX, unit_len*2.7)
add_point(points, keypoints, 9, H, W, -unit_len*offsetX, unit_len*2.7)
add_point(points, keypoints, 12, H, W, unit_len*offsetX, unit_len*2.9)
add_point(points, keypoints, 9, H, W, -unit_len*offsetX, unit_len*2.9)
else:
add_point(points, keypoints, 13, H, W, unit_len*offsetX, unit_len*0.5)
add_point(points, keypoints, 10, H, W, -unit_len*offsetX, unit_len*0.5)
add_point(points, keypoints, 13, H, W, unit_len*offsetX, unit_len*0.7)
add_point(points, keypoints, 10, H, W, -unit_len*offsetX, unit_len*0.7)
else:
addBase(points, keypoints, H, W, unit_len, closh_short)
add_point(points, keypoints, 11, H, W, unit_len*0.2, 0)
@@ -358,8 +358,9 @@ def draw_poses_mask(poses: List[PoseResult], H, W, cloth_len, line_len, cloth_sh
max_pose = findMaxPose(poses)
mask = draw_pose_mask(mask, max_pose.body.keypoints, H, W, cloth_len, line_len, cloth_short)
neck_y = max_pose.body.keypoints[1].y
return mask
return mask, neck_y
def draw_poses(poses: List[PoseResult], H, W, draw_body=True, draw_hand=True, draw_face=True, xinsr_stick_scaling=False):
"""