修改骨架颜色

axmodel_infer_fall.py

@@ -53,10 +53,8 @@ def plot_skeleton_kpts(im, kpts, steps, orig_shape=None):
     num_kpts = len(kpts) // steps
 
     skeleton = [[1, 2], [2, 3], [14, 1], [14, 4], [4, 5], [5, 6], [13, 14], [7, 14], [10, 14], [7, 8], [8, 9],[10,11],[11, 12]]    
-    
-    pose_limb_color = palette[[9, 9, 9, 9, 7, 7, 7, 0, 0, 0, 0, 0, 16, 16, 16, 16, 16, 16, 16][:(num_kpts+2)]]
-    
-    pose_kpt_color = palette[[16, 16, 16, 16, 16, 0, 0, 0, 0, 0, 0, 9, 9, 9, 9, 9, 9][:num_kpts]]
+    pose_limb_color = palette[[9, 9, 9, 9, 9, 9, 7, 0, 0, 0, 0, 0, 0]]
+    pose_kpt_color = palette[[16, 16, 16, 16, 16, 16, 0, 0, 0, 0, 0, 0, 9, 9]]
     radius = 5
     for kid in range(num_kpts):
         r, g, b = pose_kpt_color[kid]