undo pose colors

libs/easy_dwpose/draw/openpose.py

@@ -36,33 +36,24 @@ def draw_bodypose(canvas, candidate, subset):
     ]
 
     colors = [
-        [0, 255, 255],
-        [255, 255, 255],
-
-        # arm-r
-        [255, 0, 170],
-        [255, 0, 255],
-        [170, 0, 255],
-
-        # arm-l
-        [0, 85, 0],
-        [0, 255, 0],
-        [0, 170, 0],
-
-        #leg-r
-        [255, 85, 0],
         [255, 0, 0],
-        [255, 0, 85],
-
-        # leg-l
-        [0, 85, 255],
-        [0, 0, 255],
-        [85, 0, 255],
-
+        [255, 85, 0],
+        [255, 170, 0],
+        [255, 255, 0],
         [170, 255, 0],
+        [85, 255, 0],
+        [0, 255, 0],
+        [0, 255, 85],
         [0, 255, 170],
-        [255, 255, 0],
+        [0, 255, 255],
         [0, 170, 255],
+        [0, 85, 255],
+        [0, 0, 255],
+        [85, 0, 255],
+        [170, 0, 255],
+        [255, 0, 255],
+        [255, 0, 170],
+        [255, 0, 85],
     ]
 
     
@@ -81,7 +72,7 @@ def draw_bodypose(canvas, candidate, subset):
             length = ((X[0] - X[1]) ** 2 + (Y[0] - Y[1]) ** 2) ** 0.5
             angle = math.degrees(math.atan2(X[0] - X[1], Y[0] - Y[1]))
             polygon = cv2.ellipse2Poly((int(mY), int(mX)), (int(length / 2), stickwidth), int(angle), 0, 360, 1)
-            cv2.fillConvexPoly(canvas, polygon, colors[j[1]])
+            cv2.fillConvexPoly(canvas, polygon, colors[i])
 
     canvas = (canvas * 0.6).astype(np.uint8)