From 1f53c3d42b51c83b865be08e5a5098aaae811045 Mon Sep 17 00:00:00 2001 From: xsl Date: Wed, 13 Aug 2025 10:12:45 +0800 Subject: [PATCH] save code --- src/custom_controlnet_aux/dwpose/__init__.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/custom_controlnet_aux/dwpose/__init__.py b/src/custom_controlnet_aux/dwpose/__init__.py index 7455686..a76d947 100644 --- a/src/custom_controlnet_aux/dwpose/__init__.py +++ b/src/custom_controlnet_aux/dwpose/__init__.py @@ -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): """