rebase

open-mmlab · Jun 25, 2021 · 7e39b35 · 7e39b35
1 parent 7f6bb9d
commit 7e39b35
Showing 1 changed file with 24 additions and 54 deletions.
diff --git a/mmpose/apis/inference_3d.py b/mmpose/apis/inference_3d.py
@@ -240,75 +240,45 @@ def vis_3d_pose_result(model,
             ]]
 
         elif dataset == 'InterHand3DDataset':
-            skeleton = [[ 0,  1],
-                   [ 1,  2],
-                   [ 2,  3],
-                   [ 3, 20],
-                   [ 4,  5],
-                   [ 5,  6],
-                   [ 6,  7],
-                   [ 7, 20],
-                   [ 8,  9],
-                   [ 9, 10],
-                   [10, 11],
-                   [11, 20],
-                   [12, 13],
-                   [13, 14],
-                   [14, 15],
-                   [15, 20],
-                   [16, 17],
-                   [17, 18],
-                   [18, 19],
-                   [19, 20],
-                   [21, 22],
-                   [22, 23],
-                   [23, 24],
-                   [24, 41],
-                   [25, 26],
-                   [26, 27],
-                   [27, 28],
-                   [28, 41],
-                   [29, 30],
-                   [30, 31],
-                   [31, 32],
-                   [32, 41],
-                   [33, 34],
-                   [34, 35],
-                   [35, 36],
-                   [36, 41],
-                   [37, 38],
-                   [38, 39],
-                   [39, 40],
-                   [40, 41]]
+            skeleton = [[0, 1], [1, 2], [2, 3], [3, 20], [4, 5], [5, 6],
+                        [6, 7], [7, 20], [8, 9], [9, 10], [10, 11], [11, 20],
+                        [12, 13], [13, 14], [14, 15], [15, 20], [16, 17],
+                        [17, 18], [18, 19], [19, 20], [21, 22], [22, 23],
+                        [23, 24], [24, 41], [25, 26], [26, 27], [27, 28],
+                        [28, 41], [29, 30], [30, 31], [31, 32], [32, 41],
+                        [33, 34], [34, 35], [35, 36], [36, 41], [37, 38],
+                        [38, 39], [39, 40], [40, 41]]
 
             pose_kpt_color = [[14, 128, 250], [14, 128, 250], [14, 128, 250],
                               [14, 128, 250], [80, 127, 255], [80, 127, 255],
                               [80, 127, 255], [80, 127, 255], [71, 99, 255],
                               [71, 99, 255], [71, 99, 255], [71, 99, 255],
                               [0, 36, 255], [0, 36, 255], [0, 36, 255],
-                              [0, 36, 255], [0, 0, 230], [0, 0, 230], [0, 0, 230],
-                              [0, 0, 230], [0, 0, 139], [237, 149, 100],
-                              [237, 149, 100], [237, 149, 100], [237, 149, 100],
+                              [0, 36, 255], [0, 0, 230], [0, 0, 230],
+                              [0, 0, 230], [0, 0, 230], [0, 0, 139],
+                              [237, 149, 100], [237, 149, 100],
+                              [237, 149, 100], [237, 149, 100], [230, 128, 77],
                               [230, 128, 77], [230, 128, 77], [230, 128, 77],
-                              [230, 128, 77], [255, 144, 30], [255, 144, 30],
-                              [255, 144, 30], [255, 144, 30], [153, 51, 0],
-                              [153, 51, 0], [153, 51, 0], [153, 51, 0],
+                              [255, 144, 30], [255, 144, 30], [255, 144, 30],
+                              [255, 144, 30], [153, 51, 0], [153, 51, 0],
+                              [153, 51, 0], [153, 51, 0], [255, 51, 13],
                               [255, 51, 13], [255, 51, 13], [255, 51, 13],
-                              [255, 51, 13], [103, 37, 8]]
+                              [103, 37, 8]]
 
             pose_link_color = [[14, 128, 250], [14, 128, 250], [14, 128, 250],
                                [14, 128, 250], [80, 127, 255], [80, 127, 255],
                                [80, 127, 255], [80, 127, 255], [71, 99, 255],
                                [71, 99, 255], [71, 99, 255], [71, 99, 255],
                                [0, 36, 255], [0, 36, 255], [0, 36, 255],
-                               [0, 36, 255], [0, 0, 230], [0, 0, 230], [0, 0, 230],
-                               [0, 0, 230], [237, 149, 100], [237, 149, 100],
-                               [237, 149, 100], [237, 149, 100], [230, 128, 77],
-                               [230, 128, 77], [230, 128, 77], [230, 128, 77],
+                               [0, 36, 255], [0, 0, 230], [0, 0, 230],
+                               [0, 0, 230], [0, 0, 230], [237, 149, 100],
+                               [237, 149, 100], [237, 149, 100],
+                               [237, 149, 100], [230, 128, 77], [230, 128, 77],
+                               [230, 128, 77], [230, 128, 77], [255, 144, 30],
                                [255, 144, 30], [255, 144, 30], [255, 144, 30],
-                               [255, 144, 30], [153, 51, 0], [153, 51, 0],
-                               [153, 51, 0], [153, 51, 0], [255, 51, 13],
-                               [255, 51, 13], [255, 51, 13], [255, 51, 13]]
+                               [153, 51, 0], [153, 51, 0], [153, 51, 0],
+                               [153, 51, 0], [255, 51, 13], [255, 51, 13],
+                               [255, 51, 13], [255, 51, 13]]
         else:
             raise NotImplementedError