visualize_skeleton.py

AIST_SPIN_3D_EDGES = [
    [0, 1, 0],
    [1, 2, 0],
    [2, 8, 0],
    [3, 9, 1],
    [4, 3, 1],
    [5, 4, 1],
    [6, 7, 0],
    [7, 8, 0],
    [8, 12, 0],
    [9, 12, 1],
    [10, 9, 1],
    [11, 10, 1],
    [12, 13, 1],
]

AIST_SPIN_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 1],
    [4, 1],
    [5, 1],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
]


AIST_TCMR_3D_EDGES = [
    [0, 1, 0],
    [1, 2, 0],
    [2, 8, 0],
    [3, 9, 1],
    [4, 3, 1],
    [5, 4, 1],
    [6, 7, 0],
    [7, 8, 0],
    [8, 12, 0],
    [9, 12, 1],
    [10, 9, 1],
    [11, 10, 1],
    [12, 13, 1],
]

AIST_TCMR_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 1],
    [4, 1],
    [5, 1],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
]


AIST_VIBE_3D_EDGES = [
    [0, 1, 0],
    [1, 2, 0],
    [2, 8, 0],
    [3, 9, 1],
    [4, 3, 1],
    [5, 4, 1],
    [6, 7, 0],
    [7, 8, 0],
    [8, 12, 0],
    [9, 12, 1],
    [10, 9, 1],
    [11, 10, 1],
    [12, 13, 1],
]

AIST_VIBE_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 1],
    [4, 1],
    [5, 1],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
]


H36M_FCN_3D_EDGES = [
    [0, 1, 0],
    [0, 7, 0],
    [7, 8, 0],
    [8, 9, 0],
    [9, 10, 0],
    [1, 2, 0],
    [2, 3, 0],
    [7, 14, 0],
    [14, 15, 0],
    [15, 16, 0],
    [0, 4, 1],
    [4, 5, 1],
    [5, 6, 1],
    [7, 11, 1],
    [11, 12, 1],
    [12, 13, 1],
]

H36M_FCN_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 0],
    [4, 1],
    [5, 1],
    [6, 1],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
    [14, 0],
    [15, 0],
    [16, 0],
]


H36M_RLE_3D_EDGES = [
    [0, 1, 0],
    [0, 7, 0],
    [7, 8, 0],
    [8, 9, 0],
    [9, 10, 0],
    [1, 2, 0],
    [2, 3, 0],
    [7, 14, 0],
    [14, 15, 0],
    [15, 16, 0],
    [0, 4, 1],
    [4, 5, 1],
    [5, 6, 1],
    [7, 11, 1],
    [11, 12, 1],
    [12, 13, 1],
]

H36M_RLE_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 0],
    [4, 1],
    [5, 1],
    [6, 1],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
    [14, 0],
    [15, 0],
    [16, 0],
]


H36M_VIBE_3D_EDGES = [
   [0, 1, 0],
    [1, 2, 0],
    [2, 8, 0],
    [3, 9, 1],
    [4, 3, 1],
    [5, 4, 1],
    [6, 7, 0],
    [7, 8, 0],
    [8, 12, 0],
    [9, 12, 1],
    [10, 9, 1],
    [11, 10, 1],
    [12, 13, 1],
]

H36M_VIBE_3D_JOINTS = [
   [0, 0],
    [1, 0],
    [2, 0],
    [3, 1],
    [4, 1],
    [5, 1],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
]


H36M_TCMR_3D_EDGES = [
   [0, 1, 0],
    [1, 2, 0],
    [2, 8, 0],
    [3, 9, 1],
    [4, 3, 1],
    [5, 4, 1],
    [6, 7, 0],
    [7, 8, 0],
    [8, 12, 0],
    [9, 12, 1],
    [10, 9, 1],
    [11, 10, 1],
    [12, 13, 1],
]

H36M_TCMR_3D_JOINTS = [
   [0, 0],
    [1, 0],
    [2, 0],
    [3, 1],
    [4, 1],
    [5, 1],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
]


H36M_VIDEOPOSET243_3D_EDGES = [
   [0, 1, 0],
    [0, 7, 0],
    [7, 8, 0],
    [8, 9, 0],
    [9, 10, 0],
    [1, 2, 0],
    [2, 3, 0],
    [7, 14, 0],
    [14, 15, 0],
    [15, 16, 0],
    [0, 4, 1],
    [4, 5, 1],
    [5, 6, 1],
    [7, 11, 1],
    [11, 12, 1],
    [12, 13, 1],
]

