From 7e39b35f0ebc51529e7b322f3810c46d3791d1f9 Mon Sep 17 00:00:00 2001 From: jin-s13 Date: Fri, 25 Jun 2021 20:45:39 +0800 Subject: [PATCH] rebase --- mmpose/apis/inference_3d.py | 78 ++++++++++++------------------------- 1 file changed, 24 insertions(+), 54 deletions(-) diff --git a/mmpose/apis/inference_3d.py b/mmpose/apis/inference_3d.py index 2e547e8714..28a367ad49 100644 --- 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