From b6ef65acbef127d2404c98561bc382648c823980 Mon Sep 17 00:00:00 2001 From: xsl Date: Mon, 11 Aug 2025 23:08:08 +0800 Subject: [PATCH] savef code --- src/custom_controlnet_aux/dwpose/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/custom_controlnet_aux/dwpose/__init__.py b/src/custom_controlnet_aux/dwpose/__init__.py index e858f87..59394ff 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*3.2) - add_point(points, keypoints, 9, H, W, -unit_len*offsetX, unit_len*3.2) + add_point(points, keypoints, 12, H, W, unit_len*offsetX, unit_len*2.2) + add_point(points, keypoints, 9, H, W, -unit_len*offsetX, unit_len*2.2) else: - add_point(points, keypoints, 13, H, W, unit_len*offsetX, unit_len*1) - add_point(points, keypoints, 10, H, W, -unit_len*offsetX, unit_len*1) + add_point(points, keypoints, 13, H, W, unit_len*offsetX, unit_len*0) + add_point(points, keypoints, 10, H, W, -unit_len*offsetX, unit_len*0) else: addBase(points, keypoints, H, W, unit_len, closh_short) add_point(points, keypoints, 11, H, W, unit_len*0.2, 0)