H36M_VIDEOPOSET243_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 0],
    [4, 1],
    [5, 1],
    [6, 1],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
    [14, 0],
    [15, 0],
    [16, 0],
]


H36M_VIDEOPOSET81_3D_EDGES = [
   [0, 1, 0],
    [0, 7, 0],
    [7, 8, 0],
    [8, 9, 0],
    [9, 10, 0],
    [1, 2, 0],
    [2, 3, 0],
    [7, 14, 0],
    [14, 15, 0],
    [15, 16, 0],
    [0, 4, 1],
    [4, 5, 1],
    [5, 6, 1],
    [7, 11, 1],
    [11, 12, 1],
    [12, 13, 1],
]

H36M_VIDEOPOSET81_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 0],
    [4, 1],
    [5, 1],
    [6, 1],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
    [14, 0],
    [15, 0],
    [16, 0],
]


H36M_VIDEOPOSET27_3D_EDGES = [
   [0, 1, 0],
    [0, 7, 0],
    [7, 8, 0],
    [8, 9, 0],
    [9, 10, 0],
    [1, 2, 0],
    [2, 3, 0],
    [7, 14, 0],
    [14, 15, 0],
    [15, 16, 0],
    [0, 4, 1],
    [4, 5, 1],
    [5, 6, 1],
    [7, 11, 1],
    [11, 12, 1],
    [12, 13, 1],
]

H36M_VIDEOPOSET27_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 0],
    [4, 1],
    [5, 1],
    [6, 1],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
    [14, 0],
    [15, 0],
    [16, 0],
]

MPIINF3DHP_SPIN_3D_EDGES = [
   [0, 16, 0],
   [16, 1, 0],
   [1, 15, 0],
   [15, 14, 0],
   [1, 5, 0],
   [5, 6, 0],
   [6, 7, 0],
   [14, 11, 0],
   [11, 12, 0],
   [12, 13, 0],
   [1, 2, 1],
   [2, 3, 1],
   [3, 4, 1],
   [14, 8, 1],
   [8, 9, 1],
   [9, 10, 1],
    
]

MPIINF3DHP_SPIN_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 1],
    [3, 1],
    [4, 1],
    [5, 0],
    [6, 0],
    [7, 0],
    [8, 1],
    [9, 1],
    [10, 1],
    [11, 0],
    [12, 0],
    [13, 0],
    [14, 0],
    [15, 0],
    [16, 0],
]


MPIINF3DHP_TCMR_3D_EDGES = [
   [0, 16, 0],
   [16, 1, 0],
   [1, 15, 0],
   [15, 14, 0],
   [1, 5, 0],
   [5, 6, 0],
   [6, 7, 0],
   [14, 11, 0],
   [11, 12, 0],
   [12, 13, 0],
   [1, 2, 1],
   [2, 3, 1],
   [3, 4, 1],
   [14, 8, 1],
   [8, 9, 1],
   [9, 10, 1],
    
]

MPIINF3DHP_TCMR_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 1],
    [3, 1],
    [4, 1],
    [5, 0],
    [6, 0],
    [7, 0],
    [8, 1],
    [9, 1],
    [10, 1],
    [11, 0],
    [12, 0],
    [13, 0],
    [14, 0],
    [15, 0],
    [16, 0],
]


MPIINF3DHP_VIBE_3D_EDGES = [
   [0, 16, 0],
   [16, 1, 0],
   [1, 15, 0],
   [15, 14, 0],
   [1, 5, 0],
   [5, 6, 0],
   [6, 7, 0],
   [14, 11, 0],
   [11, 12, 0],
   [12, 13, 0],
   [1, 2, 1],
   [2, 3, 1],
   [3, 4, 1],
   [14, 8, 1],
   [8, 9, 1],
   [9, 10, 1],
    
]

MPIINF3DHP_VIBE_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 1],
    [3, 1],
    [4, 1],
    [5, 0],
    [6, 0],
    [7, 0],
    [8, 1],
    [9, 1],
    [10, 1],
    [11, 0],
    [12, 0],
    [13, 0],
    [14, 0],
    [15, 0],
    [16, 0],
]


