change pose colors

libs/easy_dwpose/draw/openpose.py

@@ -40,9 +40,9 @@ def draw_bodypose(canvas, candidate, subset):
         [255, 255, 255],
 
         # arm-r
-        [170, 255, 0],
-        [255, 255, 0],
-        [255, 170, 0],
+        [255, 0, 170],
+        [255, 0, 255],
+        [170, 0, 255],
 
         # arm-l
         [0, 85, 0],
@@ -59,9 +59,9 @@ def draw_bodypose(canvas, candidate, subset):
         [0, 0, 255],
         [85, 0, 255],
 
-        [170, 0, 255],
+        [170, 255, 0],
         [0, 255, 170],
-        [255, 0, 255],
+        [255, 255, 0],
         [0, 170, 255],
     ]
 

main.py

@@ -353,7 +353,7 @@ def prepare_inputs_inference(in_img, in_vid, frames, fps, dwpose, rembg_session,
             tpose.save("out/"+"tar_pose"+str(len(target_poses_cropped))+".png")
         target_poses_cropped.append(tpose)
         
-    target_poses_cropped[0].save("pose.png")
+    #target_poses_cropped[0].save("pose.png")
     return in_img, target_poses_cropped, in_pose, target_poses_coords, frames