save code

src/custom_controlnet_aux/dwpose/__init__.py

@@ -241,9 +241,9 @@ def distance(p1, p2):
 
 
 def addBase(points, keypoints, H, W, unit_len, closh_short):
-    add_point(points, keypoints, 2, H, W, -unit_len*0.4, 0)
+    add_point(points, keypoints, 2, H, W, -unit_len*0.4, -unit_len*0.1)
-    add_point(points, keypoints, 1, H, W, 0, -unit_len*0.2)
+    add_point(points, keypoints, 1, H, W, 0, -unit_len*0.25)
-    add_point(points, keypoints, 5, H, W, unit_len*0.4, 0)
+    add_point(points, keypoints, 5, H, W, unit_len*0.4, -unit_len*0.1)
     add_point(points, keypoints, 6, H, W, unit_len*0.2)
     add_point(points, keypoints, 3, H, W, -unit_len*0.2)
     if not closh_short:
@@ -326,7 +326,7 @@ def draw_pose_mask(mask: torch.Tensor, keypoints, H, W, cloth_len, line_len, clo
         add_point(points, keypoints, 8, H, W, -unit_len*0.2, 0)
 
     points = convex_hull(points)
-
+    line_len = 0.7
     for i in range(len(points)):
         if i == len(points) - 1:
             p1 = points[i]