MUPOTS_TPOSENET_3D_EDGES = [
   [0, 16, 0],
   [16, 1, 0],
   [1, 15, 0],
   [15, 14, 0],
   [1, 5, 0],
   [5, 6, 0],
   [6, 7, 0],
   [14, 11, 0],
   [11, 12, 0],
   [12, 13, 0],
   [1, 2, 1],
   [2, 3, 1],
   [3, 4, 1],
   [14, 8, 1],
   [8, 9, 1],
   [9, 10, 1],
    
]

MUPOTS_TPOSENET_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 1],
    [3, 1],
    [4, 1],
    [5, 0],
    [6, 0],
    [7, 0],
    [8, 1],
    [9, 1],
    [10, 1],
    [11, 0],
    [12, 0],
    [13, 0],
    [14, 0],
    [15, 0],
    [16, 0],
]


MUPOTS_TPOSENETREFINENET_3D_EDGES = [
   [0, 16, 0],
   [16, 1, 0],
   [1, 15, 0],
   [15, 14, 0],
   [1, 5, 0],
   [5, 6, 0],
   [6, 7, 0],
   [14, 11, 0],
   [11, 12, 0],
   [12, 13, 0],
   [1, 2, 1],
   [2, 3, 1],
   [3, 4, 1],
   [14, 8, 1],
   [8, 9, 1],
   [9, 10, 1],
    
]

MUPOTS_TPOSENETREFINENET_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 1],
    [3, 1],
    [4, 1],
    [5, 0],
    [6, 0],
    [7, 0],
    [8, 1],
    [9, 1],
    [10, 1],
    [11, 0],
    [12, 0],
    [13, 0],
    [14, 0],
    [15, 0],
    [16, 0],
]


PW3D_EFT_3D_EDGES = [
   [0, 1, 0],
    [1, 2, 0],
    [2, 8, 0],
    [3, 9, 1],
    [4, 3, 1],
    [5, 4, 1],
    [6, 7, 0],
    [7, 8, 0],
    [8, 12, 0],
    [9, 12, 1],
    [10, 9, 1],
    [11, 10, 1],
    [12, 13, 1],
     
]

PW3D_EFT_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 1],
    [4, 1],
    [5, 1],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
]


PW3D_PARE_3D_EDGES = [
   [0, 1, 0],
    [1, 2, 0],
    [2, 8, 0],
    [3, 9, 1],
    [4, 3, 1],
    [5, 4, 1],
    [6, 7, 0],
    [7, 8, 0],
    [8, 12, 0],
    [9, 12, 1],
    [10, 9, 1],
    [11, 10, 1],
    [12, 13, 1],
     
]

PW3D_PARE_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 1],
    [4, 1],
    [5, 1],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
]

PW3D_SPIN_3D_EDGES = [
   [0, 1, 0],
    [1, 2, 0],
    [2, 8, 0],
    [3, 9, 1],
    [4, 3, 1],
    [5, 4, 1],
    [6, 7, 0],
    [7, 8, 0],
    [8, 12, 0],
    [9, 12, 1],
    [10, 9, 1],
    [11, 10, 1],
    [12, 13, 1],
     
]

PW3D_SPIN_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 1],
    [4, 1],
    [5, 1],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
]


PW3D_TCMR_3D_EDGES = [
    [0, 1, 0],
    [1, 2, 0],
    [2, 8, 0],
    [3, 9, 1],
    [4, 3, 1],
    [5, 4, 1],
    [6, 7, 0],
    [7, 8, 0],
    [8, 12, 0],
    [9, 12, 1],
    [10, 9, 1],
    [11, 10, 1],
    [12, 13, 1],
    
]

PW3D_TCMR_3D_JOINTS = [
  [0, 0],
    [1, 0],
    [2, 0],
    [3, 1],
    [4, 1],
    [5, 1],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
]


PW3D_VIBE_3D_EDGES = [
   [0, 1, 0],
    [1, 2, 0],
    [2, 8, 0],
    [3, 9, 1],
    [4, 3, 1],
    [5, 4, 1],
    [6, 7, 0],
    [7, 8, 0],
    [8, 12, 0],
    [9, 12, 1],
    [10, 9, 1],
    [11, 10, 1],
    [12, 13, 1],
    
]

