Upload folder using huggingface_hub

pose.py

@@ -14,21 +14,23 @@ half = device=='cuda'
 connection_color_map = {
     (0, 1): 'lightgreen',   # Nose to Left Eye
     (0, 2): 'lightgreen',   # Nose to Right Eye
-    (1, 3): 'lightgreen',   # Left Eye to Left Ear
-    (2, 4): 'lightgreen',   # Right Eye to Right Ear
-    (0, 5): 'lightblue',    # Nose to Left Shoulder
-    (0, 6): 'lightblue',    # Nose to Right Shoulder
+    (1, 2): 'lightgreen',   # Left Eye to Right Eye
+    (1, 3): 'magenta',      # Left Eye to Left Ear
+    (2, 4): 'magenta',      # Right Eye to Right Ear
+    (3, 5): 'lightblue',    # Left Ear to Left Shoulder
+    (4, 6): 'lightblue',    # Right Ear to Right Shoulder
+    (5, 6): 'purple',         # Left Shoulder to Right Shoulder
     (5, 7): 'blue',         # Left Shoulder to Left Elbow
     (7, 9): 'blue',         # Left Elbow to Left Wrist
     (6, 8): 'red',          # Right Shoulder to Right Elbow
     (8, 10): 'red',         # Right Elbow to Right Wrist
     (5, 11): 'purple',      # Left Shoulder to Left Hip
     (6, 12): 'purple',      # Right Shoulder to Right Hip
-    (11, 13): 'purple',     # Left Hip to Left Knee
-    (13, 15): 'purple',     # Left Knee to Left Ankle
-    (12, 14): 'purple',     # Right Hip to Right Knee
-    (14, 16): 'purple',     # Right Knee to Right Ankle
-    (11, 12): 'orange'      # Left Hip to Right Hip
+    (6, 12): 'purple',      # Left Hip to Right Hip
+    (11, 13): 'orange',     # Left Hip to Left Knee
+    (13, 15): 'orange',     # Left Knee to Left Ankle
+    (12, 14): 'orange',     # Right Hip to Right Knee
+    (14, 16): 'orange',     # Right Knee to Right Ankle
 }