color skeleton hands

pipelines/text_to_sign/renderer.py

@@ -7,11 +7,18 @@ from .config import WIDTH, HEIGHT, FPS, POSE_CONNECTIONS
 # =========================
 WHITE = (255, 255, 255)
 
-# #190124 in HEX
-# RGB = (25, 1, 36)
-# OpenCV uses BGR
+# Background (#190124 in HEX)
 BACKGROUND = (36, 1, 25)
 
+# Finger colors (BGR)
+FINGER_COLORS = {
+    "thumb":  (0, 0, 255),      # Red
+    "index":  (0, 255, 0),      # Green
+    "middle": (255, 0, 0),      # Blue
+    "ring":   (0, 255, 255),    # Yellow
+    "pinky":  (255, 0, 255)     # Magenta
+}
+
 # =========================
 # HAND
 # =========================
@@ -21,14 +28,16 @@ def draw_hand_stylish(img, hand_points, thickness=5):
     pts = [(int(x * w), int(y * h)) for x, y, _ in hand_points]
 
     fingers = {
-        "thumb": [0, 1, 2, 3, 4],
-        "index": [0, 5, 6, 7, 8],
+        "thumb":  [0, 1, 2, 3, 4],
+        "index":  [0, 5, 6, 7, 8],
         "middle": [0, 9, 10, 11, 12],
-        "ring": [0, 13, 14, 15, 16],
-        "pinky": [0, 17, 18, 19, 20]
+        "ring":   [0, 13, 14, 15, 16],
+        "pinky":  [0, 17, 18, 19, 20]
     }
 
-    for idxs in fingers.values():
+    for finger_name, idxs in fingers.items():
+        color = FINGER_COLORS[finger_name]
+
         for i in range(len(idxs) - 1):
             x1, y1 = pts[idxs[i]]
             x2, y2 = pts[idxs[i + 1]]
@@ -37,7 +46,7 @@ def draw_hand_stylish(img, hand_points, thickness=5):
                 img,
                 (x1, y1),
                 (x2, y2),
-                WHITE,
+                color,
                 thickness,
                 cv2.LINE_AA
             )
@@ -123,7 +132,6 @@ def render(frames, output="output.mp4"):
     )
 
     for frame in frames:
-        # Create background using #190124
         img = np.zeros((HEIGHT, WIDTH, 3), dtype=np.uint8)
         img[:] = BACKGROUND