PW3D_VIBE_3D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 1],
    [4, 1],
    [5, 1],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 1],
    [10, 1],
    [11, 1],
    [12, 1],
    [13, 1],
]




H36M_CPN_2D_EDGES = [
   [9, 8, 0],
   [8, 7, 0],
   [7, 0, 0],
   [4, 0, 0],
   [4, 5, 0],
   [5, 6, 0],
   [10, 8, 0],
   [8, 11, 0],
   [11, 12, 0],
   [13, 12, 1],
   [8, 14, 1],
   [14, 15, 1],
   [0, 1, 1],
   [1, 2, 1],
   [2, 3, 1],
   [15, 16, 1],
]

H36M_CPN_2D_JOINTS = [
 [0, 0],
    [1, 1],
    [2, 1],
    [3, 1],
    [4, 0],
    [5, 0],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 0],
    [10, 0],
    [11, 0],
    [12, 0],
    [13, 1],
    [14, 1],
    [15, 1],
    [16, 1],
]

H36M_HOURGLASS_2D_EDGES = [
   [9, 8, 0],
   [8, 7, 0],
   [7, 0, 0],
   [4, 0, 0],
   [4, 5, 0],
   [4, 6, 0],
   [10, 8, 0],
   [10, 11, 0],
   [11, 12, 0],
   [13, 8, 1],
   [13, 14, 1],
   [14, 15, 1],
   [0, 1, 1],
   [1, 2, 1],
   [2, 3, 1],
]

H36M_HOURGLASS_2D_JOINTS = [
    [0, 0],
    [1, 1],
    [2, 1],
    [3, 1],
    [4, 0],
    [5, 0],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 0],
    [10, 0],
    [11, 0],
    [12, 0],
    [13, 1],
    [14, 1],
    [15, 1],
]


H36M_HRNET_2D_EDGES = [
   [9, 8, 0],
   [8, 7, 0],
   [7, 0, 0],
   [4, 0, 0],
   [4, 5, 0],
   [4, 6, 0],
   [10, 8, 0],
   [10, 11, 0],
   [11, 12, 0],
   [13, 8, 1],
   [13, 14, 1],
   [14, 15, 1],
   [0, 1, 1],
   [1, 2, 1],
   [2, 3, 1],
]

H36M_HRNET_2D_JOINTS = [
    [0, 0],
    [1, 1],
    [2, 1],
    [3, 1],
    [4, 0],
    [5, 0],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 0],
    [10, 0],
    [11, 0],
    [12, 0],
    [13, 1],
    [14, 1],
    [15, 1],
]


H36M_RLE_2D_EDGES = [
   [9, 8, 0],
   [8, 7, 0],
   [7, 0, 0],
   [4, 0, 0],
   [4, 5, 0],
   [4, 6, 0],
   [10, 8, 0],
   [10, 11, 0],
   [11, 12, 0],
   [13, 8, 1],
   [13, 14, 1],
   [14, 15, 1],
   [0, 1, 1],
   [1, 2, 1],
   [2, 3, 1],
]

H36M_RLE_2D_JOINTS = [
    [0, 0],
    [1, 1],
    [2, 1],
    [3, 1],
    [4, 0],
    [5, 0],
    [6, 0],
    [7, 0],
    [8, 0],
    [9, 0],
    [10, 0],
    [11, 0],
    [12, 0],
    [13, 1],
    [14, 1],
    [15, 1],
]


JHMDB_SIMPLEPOSE_2D_EDGES = [
   [2, 0, 0],
   [1, 0, 0],
   [4, 0, 0],
   [4, 8, 0],
   [8, 12, 0],
   [1, 6, 0],
   [6, 10, 0],
   [10, 14, 0],
   [0, 3, 1],
   [3, 7, 1],
   [7, 11, 1],
   [1, 5, 1],
   [5, 9, 1],
   [9, 13, 1],
]

JHMDB_SIMPLEPOSE_2D_JOINTS = [
    [0, 0],
    [1, 0],
    [2, 0],
    [3, 1],
    [4, 0],
    [5, 1],
    [6, 0],
    [7, 1],
    [8, 0],
    [9, 1],
    [10, 0],
    [11, 1],
    [12, 0],
    [13, 1],
    [14, 0],
]