realease: publish the huggingface space demo of SDPose

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+assets/logo/logo.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,3 +1,829 @@
-import trackio
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+SDPose Gradio Space - Deployment Version
+Optimized for Hugging Face Spaces with model selection
+"""
+
+import gradio as gr
+import cv2
+import numpy as np
+import torch
+import os
+import sys
+import math
+import json
+import matplotlib.colors
+from pathlib import Path
+from PIL import Image
+from torchvision import transforms
+from typing import Optional, Tuple, List
+import tempfile
+from huggingface_hub import hf_hub_download, snapshot_download
+
+# Add current directory to path for imports
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+# Import required modules
+from diffusers import DDPMScheduler, AutoencoderKL, UNet2DConditionModel
+from transformers import CLIPTokenizer, CLIPTextModel
+from models.HeatmapHead import get_heatmap_head
+from models.ModifiedUNet import Modified_forward
+from pipelines.SDPose_D_Pipeline import SDPose_D_Pipeline
+from safetensors.torch import load_file
+
+try:
+    from diffusers.utils import is_xformers_available
+except ImportError:
+    def is_xformers_available():
+        return False
+
+# Try to import YOLO
+try:
+    from ultralytics import YOLO
+    YOLO_AVAILABLE = True
+except ImportError:
+    YOLO_AVAILABLE = False
+    print("⚠️  ultralytics not available, YOLO detection will be disabled")
+
+# Constants for Gradio Space
+MODEL_REPOS = {
+    "body": "teemosliang/SDPose-Body",
+    "wholebody": "teemosliang/SDPose-Wholebody"
+}
+DEFAULT_YOLO_MODEL = "yolov8n.pt"  # Will auto-download
+
+def draw_body17_keypoints_openpose_style(canvas, keypoints, scores=None, threshold=0.3, overlay_mode=False, overlay_alpha=0.6):
+    """Draw body keypoints in OpenPose style"""
+    H, W, C = canvas.shape
+    
+    if len(keypoints) >= 7:
+        neck = (keypoints[5] + keypoints[6]) / 2
+        neck_score = min(scores[5], scores[6]) if scores is not None else 1.0
+        
+        candidate = np.zeros((18, 2))
+        candidate_scores = np.zeros(18)
+        
+        candidate[0] = keypoints[0]
+        candidate[1] = neck
+        candidate[2] = keypoints[6]
+        candidate[3] = keypoints[8]
+        candidate[4] = keypoints[10]
+        candidate[5] = keypoints[5]
+        candidate[6] = keypoints[7]
+        candidate[7] = keypoints[9]
+        candidate[8] = keypoints[12]
+        candidate[9] = keypoints[14]
+        candidate[10] = keypoints[16]
+        candidate[11] = keypoints[11]
+        candidate[12] = keypoints[13]
+        candidate[13] = keypoints[15]
+        candidate[14] = keypoints[2]
+        candidate[15] = keypoints[1]
+        candidate[16] = keypoints[4]
+        candidate[17] = keypoints[3]
+        
+        if scores is not None:
+            candidate_scores[0] = scores[0]
+            candidate_scores[1] = neck_score
+            candidate_scores[2] = scores[6]
+            candidate_scores[3] = scores[8]
+            candidate_scores[4] = scores[10]
+            candidate_scores[5] = scores[5]
+            candidate_scores[6] = scores[7]
+            candidate_scores[7] = scores[9]
+            candidate_scores[8] = scores[12]
+            candidate_scores[9] = scores[14]
+            candidate_scores[10] = scores[16]
+            candidate_scores[11] = scores[11]
+            candidate_scores[12] = scores[13]
+            candidate_scores[13] = scores[15]
+            candidate_scores[14] = scores[2]
+            candidate_scores[15] = scores[1]
+            candidate_scores[16] = scores[4]
+            candidate_scores[17] = scores[3]
+    else:
+        return canvas
+    
+    avg_size = (H + W) / 2
+    stickwidth = max(1, int(avg_size / 256))
+    circle_radius = max(2, int(avg_size / 192))
+    
+    limbSeq = [
+        [2, 3], [2, 6], [3, 4], [4, 5], [6, 7], [7, 8], [2, 9], [9, 10],
+        [10, 11], [2, 12], [12, 13], [13, 14], [2, 1], [1, 15], [15, 17],
+        [1, 16], [16, 18]
+    ]
+    
+    colors = [
+        [255, 0, 0], [255, 85, 0], [255, 170, 0], [255, 255, 0], [170, 255, 0],
+        [85, 255, 0], [0, 255, 0], [0, 255, 85], [0, 255, 170], [0, 255, 255],
+        [0, 170, 255], [0, 85, 255], [0, 0, 255], [85, 0, 255],
+        [170, 0, 255], [255, 0, 255], [255, 0, 170], [255, 0, 85]
+    ]
+    
+    for i in range(len(limbSeq)):
+        index = np.array(limbSeq[i]) - 1
+        if index[0] >= len(candidate) or index[1] >= len(candidate):
+            continue
+            
+        if scores is not None:
+            if candidate_scores[index[0]] < threshold or candidate_scores[index[1]] < threshold:
+                continue
+        
+        Y = candidate[index.astype(int), 0]
+        X = candidate[index.astype(int), 1]
+        mX = np.mean(X)
+        mY = np.mean(Y)
+        length = ((X[0] - X[1]) ** 2 + (Y[0] - Y[1]) ** 2) ** 0.5
+        
+        if length < 1:
+            continue
+            
+        angle = math.degrees(math.atan2(X[0] - X[1], Y[0] - Y[1]))
+        polygon = cv2.ellipse2Poly(
+            (int(mY), int(mX)), (int(length / 2), stickwidth), int(angle), 0, 360, 1
+        )
+        cv2.fillConvexPoly(canvas, polygon, colors[i % len(colors)])
+    
+    for i in range(18):
+        if scores is not None and candidate_scores[i] < threshold:
+            continue
+            
+        x, y = candidate[i]
+        x = int(x)
+        y = int(y)
+        
+        if x < 0 or y < 0 or x >= W or y >= H:
+            continue
+            
+        cv2.circle(canvas, (int(x), int(y)), circle_radius, colors[i % len(colors)], thickness=-1)
+    
+    return canvas
+
+
+def draw_wholebody_keypoints_openpose_style(canvas, keypoints, scores=None, threshold=0.3, overlay_mode=False, overlay_alpha=0.6):
+    """Draw wholebody keypoints (134 keypoints) in DWPose style"""
+    H, W, C = canvas.shape
+    
+    stickwidth = 4
+    
+    # Body connections
+    body_limbSeq = [
+        [2, 3], [2, 6], [3, 4], [4, 5], [6, 7], [7, 8], [2, 9], [9, 10],
+        [10, 11], [2, 12], [12, 13], [13, 14], [2, 1], [1, 15], [15, 17],
+        [1, 16], [16, 18]
+    ]
+    
+    # Hand connections
+    hand_edges = [
+        [0, 1], [1, 2], [2, 3], [3, 4],      # thumb
+        [0, 5], [5, 6], [6, 7], [7, 8],      # index
+        [0, 9], [9, 10], [10, 11], [11, 12], # middle
+        [0, 13], [13, 14], [14, 15], [15, 16], # ring
+        [0, 17], [17, 18], [18, 19], [19, 20], # pinky
+    ]
+    
+    colors = [
+        [255, 0, 0], [255, 85, 0], [255, 170, 0], [255, 255, 0], [170, 255, 0],
+        [85, 255, 0], [0, 255, 0], [0, 255, 85], [0, 255, 170], [0, 255, 255],
+        [0, 170, 255], [0, 85, 255], [0, 0, 255], [85, 0, 255],
+        [170, 0, 255], [255, 0, 255], [255, 0, 170], [255, 0, 85]
+    ]
+    
+    # Draw body limbs
+    if len(keypoints) >= 18:
+        for i, limb in enumerate(body_limbSeq):
+            idx1, idx2 = limb[0] - 1, limb[1] - 1
+            if idx1 >= 18 or idx2 >= 18:
+                continue
+            if scores is not None:
+                if scores[idx1] < threshold or scores[idx2] < threshold:
+                    continue
+            
+            Y = np.array([keypoints[idx1][0], keypoints[idx2][0]])
+            X = np.array([keypoints[idx1][1], keypoints[idx2][1]])
+            mX = np.mean(X)
+            mY = np.mean(Y)
+            length = ((X[0] - X[1]) ** 2 + (Y[0] - Y[1]) ** 2) ** 0.5
+            
+            if length < 1:
+                continue
+            
+            angle = math.degrees(math.atan2(X[0] - X[1], Y[0] - Y[1]))
+            polygon = cv2.ellipse2Poly(
+                (int(mY), int(mX)), (int(length / 2), stickwidth), int(angle), 0, 360, 1
+            )
+            cv2.fillConvexPoly(canvas, polygon, colors[i % len(colors)])
+    
+    # Draw body keypoints
+    if len(keypoints) >= 18:
+        for i in range(18):
+            if scores is not None and scores[i] < threshold:
+                continue
+            x, y = int(keypoints[i][0]), int(keypoints[i][1])
+            if 0 <= x < W and 0 <= y < H:
+                cv2.circle(canvas, (x, y), 4, colors[i % len(colors)], thickness=-1)
+    
+    # Draw foot keypoints
+    if len(keypoints) >= 24:
+        for i in range(18, 24):
+            if scores is not None and scores[i] < threshold:
+                continue
+            x, y = int(keypoints[i][0]), int(keypoints[i][1])
+            if 0 <= x < W and 0 <= y < H:
+                cv2.circle(canvas, (x, y), 4, colors[i % len(colors)], thickness=-1)
+    
+    # Draw right hand
+    if len(keypoints) >= 113:
+        eps = 0.01
+        for ie, edge in enumerate(hand_edges):
+            idx1, idx2 = 92 + edge[0], 92 + edge[1]
+            if scores is not None:
+                if scores[idx1] < threshold or scores[idx2] < threshold:
+                    continue
+            
+            x1, y1 = int(keypoints[idx1][0]), int(keypoints[idx1][1])
+            x2, y2 = int(keypoints[idx2][0]), int(keypoints[idx2][1])
+            
+            if x1 > eps and y1 > eps and x2 > eps and y2 > eps:
+                if 0 <= x1 < W and 0 <= y1 < H and 0 <= x2 < W and 0 <= y2 < H:
+                    color = matplotlib.colors.hsv_to_rgb([ie / float(len(hand_edges)), 1.0, 1.0]) * 255
+                    cv2.line(canvas, (x1, y1), (x2, y2), color, thickness=2)
+        
+        for i in range(92, 113):
+            if scores is not None and scores[i] < threshold:
+                continue
+            x, y = int(keypoints[i][0]), int(keypoints[i][1])
+            if x > eps and y > eps and 0 <= x < W and 0 <= y < H:
+                cv2.circle(canvas, (x, y), 4, (0, 0, 255), thickness=-1)
+    
+    # Draw left hand
+    if len(keypoints) >= 134:
+        eps = 0.01
+        for ie, edge in enumerate(hand_edges):
+            idx1, idx2 = 113 + edge[0], 113 + edge[1]
+            if scores is not None:
+                if scores[idx1] < threshold or scores[idx2] < threshold:
+                    continue
+            
+            x1, y1 = int(keypoints[idx1][0]), int(keypoints[idx1][1])
+            x2, y2 = int(keypoints[idx2][0]), int(keypoints[idx2][1])
+            
+            if x1 > eps and y1 > eps and x2 > eps and y2 > eps:
+                if 0 <= x1 < W and 0 <= y1 < H and 0 <= x2 < W and 0 <= y2 < H:
+                    color = matplotlib.colors.hsv_to_rgb([ie / float(len(hand_edges)), 1.0, 1.0]) * 255
+                    cv2.line(canvas, (x1, y1), (x2, y2), color, thickness=2)
+        
+        for i in range(113, 134):
+            if scores is not None and i < len(scores) and scores[i] < threshold:
+                continue
+            x, y = int(keypoints[i][0]), int(keypoints[i][1])
+            if x > eps and y > eps and 0 <= x < W and 0 <= y < H:
+                cv2.circle(canvas, (x, y), 4, (0, 0, 255), thickness=-1)
+    
+    # Draw face keypoints
+    if len(keypoints) >= 92:
+        eps = 0.01
+        for i in range(24, 92):
+            if scores is not None and scores[i] < threshold:
+                continue
+            x, y = int(keypoints[i][0]), int(keypoints[i][1])
+            if x > eps and y > eps and 0 <= x < W and 0 <= y < H:
+                cv2.circle(canvas, (x, y), 3, (255, 255, 255), thickness=-1)
+    
+    return canvas
+
+
+def detect_person_yolo(image, confidence_threshold=0.5):
+    """Detect person using YOLO with auto-download"""
+    if not YOLO_AVAILABLE:
+        h, w = image.shape[:2]
+        return [[0, 0, w, h]], False
+    
+    try:
+        # Auto-download YOLOv8n if not exists
+        model = YOLO(DEFAULT_YOLO_MODEL)
+        results = model(image, verbose=False)
+        
+        person_bboxes = []
+        for result in results:
+            boxes = result.boxes
+            if boxes is not None:
+                for box in boxes:
+                    cls = int(box.cls[0].cpu().numpy())
+                    conf = float(box.conf[0].cpu().numpy())
+                    if cls == 0 and conf > confidence_threshold:
+                        x1, y1, x2, y2 = box.xyxy[0].cpu().numpy()
+                        person_bboxes.append([float(x1), float(y1), float(x2), float(y2), conf])
+        
+        if person_bboxes:
+            person_bboxes.sort(key=lambda x: x[4], reverse=True)
+            bboxes = [bbox[:4] for bbox in person_bboxes]
+            return bboxes, True
+        else:
+            h, w = image.shape[:2]
+            return [[0, 0, w, h]], False
+        
+    except Exception as e:
+        print(f"YOLO detection failed: {e}")
+        h, w = image.shape[:2]
+        return [[0, 0, w, h]], False
+
+
+def preprocess_image_for_sdpose(image, bbox=None, input_size=(768, 1024)):
+    """Preprocess image for SDPose inference"""
+    if isinstance(image, np.ndarray):
+        if len(image.shape) == 3 and image.shape[2] == 3:
+            image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+        else:
+            image_rgb = image
+        pil_image = Image.fromarray(image_rgb)
+        original_size = (image.shape[1], image.shape[0])
+    else:
+        pil_image = image
+        original_size = pil_image.size
+    
+    crop_info = None
+    if bbox is not None:
+        x1, y1, x2, y2 = map(int, bbox)
+        x1 = max(0, x1)
+        y1 = max(0, y1)
+        x2 = min(pil_image.width, x2)
+        y2 = min(pil_image.height, y2)
+        
+        if x2 > x1 and y2 > y1:
+            cropped_image = pil_image.crop((x1, y1, x2, y2))
+            crop_info = (x1, y1, x2 - x1, y2 - y1)
+            pil_image = cropped_image
+        else:
+            crop_info = (0, 0, pil_image.width, pil_image.height)
+    else:
+        crop_info = (0, 0, pil_image.width, pil_image.height)
+    
+    transform_list = [
+        transforms.Resize((input_size[1], input_size[0])),
+        transforms.ToTensor(),
+        transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
+    ]
+    
+    val_transform = transforms.Compose(transform_list)
+    input_tensor = val_transform(pil_image).unsqueeze(0)
+    
+    return input_tensor, original_size, crop_info
+
+
+def restore_keypoints_to_original(keypoints, crop_info, input_size, original_size):
+    """Restore keypoints from cropped/resized space to original image space"""
+    x1, y1, crop_w, crop_h = crop_info
+    input_w, input_h = input_size
+    
+    scale_x = crop_w / input_w
+    scale_y = crop_h / input_h
+    
+    keypoints_restored = keypoints.copy()
+    keypoints_restored[:, 0] = keypoints[:, 0] * scale_x + x1
+    keypoints_restored[:, 1] = keypoints[:, 1] * scale_y + y1
+    
+    return keypoints_restored
+
+
+class SDPoseInference:
+    """SDPose inference class for Gradio Space"""
+    
+    def __init__(self):
+        self.pipeline = None
+        self.device = None
+        self.model_loaded = False
+        self.keypoint_scheme = "body"
+        self.input_size = (768, 1024)
+        self.model_cache_dir = None
+        
+    def load_model_from_hub(self, repo_id=None, keypoint_scheme="body"):
+        """Load model from Hugging Face Hub"""
+        try:
+            if repo_id is None:
+                repo_id = MODEL_REPOS.get(keypoint_scheme, MODEL_REPOS["body"])
+                
+            self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+            self.keypoint_scheme = keypoint_scheme
+            
+            print(f"🔄 Loading model from Hugging Face Hub: {repo_id}")
+            print(f"📱 Device: {self.device}")
+            print(f"🎯 Keypoint scheme: {keypoint_scheme}")
+            
+            # Download entire model repository
+            print("📥 Downloading model files...")
+            cache_dir = snapshot_download(
+                repo_id=repo_id,
+                allow_patterns=["*.safetensors", "*.json", "*.txt", "*.model"],
+                cache_dir="./model_cache"
+            )
+            self.model_cache_dir = cache_dir
+            print(f"✅ Model files downloaded to: {cache_dir}")
+            
+            # Load components from downloaded files
+            print("🔧 Loading model components...")
+            
+            # Load UNet
+            unet_path = os.path.join(cache_dir, "unet")
+            if os.path.exists(unet_path):
+                unet = UNet2DConditionModel.from_pretrained(
+                    unet_path,
+                    class_embed_type="projection",
+                    projection_class_embeddings_input_dim=4,
+                    low_cpu_mem_usage=False,
+                    device_map=None,
+                )
+            else:
+                # Try loading from cache_dir directly
+                unet = UNet2DConditionModel.from_pretrained(
+                    cache_dir,
+                    subfolder="unet",
+                    class_embed_type="projection",
+                    projection_class_embeddings_input_dim=4,
+                    low_cpu_mem_usage=False,
+                    device_map=None,
+                )
+            
+            unet = Modified_forward(unet, keypoint_scheme=keypoint_scheme)
+            
+            # Load VAE
+            vae_path = os.path.join(cache_dir, "vae")
+            if os.path.exists(vae_path):
+                vae = AutoencoderKL.from_pretrained(vae_path)
+            else:
+                vae = AutoencoderKL.from_pretrained(cache_dir, subfolder="vae")
+            
+            # Load tokenizer and text encoder
+            tokenizer_path = os.path.join(cache_dir, "tokenizer")
+            text_encoder_path = os.path.join(cache_dir, "text_encoder")
+            
+            if os.path.exists(tokenizer_path):
+                tokenizer = CLIPTokenizer.from_pretrained(tokenizer_path)
+            else:
+                tokenizer = CLIPTokenizer.from_pretrained(cache_dir, subfolder="tokenizer")
+                
+            if os.path.exists(text_encoder_path):
+                text_encoder = CLIPTextModel.from_pretrained(text_encoder_path)
+            else:
+                text_encoder = CLIPTextModel.from_pretrained(cache_dir, subfolder="text_encoder")
+            
+            # Load decoder
+            hm_decoder = get_heatmap_head(mode=keypoint_scheme)
+            decoder_file = os.path.join(cache_dir, "decoder", "decoder.safetensors")
+            if not os.path.exists(decoder_file):
+                # Try alternative path
+                decoder_file = os.path.join(cache_dir, "decoder.safetensors")
+            
+            if os.path.exists(decoder_file):
+                print(f"📂 Loading decoder from: {decoder_file}")
+                hm_decoder.load_state_dict(load_file(decoder_file, device="cpu"), strict=True)
+            else:
+                print("⚠️  Decoder weights not found, using random initialization")
+            
+            # Load scheduler
+            scheduler_path = os.path.join(cache_dir, "scheduler")
+            if os.path.exists(scheduler_path):
+                noise_scheduler = DDPMScheduler.from_pretrained(scheduler_path)
+            else:
+                noise_scheduler = DDPMScheduler.from_pretrained(cache_dir, subfolder="scheduler")
+            
+            # Move to device
+            print("🚀 Moving models to device...")
+            unet = unet.to(self.device)
+            vae = vae.to(self.device)
+            text_encoder = text_encoder.to(self.device)
+            hm_decoder = hm_decoder.to(self.device)
+            
+            # Create pipeline
+            self.pipeline = SDPose_D_Pipeline(
+                unet=unet,
+                vae=vae,
+                tokenizer=tokenizer,
+                text_encoder=text_encoder,
+                scheduler=noise_scheduler,
+                decoder=hm_decoder
+            )
+            
+            # Enable xformers if available
+            if is_xformers_available():
+                try:
+                    self.pipeline.unet.enable_xformers_memory_efficient_attention()
+                    print("✅ xformers enabled")
+                except:
+                    print("⚠️  xformers not enabled")
+            
+            self.model_loaded = True
+            print("✅ Model loaded successfully!")
+            return True
+            
+        except Exception as e:
+            print(f"❌ Error loading model: {e}")
+            import traceback
+            traceback.print_exc()
+            return False
+    
+    def predict_image(self, image, enable_yolo=True, score_threshold=0.3, overlay_alpha=0.6):
+        """Run inference on a single image"""
+        if not self.model_loaded or self.pipeline is None:
+            return None, "❌ Model not loaded. Please wait for model to load."
+        
+        try:
+            if isinstance(image, np.ndarray):
+                original_image_rgb = image.copy()
+            else:
+                original_image_rgb = np.array(image)
+            
+            original_image_bgr = cv2.cvtColor(original_image_rgb, cv2.COLOR_RGB2BGR)
+            
+            # Person detection
+            bboxes_list = []
+            if enable_yolo:
+                bboxes, used_yolo = detect_person_yolo(original_image_bgr)
+                bboxes_list = bboxes if bboxes else [None]
+            else:
+                bboxes_list = [None]
+            
+            # Process each person
+            pose_canvas = np.zeros_like(original_image_rgb)
+            all_keypoints = []
+            all_scores = []
+            
+            for bbox in bboxes_list:
+                input_tensor, original_size, crop_info = preprocess_image_for_sdpose(
+                    original_image_bgr, bbox, self.input_size
+                )
+                input_tensor = input_tensor.to(self.device)
+                
+                with torch.no_grad():
+                    out = self.pipeline(
+                        input_tensor,
+                        timesteps=[999],
+                        test_cfg={'flip_test': False},
+                        show_progress_bar=False,
+                        mode="inference",
+                    )
+                    
+                    heatmap_inst = out[0]
+                    keypoints = heatmap_inst.keypoints[0].cpu().numpy()
+                    scores = heatmap_inst.keypoint_scores[0].cpu().numpy()
+                
+                if bbox is not None:
+                    keypoints_original = restore_keypoints_to_original(
+                        keypoints, crop_info, self.input_size, original_size
+                    )
+                else:
+                    scale_x = original_size[0] / self.input_size[0]
+                    scale_y = original_size[1] / self.input_size[1]
+                    keypoints_original = keypoints.copy()
+                    keypoints_original[:, 0] *= scale_x
+                    keypoints_original[:, 1] *= scale_y
+                
+                all_keypoints.append(keypoints_original)
+                all_scores.append(scores)
+                
+                # Draw keypoints
+                if self.keypoint_scheme == "body" and len(keypoints_original) >= 17:
+                    pose_canvas = draw_body17_keypoints_openpose_style(
+                        pose_canvas, keypoints_original[:17], scores[:17], 
+                        threshold=score_threshold
+                    )
+                elif self.keypoint_scheme == "wholebody":
+                    # Process wholebody keypoints
+                    keypoints_with_neck = keypoints_original.copy()
+                    scores_with_neck = scores.copy()
+                    
+                    if len(keypoints_original) >= 17:
+                        # Insert neck
+                        neck = (keypoints_original[5] + keypoints_original[6]) / 2
+                        neck_score = min(scores[5], scores[6]) if scores[5] > 0.3 and scores[6] > 0.3 else 0
+                        
+                        keypoints_with_neck = np.insert(keypoints_original, 17, neck, axis=0)
+                        scores_with_neck = np.insert(scores, 17, neck_score)
+                        
+                        # Remap from mmpose to openpose format
+                        mmpose_idx = np.array([17, 6, 8, 10, 7, 9, 12, 14, 16, 13, 15, 2, 1, 4, 3])
+                        openpose_idx = np.array([1, 2, 3, 4, 6, 7, 8, 9, 10, 12, 13, 14, 15, 16, 17])
+                        
+                        temp_kpts = keypoints_with_neck.copy()
+                        temp_scores = scores_with_neck.copy()
+                        temp_kpts[openpose_idx] = keypoints_with_neck[mmpose_idx]
+                        temp_scores[openpose_idx] = scores_with_neck[mmpose_idx]
+                        
+                        keypoints_with_neck = temp_kpts
+                        scores_with_neck = temp_scores
+                    
+                    pose_canvas = draw_wholebody_keypoints_openpose_style(
+                        pose_canvas, keypoints_with_neck, scores_with_neck, 
+                        threshold=score_threshold
+                    )
+            
+            # Blend images
+            result_image = cv2.addWeighted(original_image_rgb, 1.0 - overlay_alpha, pose_canvas, overlay_alpha, 0)
+            
+            info_text = f"✅ Detected {len(all_keypoints)} person(s)\n"
+            info_text += f"📊 Keypoint scheme: {self.keypoint_scheme}\n"
+            if self.keypoint_scheme == "body":
+                info_text += f"🔢 Keypoints: 17 (COCO body)\n"
+            else:
+                info_text += f"��� Keypoints: 133 (body + face + hands + feet)\n"
+            info_text += f"🎯 Score threshold: {score_threshold}\n"
+            info_text += f"🎨 Overlay alpha: {overlay_alpha}"
+            
+            return result_image, info_text
+                
+        except Exception as e:
+            import traceback
+            error_msg = f"❌ Error during inference: {str(e)}\n{traceback.format_exc()}"
+            print(error_msg)
+            return image, error_msg
+
+
+# Global inference instances for both models
+inference_engines = {
+    "body": SDPoseInference(),
+    "wholebody": SDPoseInference()
+}
+current_model = None
+
+
+def switch_model(model_type):
+    """Switch between body and wholebody models"""
+    global current_model
+    current_model = model_type
+    
+    if not inference_engines[model_type].model_loaded:
+        print(f"🔄 Loading {model_type} model...")
+        success = inference_engines[model_type].load_model_from_hub(keypoint_scheme=model_type)
+        if success:
+            return f"✅ {model_type.capitalize()} model loaded successfully!"
+        else:
+            return f"❌ Failed to load {model_type} model"
+    else:
+        return f"✅ {model_type.capitalize()} model is ready"
+
+
+def run_inference(image, model_type, enable_yolo, score_threshold, overlay_alpha):
+    """Gradio interface function"""
+    if image is None:
+        return None, "⚠️ Please upload an image first!"
+    
+    # Ensure model is loaded
+    if not inference_engines[model_type].model_loaded:
+        status = switch_model(model_type)
+        if "Failed" in status:
+            return image, status
+    
+    result_image, info_text = inference_engines[model_type].predict_image(
+        image, 
+        enable_yolo=enable_yolo,
+        score_threshold=score_threshold,
+        overlay_alpha=overlay_alpha
+    )
+    
+    return result_image, info_text
+
+
+def create_gradio_interface():
+    """Create Gradio interface for Spaces"""
+    
+    with gr.Blocks(title="SDPose - Pose Estimation", theme=gr.themes.Soft()) as demo:
+        gr.Markdown("""
+        # 🎯 SDPose: Diffusion-based Pose Estimation
+        
+        **Exploiting Diffusion Priors for Out-of-Domain and Robust Pose Estimation**
+        
+        Upload an image to detect human poses with high accuracy, even in challenging conditions!
+        
+        📄 [Paper](https://arxiv.org/abs/2409.24980) | 💻 [GitHub](https://github.com/T-S-Liang/SDPose-OOD) | 🤗 [Body Model](https://huggingface.co/teemosliang/SDPose-Body) | 🤗 [WholeBody Model](https://huggingface.co/teemosliang/SDPose-Wholebody)
+        """)
+        
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown("### 📥 Input")
+                input_image = gr.Image(
+                    label="Upload Image",
+                    type="numpy",
+                    height=400
+                )
+                
+                gr.Markdown("### ⚙️ Settings")
+                
+                model_type = gr.Radio(
+                    choices=["body", "wholebody"],
+                    value="body",
+                    label="Model Selection",
+                    info="Body (17 keypoints) or WholeBody (133 keypoints: body + face + hands + feet)"
+                )
+                
+                model_status = gr.Textbox(
+                    label="Model Status",
+                    value="Select a model and upload an image to begin",
+                    interactive=False,
+                    lines=2
+                )
+                
+                enable_yolo = gr.Checkbox(
+                    label="Enable YOLO Person Detection",
+                    value=True,
+                    info="Automatically detect persons in the image (recommended for multi-person)"
+                )
+                
+                score_threshold = gr.Slider(
+                    minimum=0.1,
+                    maximum=0.9,
+                    value=0.3,
+                    step=0.05,
+                    label="Confidence Threshold",
+                    info="Minimum confidence score for keypoints"
+                )
+                
+                overlay_alpha = gr.Slider(
+                    minimum=0.0,
+                    maximum=1.0,
+                    value=0.6,
+                    step=0.05,
+                    label="Pose Overlay Opacity",
+                    info="Transparency of pose visualization"
+                )
+                
+                run_btn = gr.Button(
+                    "🚀 Run Pose Estimation",
+                    variant="primary",
+                    size="lg"
+                )
+            
+            with gr.Column():
+                gr.Markdown("### 📤 Output")
+                output_image = gr.Image(
+                    label="Result with Keypoints",
+                    height=400
+                )
+                
+                info_output = gr.Textbox(
+                    label="Detection Info",
+                    lines=6,
+                    interactive=False
+                )
+        
+        gr.Markdown("""
+        ### 📝 Usage Instructions
+        
+        1. **Select Model**: Choose Body (17 keypoints) or WholeBody (133 keypoints)
+        2. **Upload Image**: Upload an image containing one or more persons
+        3. **Configure Settings**: Adjust YOLO detection, threshold, and opacity
+        4. **Run Estimation**: Click the button to process
+        5. **View Results**: See detected keypoints overlaid on the image
+        
+        ### 📊 Model Details
+        
+        - **Body Model**: 17 COCO keypoints (nose, eyes, ears, shoulders, elbows, wrists, hips, knees, ankles)
+        - **WholeBody Model**: 133 keypoints (17 body + 68 face + 42 hands + 6 feet)
+        - YOLO detection recommended for multi-person scenes
+        - First model load may take 1-2 minutes depending on your connection
+        - Switching models will load the new model automatically
+        
+        ### 🔗 Links
+        
+        - 📄 [Read the Paper](https://arxiv.org/abs/2409.24980)
+        - 💻 [GitHub Repository](https://github.com/T-S-Liang/SDPose-OOD)
+        - 🤗 [Body Model Hub](https://huggingface.co/teemosliang/SDPose-Body)
+        - 🤗 [WholeBody Model Hub](https://huggingface.co/teemosliang/SDPose-Wholebody)
+        """)
+        
+        # Event handlers
+        model_type.change(
+            fn=switch_model,
+            inputs=[model_type],
+            outputs=[model_status]
+        )
+        
+        run_btn.click(
+            fn=run_inference,
+            inputs=[input_image, model_type, enable_yolo, score_threshold, overlay_alpha],
+            outputs=[output_image, info_output]
+        )
+    
+    return demo
+
+
+# Pre-load body model on startup (wholebody loads on demand)
+print("=" * 50)
+print("🚀 SDPose Gradio Space Starting...")
+print("=" * 50)
+print("🔄 Pre-loading Body model...")
+success = inference_engines["body"].load_model_from_hub(keypoint_scheme="body")
+if success:
+    print("✅ Body model loaded and ready!")
+    current_model = "body"
+else:
+    print("⚠️ Body model loading failed, will try on demand")
+
+print("ℹ️  WholeBody model will load when selected")
+print("=" * 50)
+
+if __name__ == "__main__":
+    demo = create_gradio_interface()
+    demo.launch()
 
-trackio.show()

configs/_base_/datasets/300w.py

@@ -0,0 +1,140 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='300w',
+    paper_info=dict(
+        author='Sagonas, Christos and Antonakos, Epameinondas '
+        'and Tzimiropoulos, Georgios and Zafeiriou, Stefanos '
+        'and Pantic, Maja',
+        title='300 faces in-the-wild challenge: '
+        'Database and results',
+        container='Image and vision computing',
+        year='2016',
+        homepage='https://ibug.doc.ic.ac.uk/resources/300-W/',
+    ),
+    keypoint_info={
+        0: dict(name='kpt-0', id=0, color=[255, 0, 0], type='', swap='kpt-16'),
+        1: dict(name='kpt-1', id=1, color=[255, 0, 0], type='', swap='kpt-15'),
+        2: dict(name='kpt-2', id=2, color=[255, 0, 0], type='', swap='kpt-14'),
+        3: dict(name='kpt-3', id=3, color=[255, 0, 0], type='', swap='kpt-13'),
+        4: dict(name='kpt-4', id=4, color=[255, 0, 0], type='', swap='kpt-12'),
+        5: dict(name='kpt-5', id=5, color=[255, 0, 0], type='', swap='kpt-11'),
+        6: dict(name='kpt-6', id=6, color=[255, 0, 0], type='', swap='kpt-10'),
+        7: dict(name='kpt-7', id=7, color=[255, 0, 0], type='', swap='kpt-9'),
+        8: dict(name='kpt-8', id=8, color=[255, 0, 0], type='', swap=''),
+        9: dict(name='kpt-9', id=9, color=[255, 0, 0], type='', swap='kpt-7'),
+        10:
+        dict(name='kpt-10', id=10, color=[255, 0, 0], type='', swap='kpt-6'),
+        11:
+        dict(name='kpt-11', id=11, color=[255, 0, 0], type='', swap='kpt-5'),
+        12:
+        dict(name='kpt-12', id=12, color=[255, 0, 0], type='', swap='kpt-4'),
+        13:
+        dict(name='kpt-13', id=13, color=[255, 0, 0], type='', swap='kpt-3'),
+        14:
+        dict(name='kpt-14', id=14, color=[255, 0, 0], type='', swap='kpt-2'),
+        15:
+        dict(name='kpt-15', id=15, color=[255, 0, 0], type='', swap='kpt-1'),
+        16:
+        dict(name='kpt-16', id=16, color=[255, 0, 0], type='', swap='kpt-0'),
+        17:
+        dict(name='kpt-17', id=17, color=[255, 0, 0], type='', swap='kpt-26'),
+        18:
+        dict(name='kpt-18', id=18, color=[255, 0, 0], type='', swap='kpt-25'),
+        19:
+        dict(name='kpt-19', id=19, color=[255, 0, 0], type='', swap='kpt-24'),
+        20:
+        dict(name='kpt-20', id=20, color=[255, 0, 0], type='', swap='kpt-23'),
+        21:
+        dict(name='kpt-21', id=21, color=[255, 0, 0], type='', swap='kpt-22'),
+        22:
+        dict(name='kpt-22', id=22, color=[255, 0, 0], type='', swap='kpt-21'),
+        23:
+        dict(name='kpt-23', id=23, color=[255, 0, 0], type='', swap='kpt-20'),
+        24:
+        dict(name='kpt-24', id=24, color=[255, 0, 0], type='', swap='kpt-19'),
+        25:
+        dict(name='kpt-25', id=25, color=[255, 0, 0], type='', swap='kpt-18'),
+        26:
+        dict(name='kpt-26', id=26, color=[255, 0, 0], type='', swap='kpt-17'),
+        27: dict(name='kpt-27', id=27, color=[255, 0, 0], type='', swap=''),
+        28: dict(name='kpt-28', id=28, color=[255, 0, 0], type='', swap=''),
+        29: dict(name='kpt-29', id=29, color=[255, 0, 0], type='', swap=''),
+        30: dict(name='kpt-30', id=30, color=[255, 0, 0], type='', swap=''),
+        31:
+        dict(name='kpt-31', id=31, color=[255, 0, 0], type='', swap='kpt-35'),
+        32:
+        dict(name='kpt-32', id=32, color=[255, 0, 0], type='', swap='kpt-34'),
+        33: dict(name='kpt-33', id=33, color=[255, 0, 0], type='', swap=''),
+        34:
+        dict(name='kpt-34', id=34, color=[255, 0, 0], type='', swap='kpt-32'),
+        35:
+        dict(name='kpt-35', id=35, color=[255, 0, 0], type='', swap='kpt-31'),
+        36:
+        dict(name='kpt-36', id=36, color=[255, 0, 0], type='', swap='kpt-45'),
+        37:
+        dict(name='kpt-37', id=37, color=[255, 0, 0], type='', swap='kpt-44'),
+        38:
+        dict(name='kpt-38', id=38, color=[255, 0, 0], type='', swap='kpt-43'),
+        39:
+        dict(name='kpt-39', id=39, color=[255, 0, 0], type='', swap='kpt-42'),
+        40:
+        dict(name='kpt-40', id=40, color=[255, 0, 0], type='', swap='kpt-47'),
+        41: dict(
+            name='kpt-41', id=41, color=[255, 0, 0], type='', swap='kpt-46'),
+        42: dict(
+            name='kpt-42', id=42, color=[255, 0, 0], type='', swap='kpt-39'),
+        43: dict(
+            name='kpt-43', id=43, color=[255, 0, 0], type='', swap='kpt-38'),
+        44: dict(
+            name='kpt-44', id=44, color=[255, 0, 0], type='', swap='kpt-37'),
+        45: dict(
+            name='kpt-45', id=45, color=[255, 0, 0], type='', swap='kpt-36'),
+        46: dict(
+            name='kpt-46', id=46, color=[255, 0, 0], type='', swap='kpt-41'),
+        47: dict(
+            name='kpt-47', id=47, color=[255, 0, 0], type='', swap='kpt-40'),
+        48: dict(
+            name='kpt-48', id=48, color=[255, 0, 0], type='', swap='kpt-54'),
+        49: dict(
+            name='kpt-49', id=49, color=[255, 0, 0], type='', swap='kpt-53'),
+        50: dict(
+            name='kpt-50', id=50, color=[255, 0, 0], type='', swap='kpt-52'),
+        51: dict(name='kpt-51', id=51, color=[255, 0, 0], type='', swap=''),
+        52: dict(
+            name='kpt-52', id=52, color=[255, 0, 0], type='', swap='kpt-50'),
+        53: dict(
+            name='kpt-53', id=53, color=[255, 0, 0], type='', swap='kpt-49'),
+        54: dict(
+            name='kpt-54', id=54, color=[255, 0, 0], type='', swap='kpt-48'),
+        55: dict(
+            name='kpt-55', id=55, color=[255, 0, 0], type='', swap='kpt-59'),
+        56: dict(
+            name='kpt-56', id=56, color=[255, 0, 0], type='', swap='kpt-58'),
+        57: dict(name='kpt-57', id=57, color=[255, 0, 0], type='', swap=''),
+        58: dict(
+            name='kpt-58', id=58, color=[255, 0, 0], type='', swap='kpt-56'),
+        59: dict(
+            name='kpt-59', id=59, color=[255, 0, 0], type='', swap='kpt-55'),
+        60: dict(
+            name='kpt-60', id=60, color=[255, 0, 0], type='', swap='kpt-64'),
+        61: dict(
+            name='kpt-61', id=61, color=[255, 0, 0], type='', swap='kpt-63'),
+        62: dict(name='kpt-62', id=62, color=[255, 0, 0], type='', swap=''),
+        63: dict(
+            name='kpt-63', id=63, color=[255, 0, 0], type='', swap='kpt-61'),
+        64: dict(
+            name='kpt-64', id=64, color=[255, 0, 0], type='', swap='kpt-60'),
+        65: dict(
+            name='kpt-65', id=65, color=[255, 0, 0], type='', swap='kpt-67'),
+        66: dict(name='kpt-66', id=66, color=[255, 0, 0], type='', swap=''),
+        67: dict(
+            name='kpt-67', id=67, color=[255, 0, 0], type='', swap='kpt-65'),
+    },
+    skeleton_info={},
+    joint_weights=[1.] * 68,
+    sigmas=[])

configs/_base_/datasets/aflw.py

@@ -0,0 +1,50 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='aflw',
+    paper_info=dict(
+        author='Koestinger, Martin and Wohlhart, Paul and '
+        'Roth, Peter M and Bischof, Horst',
+        title='Annotated facial landmarks in the wild: '
+        'A large-scale, real-world database for facial '
+        'landmark localization',
+        container='2011 IEEE international conference on computer '
+        'vision workshops (ICCV workshops)',
+        year='2011',
+        homepage='https://www.tugraz.at/institute/icg/research/'
+        'team-bischof/lrs/downloads/aflw/',
+    ),
+    keypoint_info={
+        0: dict(name='kpt-0', id=0, color=[255, 0, 0], type='', swap='kpt-5'),
+        1: dict(name='kpt-1', id=1, color=[255, 0, 0], type='', swap='kpt-4'),
+        2: dict(name='kpt-2', id=2, color=[255, 0, 0], type='', swap='kpt-3'),
+        3: dict(name='kpt-3', id=3, color=[255, 0, 0], type='', swap='kpt-2'),
+        4: dict(name='kpt-4', id=4, color=[255, 0, 0], type='', swap='kpt-1'),
+        5: dict(name='kpt-5', id=5, color=[255, 0, 0], type='', swap='kpt-0'),
+        6: dict(name='kpt-6', id=6, color=[255, 0, 0], type='', swap='kpt-11'),
+        7: dict(name='kpt-7', id=7, color=[255, 0, 0], type='', swap='kpt-10'),
+        8: dict(name='kpt-8', id=8, color=[255, 0, 0], type='', swap='kpt-9'),
+        9: dict(name='kpt-9', id=9, color=[255, 0, 0], type='', swap='kpt-8'),
+        10:
+        dict(name='kpt-10', id=10, color=[255, 0, 0], type='', swap='kpt-7'),
+        11:
+        dict(name='kpt-11', id=11, color=[255, 0, 0], type='', swap='kpt-6'),
+        12:
+        dict(name='kpt-12', id=12, color=[255, 0, 0], type='', swap='kpt-14'),
+        13: dict(name='kpt-13', id=13, color=[255, 0, 0], type='', swap=''),
+        14:
+        dict(name='kpt-14', id=14, color=[255, 0, 0], type='', swap='kpt-12'),
+        15:
+        dict(name='kpt-15', id=15, color=[255, 0, 0], type='', swap='kpt-17'),
+        16: dict(name='kpt-16', id=16, color=[255, 0, 0], type='', swap=''),
+        17:
+        dict(name='kpt-17', id=17, color=[255, 0, 0], type='', swap='kpt-15'),
+        18: dict(name='kpt-18', id=18, color=[255, 0, 0], type='', swap='')
+    },
+    skeleton_info={},
+    joint_weights=[1.] * 19,
+    sigmas=[])

configs/_base_/datasets/aic.py

@@ -0,0 +1,146 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='aic',
+    paper_info=dict(
+        author='Wu, Jiahong and Zheng, He and Zhao, Bo and '
+        'Li, Yixin and Yan, Baoming and Liang, Rui and '
+        'Wang, Wenjia and Zhou, Shipei and Lin, Guosen and '
+        'Fu, Yanwei and others',
+        title='Ai challenger: A large-scale dataset for going '
+        'deeper in image understanding',
+        container='arXiv',
+        year='2017',
+        homepage='https://github.com/AIChallenger/AI_Challenger_2017',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='right_shoulder',
+            id=0,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        1:
+        dict(
+            name='right_elbow',
+            id=1,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        2:
+        dict(
+            name='right_wrist',
+            id=2,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        3:
+        dict(
+            name='left_shoulder',
+            id=3,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        4:
+        dict(
+            name='left_elbow',
+            id=4,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        5:
+        dict(
+            name='left_wrist',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        6:
+        dict(
+            name='right_hip',
+            id=6,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        7:
+        dict(
+            name='right_knee',
+            id=7,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        8:
+        dict(
+            name='right_ankle',
+            id=8,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle'),
+        9:
+        dict(
+            name='left_hip',
+            id=9,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        10:
+        dict(
+            name='left_knee',
+            id=10,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        11:
+        dict(
+            name='left_ankle',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        12:
+        dict(
+            name='head_top',
+            id=12,
+            color=[51, 153, 255],
+            type='upper',
+            swap=''),
+        13:
+        dict(name='neck', id=13, color=[51, 153, 255], type='upper', swap='')
+    },
+    skeleton_info={
+        0:
+        dict(link=('right_wrist', 'right_elbow'), id=0, color=[255, 128, 0]),
+        1: dict(
+            link=('right_elbow', 'right_shoulder'), id=1, color=[255, 128, 0]),
+        2: dict(link=('right_shoulder', 'neck'), id=2, color=[51, 153, 255]),
+        3: dict(link=('neck', 'left_shoulder'), id=3, color=[51, 153, 255]),
+        4: dict(link=('left_shoulder', 'left_elbow'), id=4, color=[0, 255, 0]),
+        5: dict(link=('left_elbow', 'left_wrist'), id=5, color=[0, 255, 0]),
+        6: dict(link=('right_ankle', 'right_knee'), id=6, color=[255, 128, 0]),
+        7: dict(link=('right_knee', 'right_hip'), id=7, color=[255, 128, 0]),
+        8: dict(link=('right_hip', 'left_hip'), id=8, color=[51, 153, 255]),
+        9: dict(link=('left_hip', 'left_knee'), id=9, color=[0, 255, 0]),
+        10: dict(link=('left_knee', 'left_ankle'), id=10, color=[0, 255, 0]),
+        11: dict(link=('head_top', 'neck'), id=11, color=[51, 153, 255]),
+        12: dict(
+            link=('right_shoulder', 'right_hip'), id=12, color=[51, 153, 255]),
+        13:
+        dict(link=('left_shoulder', 'left_hip'), id=13, color=[51, 153, 255])
+    },
+    joint_weights=[
+        1., 1.2, 1.5, 1., 1.2, 1.5, 1., 1.2, 1.5, 1., 1.2, 1.5, 1., 1.
+    ],
+
+    # 'https://github.com/AIChallenger/AI_Challenger_2017/blob/master/'
+    # 'Evaluation/keypoint_eval/keypoint_eval.py#L50'
+    # delta = 2 x sigma
+    sigmas=[
+        0.01388152, 0.01515228, 0.01057665, 0.01417709, 0.01497891, 0.01402144,
+        0.03909642, 0.03686941, 0.01981803, 0.03843971, 0.03412318, 0.02415081,
+        0.01291456, 0.01236173
+    ])

configs/_base_/datasets/aic2coco.py

@@ -0,0 +1,187 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='aic2coco',
+    paper_info=dict(
+        author='Lin, Tsung-Yi and Maire, Michael and '
+        'Belongie, Serge and Hays, James and '
+        'Perona, Pietro and Ramanan, Deva and '
+        r'Doll{\'a}r, Piotr and Zitnick, C Lawrence',
+        title='Microsoft coco: Common objects in context',
+        container='European conference on computer vision',
+        year='2014',
+        homepage='http://cocodataset.org/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
+        1:
+        dict(
+            name='left_eye',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_eye'),
+        2:
+        dict(
+            name='right_eye',
+            id=2,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_eye'),
+        3:
+        dict(
+            name='left_ear',
+            id=3,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_ear'),
+        4:
+        dict(
+            name='right_ear',
+            id=4,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_ear'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='right_shoulder',
+            id=6,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        7:
+        dict(
+            name='left_elbow',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        8:
+        dict(
+            name='right_elbow',
+            id=8,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        9:
+        dict(
+            name='left_wrist',
+            id=9,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='right_hip',
+            id=12,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        13:
+        dict(
+            name='left_knee',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        14:
+        dict(
+            name='right_knee',
+            id=14,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        15:
+        dict(
+            name='left_ankle',
+            id=15,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        16:
+        dict(
+            name='right_ankle',
+            id=16,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle')
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
+        5:
+        dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
+        6:
+        dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
+        7:
+        dict(
+            link=('left_shoulder', 'right_shoulder'),
+            id=7,
+            color=[51, 153, 255]),
+        8:
+        dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
+        9:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
+        13:
+        dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
+        14:
+        dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
+        15:
+        dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
+        16:
+        dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
+        17:
+        dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
+        18:
+        dict(
+            link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255])
+    },
+    joint_weights=[
+        1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
+        1.5
+    ],
+    sigmas=[
+        0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
+        0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089
+    ])

configs/_base_/datasets/aic2goliath.py

@@ -0,0 +1,11 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+from configs._base_.datasets.goliath import dataset_info as goliath_info
+
+goliath_info = goliath_info.build()
+dataset_info = goliath_info.copy()
+dataset_info['dataset_name'] = 'aic2goliath'

configs/_base_/datasets/aic2goliath3d.py

@@ -0,0 +1,11 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+from configs._base_.datasets.goliath3d import dataset_info as goliath_info
+
+goliath_info = goliath_info.build()
+dataset_info = goliath_info.copy()
+dataset_info['dataset_name'] = 'aic2goliath3d'

configs/_base_/datasets/ak.py

@@ -0,0 +1,273 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='Animal Kingdom',
+    paper_info=dict(
+        author='Singapore University of Technology and Design, Singapore.'
+        ' Xun Long Ng, Kian Eng Ong, Qichen Zheng,'
+        ' Yun Ni, Si Yong Yeo, Jun Liu.',
+        title='Animal Kingdom: '
+        'A Large and Diverse Dataset for Animal Behavior Understanding',
+        container='Conference on Computer Vision '
+        'and Pattern Recognition (CVPR)',
+        year='2022',
+        homepage='https://sutdcv.github.io/Animal-Kingdom',
+        version='1.0 (2022-06)',
+        date_created='2022-06',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='Head_Mid_Top',
+            id=0,
+            color=(225, 0, 255),
+            type='upper',
+            swap=''),
+        1:
+        dict(
+            name='Eye_Left',
+            id=1,
+            color=[220, 20, 60],
+            type='upper',
+            swap='Eye_Right'),
+        2:
+        dict(
+            name='Eye_Right',
+            id=2,
+            color=[0, 255, 255],
+            type='upper',
+            swap='Eye_Left'),
+        3:
+        dict(
+            name='Mouth_Front_Top',
+            id=3,
+            color=(0, 255, 42),
+            type='upper',
+            swap=''),
+        4:
+        dict(
+            name='Mouth_Back_Left',
+            id=4,
+            color=[221, 160, 221],
+            type='upper',
+            swap='Mouth_Back_Right'),
+        5:
+        dict(
+            name='Mouth_Back_Right',
+            id=5,
+            color=[135, 206, 250],
+            type='upper',
+            swap='Mouth_Back_Left'),
+        6:
+        dict(
+            name='Mouth_Front_Bottom',
+            id=6,
+            color=[50, 205, 50],
+            type='upper',
+            swap=''),
+        7:
+        dict(
+            name='Shoulder_Left',
+            id=7,
+            color=[255, 182, 193],
+            type='upper',
+            swap='Shoulder_Right'),
+        8:
+        dict(
+            name='Shoulder_Right',
+            id=8,
+            color=[0, 191, 255],
+            type='upper',
+            swap='Shoulder_Left'),
+        9:
+        dict(
+            name='Elbow_Left',
+            id=9,
+            color=[255, 105, 180],
+            type='upper',
+            swap='Elbow_Right'),
+        10:
+        dict(
+            name='Elbow_Right',
+            id=10,
+            color=[30, 144, 255],
+            type='upper',
+            swap='Elbow_Left'),
+        11:
+        dict(
+            name='Wrist_Left',
+            id=11,
+            color=[255, 20, 147],
+            type='upper',
+            swap='Wrist_Right'),
+        12:
+        dict(
+            name='Wrist_Right',
+            id=12,
+            color=[0, 0, 255],
+            type='upper',
+            swap='Wrist_Left'),
+        13:
+        dict(
+            name='Torso_Mid_Back',
+            id=13,
+            color=(185, 3, 221),
+            type='upper',
+            swap=''),
+        14:
+        dict(
+            name='Hip_Left',
+            id=14,
+            color=[255, 215, 0],
+            type='lower',
+            swap='Hip_Right'),
+        15:
+        dict(
+            name='Hip_Right',
+            id=15,
+            color=[147, 112, 219],
+            type='lower',
+            swap='Hip_Left'),
+        16:
+        dict(
+            name='Knee_Left',
+            id=16,
+            color=[255, 165, 0],
+            type='lower',
+            swap='Knee_Right'),
+        17:
+        dict(
+            name='Knee_Right',
+            id=17,
+            color=[138, 43, 226],
+            type='lower',
+            swap='Knee_Left'),
+        18:
+        dict(
+            name='Ankle_Left',
+            id=18,
+            color=[255, 140, 0],
+            type='lower',
+            swap='Ankle_Right'),
+        19:
+        dict(
+            name='Ankle_Right',
+            id=19,
+            color=[128, 0, 128],
+            type='lower',
+            swap='Ankle_Left'),
+        20:
+        dict(
+            name='Tail_Top_Back',
+            id=20,
+            color=(0, 251, 255),
+            type='lower',
+            swap=''),
+        21:
+        dict(
+            name='Tail_Mid_Back',
+            id=21,
+            color=[32, 178, 170],
+            type='lower',
+            swap=''),
+        22:
+        dict(
+            name='Tail_End_Back',
+            id=22,
+            color=(0, 102, 102),
+            type='lower',
+            swap='')
+    },
+    skeleton_info={
+        0:
+        dict(link=('Eye_Left', 'Head_Mid_Top'), id=0, color=[220, 20, 60]),
+        1:
+        dict(link=('Eye_Right', 'Head_Mid_Top'), id=1, color=[0, 255, 255]),
+        2:
+        dict(
+            link=('Mouth_Front_Top', 'Mouth_Back_Left'),
+            id=2,
+            color=[221, 160, 221]),
+        3:
+        dict(
+            link=('Mouth_Front_Top', 'Mouth_Back_Right'),
+            id=3,
+            color=[135, 206, 250]),
+        4:
+        dict(
+            link=('Mouth_Front_Bottom', 'Mouth_Back_Left'),
+            id=4,
+            color=[221, 160, 221]),
+        5:
+        dict(
+            link=('Mouth_Front_Bottom', 'Mouth_Back_Right'),
+            id=5,
+            color=[135, 206, 250]),
+        6:
+        dict(
+            link=('Head_Mid_Top', 'Torso_Mid_Back'), id=6,
+            color=(225, 0, 255)),
+        7:
+        dict(
+            link=('Torso_Mid_Back', 'Tail_Top_Back'),
+            id=7,
+            color=(185, 3, 221)),
+        8:
+        dict(
+            link=('Tail_Top_Back', 'Tail_Mid_Back'), id=8,
+            color=(0, 251, 255)),
+        9:
+        dict(
+            link=('Tail_Mid_Back', 'Tail_End_Back'),
+            id=9,
+            color=[32, 178, 170]),
+        10:
+        dict(
+            link=('Head_Mid_Top', 'Shoulder_Left'),
+            id=10,
+            color=[255, 182, 193]),
+        11:
+        dict(
+            link=('Head_Mid_Top', 'Shoulder_Right'),
+            id=11,
+            color=[0, 191, 255]),
+        12:
+        dict(
+            link=('Shoulder_Left', 'Elbow_Left'), id=12, color=[255, 105,
+                                                                180]),
+        13:
+        dict(
+            link=('Shoulder_Right', 'Elbow_Right'),
+            id=13,
+            color=[30, 144, 255]),
+        14:
+        dict(link=('Elbow_Left', 'Wrist_Left'), id=14, color=[255, 20, 147]),
+        15:
+        dict(link=('Elbow_Right', 'Wrist_Right'), id=15, color=[0, 0, 255]),
+        16:
+        dict(link=('Tail_Top_Back', 'Hip_Left'), id=16, color=[255, 215, 0]),
+        17:
+        dict(
+            link=('Tail_Top_Back', 'Hip_Right'), id=17, color=[147, 112, 219]),
+        18:
+        dict(link=('Hip_Left', 'Knee_Left'), id=18, color=[255, 165, 0]),
+        19:
+        dict(link=('Hip_Right', 'Knee_Right'), id=19, color=[138, 43, 226]),
+        20:
+        dict(link=('Knee_Left', 'Ankle_Left'), id=20, color=[255, 140, 0]),
+        21:
+        dict(link=('Knee_Right', 'Ankle_Right'), id=21, color=[128, 0, 128])
+    },
+    joint_weights=[
+        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.,
+        1., 1., 1., 1., 1.
+    ],
+    sigmas=[
+        0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025,
+        0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025, 0.025,
+        0.025, 0.025, 0.025
+    ])

configs/_base_/datasets/animalpose.py

@@ -0,0 +1,172 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='animalpose',
+    paper_info=dict(
+        author='Cao, Jinkun and Tang, Hongyang and Fang, Hao-Shu and '
+        'Shen, Xiaoyong and Lu, Cewu and Tai, Yu-Wing',
+        title='Cross-Domain Adaptation for Animal Pose Estimation',
+        container='The IEEE International Conference on '
+        'Computer Vision (ICCV)',
+        year='2019',
+        homepage='https://sites.google.com/view/animal-pose/',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='L_Eye', id=0, color=[0, 255, 0], type='upper', swap='R_Eye'),
+        1:
+        dict(
+            name='R_Eye',
+            id=1,
+            color=[255, 128, 0],
+            type='upper',
+            swap='L_Eye'),
+        2:
+        dict(
+            name='L_EarBase',
+            id=2,
+            color=[0, 255, 0],
+            type='upper',
+            swap='R_EarBase'),
+        3:
+        dict(
+            name='R_EarBase',
+            id=3,
+            color=[255, 128, 0],
+            type='upper',
+            swap='L_EarBase'),
+        4:
+        dict(name='Nose', id=4, color=[51, 153, 255], type='upper', swap=''),
+        5:
+        dict(name='Throat', id=5, color=[51, 153, 255], type='upper', swap=''),
+        6:
+        dict(
+            name='TailBase', id=6, color=[51, 153, 255], type='lower',
+            swap=''),
+        7:
+        dict(
+            name='Withers', id=7, color=[51, 153, 255], type='upper', swap=''),
+        8:
+        dict(
+            name='L_F_Elbow',
+            id=8,
+            color=[0, 255, 0],
+            type='upper',
+            swap='R_F_Elbow'),
+        9:
+        dict(
+            name='R_F_Elbow',
+            id=9,
+            color=[255, 128, 0],
+            type='upper',
+            swap='L_F_Elbow'),
+        10:
+        dict(
+            name='L_B_Elbow',
+            id=10,
+            color=[0, 255, 0],
+            type='lower',
+            swap='R_B_Elbow'),
+        11:
+        dict(
+            name='R_B_Elbow',
+            id=11,
+            color=[255, 128, 0],
+            type='lower',
+            swap='L_B_Elbow'),
+        12:
+        dict(
+            name='L_F_Knee',
+            id=12,
+            color=[0, 255, 0],
+            type='upper',
+            swap='R_F_Knee'),
+        13:
+        dict(
+            name='R_F_Knee',
+            id=13,
+            color=[255, 128, 0],
+            type='upper',
+            swap='L_F_Knee'),
+        14:
+        dict(
+            name='L_B_Knee',
+            id=14,
+            color=[0, 255, 0],
+            type='lower',
+            swap='R_B_Knee'),
+        15:
+        dict(
+            name='R_B_Knee',
+            id=15,
+            color=[255, 128, 0],
+            type='lower',
+            swap='L_B_Knee'),
+        16:
+        dict(
+            name='L_F_Paw',
+            id=16,
+            color=[0, 255, 0],
+            type='upper',
+            swap='R_F_Paw'),
+        17:
+        dict(
+            name='R_F_Paw',
+            id=17,
+            color=[255, 128, 0],
+            type='upper',
+            swap='L_F_Paw'),
+        18:
+        dict(
+            name='L_B_Paw',
+            id=18,
+            color=[0, 255, 0],
+            type='lower',
+            swap='R_B_Paw'),
+        19:
+        dict(
+            name='R_B_Paw',
+            id=19,
+            color=[255, 128, 0],
+            type='lower',
+            swap='L_B_Paw')
+    },
+    skeleton_info={
+        0: dict(link=('L_Eye', 'R_Eye'), id=0, color=[51, 153, 255]),
+        1: dict(link=('L_Eye', 'L_EarBase'), id=1, color=[0, 255, 0]),
+        2: dict(link=('R_Eye', 'R_EarBase'), id=2, color=[255, 128, 0]),
+        3: dict(link=('L_Eye', 'Nose'), id=3, color=[0, 255, 0]),
+        4: dict(link=('R_Eye', 'Nose'), id=4, color=[255, 128, 0]),
+        5: dict(link=('Nose', 'Throat'), id=5, color=[51, 153, 255]),
+        6: dict(link=('Throat', 'Withers'), id=6, color=[51, 153, 255]),
+        7: dict(link=('TailBase', 'Withers'), id=7, color=[51, 153, 255]),
+        8: dict(link=('Throat', 'L_F_Elbow'), id=8, color=[0, 255, 0]),
+        9: dict(link=('L_F_Elbow', 'L_F_Knee'), id=9, color=[0, 255, 0]),
+        10: dict(link=('L_F_Knee', 'L_F_Paw'), id=10, color=[0, 255, 0]),
+        11: dict(link=('Throat', 'R_F_Elbow'), id=11, color=[255, 128, 0]),
+        12: dict(link=('R_F_Elbow', 'R_F_Knee'), id=12, color=[255, 128, 0]),
+        13: dict(link=('R_F_Knee', 'R_F_Paw'), id=13, color=[255, 128, 0]),
+        14: dict(link=('TailBase', 'L_B_Elbow'), id=14, color=[0, 255, 0]),
+        15: dict(link=('L_B_Elbow', 'L_B_Knee'), id=15, color=[0, 255, 0]),
+        16: dict(link=('L_B_Knee', 'L_B_Paw'), id=16, color=[0, 255, 0]),
+        17: dict(link=('TailBase', 'R_B_Elbow'), id=17, color=[255, 128, 0]),
+        18: dict(link=('R_B_Elbow', 'R_B_Knee'), id=18, color=[255, 128, 0]),
+        19: dict(link=('R_B_Knee', 'R_B_Paw'), id=19, color=[255, 128, 0])
+    },
+    joint_weights=[
+        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.2, 1.2,
+        1.5, 1.5, 1.5, 1.5
+    ],
+
+    # Note: The original paper did not provide enough information about
+    # the sigmas. We modified from 'https://github.com/cocodataset/'
+    # 'cocoapi/blob/master/PythonAPI/pycocotools/cocoeval.py#L523'
+    sigmas=[
+        0.025, 0.025, 0.026, 0.035, 0.035, 0.10, 0.10, 0.10, 0.107, 0.107,
+        0.107, 0.107, 0.087, 0.087, 0.087, 0.087, 0.089, 0.089, 0.089, 0.089
+    ])

configs/_base_/datasets/ap10k.py

@@ -0,0 +1,148 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='ap10k',
+    paper_info=dict(
+        author='Yu, Hang and Xu, Yufei and Zhang, Jing and '
+        'Zhao, Wei and Guan, Ziyu and Tao, Dacheng',
+        title='AP-10K: A Benchmark for Animal Pose Estimation in the Wild',
+        container='35th Conference on Neural Information Processing Systems '
+        '(NeurIPS 2021) Track on Datasets and Bench-marks.',
+        year='2021',
+        homepage='https://github.com/AlexTheBad/AP-10K',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='L_Eye', id=0, color=[0, 255, 0], type='upper', swap='R_Eye'),
+        1:
+        dict(
+            name='R_Eye',
+            id=1,
+            color=[255, 128, 0],
+            type='upper',
+            swap='L_Eye'),
+        2:
+        dict(name='Nose', id=2, color=[51, 153, 255], type='upper', swap=''),
+        3:
+        dict(name='Neck', id=3, color=[51, 153, 255], type='upper', swap=''),
+        4:
+        dict(
+            name='Root of tail',
+            id=4,
+            color=[51, 153, 255],
+            type='lower',
+            swap=''),
+        5:
+        dict(
+            name='L_Shoulder',
+            id=5,
+            color=[51, 153, 255],
+            type='upper',
+            swap='R_Shoulder'),
+        6:
+        dict(
+            name='L_Elbow',
+            id=6,
+            color=[51, 153, 255],
+            type='upper',
+            swap='R_Elbow'),
+        7:
+        dict(
+            name='L_F_Paw',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='R_F_Paw'),
+        8:
+        dict(
+            name='R_Shoulder',
+            id=8,
+            color=[0, 255, 0],
+            type='upper',
+            swap='L_Shoulder'),
+        9:
+        dict(
+            name='R_Elbow',
+            id=9,
+            color=[255, 128, 0],
+            type='upper',
+            swap='L_Elbow'),
+        10:
+        dict(
+            name='R_F_Paw',
+            id=10,
+            color=[0, 255, 0],
+            type='lower',
+            swap='L_F_Paw'),
+        11:
+        dict(
+            name='L_Hip',
+            id=11,
+            color=[255, 128, 0],
+            type='lower',
+            swap='R_Hip'),
+        12:
+        dict(
+            name='L_Knee',
+            id=12,
+            color=[255, 128, 0],
+            type='lower',
+            swap='R_Knee'),
+        13:
+        dict(
+            name='L_B_Paw',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='R_B_Paw'),
+        14:
+        dict(
+            name='R_Hip', id=14, color=[0, 255, 0], type='lower',
+            swap='L_Hip'),
+        15:
+        dict(
+            name='R_Knee',
+            id=15,
+            color=[0, 255, 0],
+            type='lower',
+            swap='L_Knee'),
+        16:
+        dict(
+            name='R_B_Paw',
+            id=16,
+            color=[0, 255, 0],
+            type='lower',
+            swap='L_B_Paw'),
+    },
+    skeleton_info={
+        0: dict(link=('L_Eye', 'R_Eye'), id=0, color=[0, 0, 255]),
+        1: dict(link=('L_Eye', 'Nose'), id=1, color=[0, 0, 255]),
+        2: dict(link=('R_Eye', 'Nose'), id=2, color=[0, 0, 255]),
+        3: dict(link=('Nose', 'Neck'), id=3, color=[0, 255, 0]),
+        4: dict(link=('Neck', 'Root of tail'), id=4, color=[0, 255, 0]),
+        5: dict(link=('Neck', 'L_Shoulder'), id=5, color=[0, 255, 255]),
+        6: dict(link=('L_Shoulder', 'L_Elbow'), id=6, color=[0, 255, 255]),
+        7: dict(link=('L_Elbow', 'L_F_Paw'), id=6, color=[0, 255, 255]),
+        8: dict(link=('Neck', 'R_Shoulder'), id=7, color=[6, 156, 250]),
+        9: dict(link=('R_Shoulder', 'R_Elbow'), id=8, color=[6, 156, 250]),
+        10: dict(link=('R_Elbow', 'R_F_Paw'), id=9, color=[6, 156, 250]),
+        11: dict(link=('Root of tail', 'L_Hip'), id=10, color=[0, 255, 255]),
+        12: dict(link=('L_Hip', 'L_Knee'), id=11, color=[0, 255, 255]),
+        13: dict(link=('L_Knee', 'L_B_Paw'), id=12, color=[0, 255, 255]),
+        14: dict(link=('Root of tail', 'R_Hip'), id=13, color=[6, 156, 250]),
+        15: dict(link=('R_Hip', 'R_Knee'), id=14, color=[6, 156, 250]),
+        16: dict(link=('R_Knee', 'R_B_Paw'), id=15, color=[6, 156, 250]),
+    },
+    joint_weights=[
+        1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
+        1.5
+    ],
+    sigmas=[
+        0.025, 0.025, 0.026, 0.035, 0.035, 0.079, 0.072, 0.062, 0.079, 0.072,
+        0.062, 0.107, 0.087, 0.089, 0.107, 0.087, 0.089
+    ])

configs/_base_/datasets/atrw.py

@@ -0,0 +1,150 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='atrw',
+    paper_info=dict(
+        author='Li, Shuyuan and Li, Jianguo and Tang, Hanlin '
+        'and Qian, Rui and Lin, Weiyao',
+        title='ATRW: A Benchmark for Amur Tiger '
+        'Re-identification in the Wild',
+        container='Proceedings of the 28th ACM '
+        'International Conference on Multimedia',
+        year='2020',
+        homepage='https://cvwc2019.github.io/challenge.html',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='left_ear',
+            id=0,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_ear'),
+        1:
+        dict(
+            name='right_ear',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_ear'),
+        2:
+        dict(name='nose', id=2, color=[51, 153, 255], type='upper', swap=''),
+        3:
+        dict(
+            name='right_shoulder',
+            id=3,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        4:
+        dict(
+            name='right_front_paw',
+            id=4,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_front_paw'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='left_front_paw',
+            id=6,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_front_paw'),
+        7:
+        dict(
+            name='right_hip',
+            id=7,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        8:
+        dict(
+            name='right_knee',
+            id=8,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        9:
+        dict(
+            name='right_back_paw',
+            id=9,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_back_paw'),
+        10:
+        dict(
+            name='left_hip',
+            id=10,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        11:
+        dict(
+            name='left_knee',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        12:
+        dict(
+            name='left_back_paw',
+            id=12,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_back_paw'),
+        13:
+        dict(name='tail', id=13, color=[51, 153, 255], type='lower', swap=''),
+        14:
+        dict(
+            name='center', id=14, color=[51, 153, 255], type='lower', swap=''),
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ear', 'nose'), id=0, color=[51, 153, 255]),
+        1:
+        dict(link=('right_ear', 'nose'), id=1, color=[51, 153, 255]),
+        2:
+        dict(link=('nose', 'center'), id=2, color=[51, 153, 255]),
+        3:
+        dict(
+            link=('left_shoulder', 'left_front_paw'), id=3, color=[0, 255, 0]),
+        4:
+        dict(link=('left_shoulder', 'center'), id=4, color=[0, 255, 0]),
+        5:
+        dict(
+            link=('right_shoulder', 'right_front_paw'),
+            id=5,
+            color=[255, 128, 0]),
+        6:
+        dict(link=('right_shoulder', 'center'), id=6, color=[255, 128, 0]),
+        7:
+        dict(link=('tail', 'center'), id=7, color=[51, 153, 255]),
+        8:
+        dict(link=('right_back_paw', 'right_knee'), id=8, color=[255, 128, 0]),
+        9:
+        dict(link=('right_knee', 'right_hip'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('right_hip', 'tail'), id=10, color=[255, 128, 0]),
+        11:
+        dict(link=('left_back_paw', 'left_knee'), id=11, color=[0, 255, 0]),
+        12:
+        dict(link=('left_knee', 'left_hip'), id=12, color=[0, 255, 0]),
+        13:
+        dict(link=('left_hip', 'tail'), id=13, color=[0, 255, 0]),
+    },
+    joint_weights=[1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.],
+    sigmas=[
+        0.0277, 0.0823, 0.0831, 0.0202, 0.0716, 0.0263, 0.0646, 0.0302, 0.0440,
+        0.0316, 0.0333, 0.0547, 0.0263, 0.0683, 0.0539
+    ])

configs/_base_/datasets/campus.py

@@ -0,0 +1,157 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='campus',
+    paper_info=dict(
+        author='Belagiannis, Vasileios and Amin, Sikandar and Andriluka, '
+        'Mykhaylo and Schiele, Bernt and Navab, Nassir and Ilic, Slobodan',
+        title='3D Pictorial Structures for Multiple Human Pose Estimation',
+        container='IEEE Computer Society Conference on Computer Vision and '
+        'Pattern Recognition (CVPR)',
+        year='2014',
+        homepage='http://campar.in.tum.de/Chair/MultiHumanPose',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='right_ankle',
+            id=0,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle'),
+        1:
+        dict(
+            name='right_knee',
+            id=1,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        2:
+        dict(
+            name='right_hip',
+            id=2,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        3:
+        dict(
+            name='left_hip',
+            id=3,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        4:
+        dict(
+            name='left_knee',
+            id=4,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        5:
+        dict(
+            name='left_ankle',
+            id=5,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        6:
+        dict(
+            name='right_wrist',
+            id=6,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        7:
+        dict(
+            name='right_elbow',
+            id=7,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        8:
+        dict(
+            name='right_shoulder',
+            id=8,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        9:
+        dict(
+            name='left_shoulder',
+            id=9,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        10:
+        dict(
+            name='left_elbow',
+            id=10,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        11:
+        dict(
+            name='left_wrist',
+            id=11,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        12:
+        dict(
+            name='bottom_head',
+            id=12,
+            color=[51, 153, 255],
+            type='upper',
+            swap=''),
+        13:
+        dict(
+            name='top_head',
+            id=13,
+            color=[51, 153, 255],
+            type='upper',
+            swap=''),
+    },
+    skeleton_info={
+        0:
+        dict(link=('right_ankle', 'right_knee'), id=0, color=[255, 128, 0]),
+        1:
+        dict(link=('right_knee', 'right_hip'), id=1, color=[255, 128, 0]),
+        2:
+        dict(link=('left_hip', 'left_knee'), id=2, color=[0, 255, 0]),
+        3:
+        dict(link=('left_knee', 'left_ankle'), id=3, color=[0, 255, 0]),
+        4:
+        dict(link=('right_hip', 'left_hip'), id=4, color=[51, 153, 255]),
+        5:
+        dict(link=('right_wrist', 'right_elbow'), id=5, color=[255, 128, 0]),
+        6:
+        dict(
+            link=('right_elbow', 'right_shoulder'), id=6, color=[255, 128, 0]),
+        7:
+        dict(link=('left_shoulder', 'left_elbow'), id=7, color=[0, 255, 0]),
+        8:
+        dict(link=('left_elbow', 'left_wrist'), id=8, color=[0, 255, 0]),
+        9:
+        dict(link=('right_hip', 'right_shoulder'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('left_hip', 'left_shoulder'), id=10, color=[0, 255, 0]),
+        11:
+        dict(
+            link=('right_shoulder', 'bottom_head'), id=11, color=[255, 128,
+                                                                  0]),
+        12:
+        dict(link=('left_shoulder', 'bottom_head'), id=12, color=[0, 255, 0]),
+        13:
+        dict(link=('bottom_head', 'top_head'), id=13, color=[51, 153, 255]),
+    },
+    joint_weights=[
+        1.5, 1.2, 1.0, 1.0, 1.2, 1.5, 1.5, 1.2, 1.0, 1.0, 1.2, 1.5, 1.0, 1.0
+    ],
+    sigmas=[
+        0.089, 0.087, 0.107, 0.107, 0.087, 0.089, 0.062, 0.072, 0.079, 0.079,
+        0.072, 0.062, 0.026, 0.026
+    ])

configs/_base_/datasets/coco.py

@@ -0,0 +1,187 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='coco',
+    paper_info=dict(
+        author='Lin, Tsung-Yi and Maire, Michael and '
+        'Belongie, Serge and Hays, James and '
+        'Perona, Pietro and Ramanan, Deva and '
+        r'Doll{\'a}r, Piotr and Zitnick, C Lawrence',
+        title='Microsoft coco: Common objects in context',
+        container='European conference on computer vision',
+        year='2014',
+        homepage='http://cocodataset.org/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
+        1:
+        dict(
+            name='left_eye',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_eye'),
+        2:
+        dict(
+            name='right_eye',
+            id=2,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_eye'),
+        3:
+        dict(
+            name='left_ear',
+            id=3,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_ear'),
+        4:
+        dict(
+            name='right_ear',
+            id=4,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_ear'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='right_shoulder',
+            id=6,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        7:
+        dict(
+            name='left_elbow',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        8:
+        dict(
+            name='right_elbow',
+            id=8,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        9:
+        dict(
+            name='left_wrist',
+            id=9,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='right_hip',
+            id=12,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        13:
+        dict(
+            name='left_knee',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        14:
+        dict(
+            name='right_knee',
+            id=14,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        15:
+        dict(
+            name='left_ankle',
+            id=15,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        16:
+        dict(
+            name='right_ankle',
+            id=16,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle')
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
+        5:
+        dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
+        6:
+        dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
+        7:
+        dict(
+            link=('left_shoulder', 'right_shoulder'),
+            id=7,
+            color=[51, 153, 255]),
+        8:
+        dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
+        9:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
+        13:
+        dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
+        14:
+        dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
+        15:
+        dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
+        16:
+        dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
+        17:
+        dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
+        18:
+        dict(
+            link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255])
+    },
+    joint_weights=[
+        1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
+        1.5
+    ],
+    sigmas=[
+        0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
+        0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089
+    ])

configs/_base_/datasets/coco_openpose.py

@@ -0,0 +1,163 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='coco_openpose',
+    paper_info=dict(
+        author='Zhe, Cao and Tomas, Simon and '
+        'Shih-En, Wei and Yaser, Sheikh',
+        title='OpenPose: Realtime Multi-Person 2D Pose '
+        'Estimation using Part Affinity Fields',
+        container='IEEE Transactions on Pattern Analysis '
+        'and Machine Intelligence',
+        year='2019',
+        homepage='https://github.com/CMU-Perceptual-Computing-Lab/openpose/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[255, 0, 0], type='upper', swap=''),
+        1:
+        dict(name='neck', id=1, color=[255, 85, 0], type='upper', swap=''),
+        2:
+        dict(
+            name='right_shoulder',
+            id=2,
+            color=[255, 170, 0],
+            type='upper',
+            swap='left_shoulder'),
+        3:
+        dict(
+            name='right_elbow',
+            id=3,
+            color=[255, 255, 0],
+            type='upper',
+            swap='left_elbow'),
+        4:
+        dict(
+            name='right_wrist',
+            id=4,
+            color=[170, 255, 0],
+            type='upper',
+            swap='left_wrist'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[85, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='left_elbow',
+            id=6,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        7:
+        dict(
+            name='left_wrist',
+            id=7,
+            color=[0, 255, 85],
+            type='upper',
+            swap='right_wrist'),
+        8:
+        dict(
+            name='right_hip',
+            id=8,
+            color=[0, 255, 170],
+            type='lower',
+            swap='left_hip'),
+        9:
+        dict(
+            name='right_knee',
+            id=9,
+            color=[0, 255, 255],
+            type='lower',
+            swap='left_knee'),
+        10:
+        dict(
+            name='right_ankle',
+            id=10,
+            color=[0, 170, 255],
+            type='lower',
+            swap='left_ankle'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 85, 255],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='left_knee',
+            id=12,
+            color=[0, 0, 255],
+            type='lower',
+            swap='right_knee'),
+        13:
+        dict(
+            name='left_ankle',
+            id=13,
+            color=[85, 0, 255],
+            type='lower',
+            swap='right_ankle'),
+        14:
+        dict(
+            name='right_eye',
+            id=14,
+            color=[170, 0, 255],
+            type='upper',
+            swap='left_eye'),
+        15:
+        dict(
+            name='left_eye',
+            id=15,
+            color=[255, 0, 255],
+            type='upper',
+            swap='right_eye'),
+        16:
+        dict(
+            name='right_ear',
+            id=16,
+            color=[255, 0, 170],
+            type='upper',
+            swap='left_ear'),
+        17:
+        dict(
+            name='left_ear',
+            id=17,
+            color=[255, 0, 85],
+            type='upper',
+            swap='right_ear'),
+    },
+    skeleton_info={
+        0: dict(link=('neck', 'right_shoulder'), id=0, color=[255, 0, 0]),
+        1: dict(link=('neck', 'left_shoulder'), id=1, color=[255, 85, 0]),
+        2: dict(
+            link=('right_shoulder', 'right_elbow'), id=2, color=[255, 170, 0]),
+        3:
+        dict(link=('right_elbow', 'right_wrist'), id=3, color=[255, 255, 0]),
+        4:
+        dict(link=('left_shoulder', 'left_elbow'), id=4, color=[170, 255, 0]),
+        5: dict(link=('left_elbow', 'left_wrist'), id=5, color=[85, 255, 0]),
+        6: dict(link=('neck', 'right_hip'), id=6, color=[0, 255, 0]),
+        7: dict(link=('right_hip', 'right_knee'), id=7, color=[0, 255, 85]),
+        8: dict(link=('right_knee', 'right_ankle'), id=8, color=[0, 255, 170]),
+        9: dict(link=('neck', 'left_hip'), id=9, color=[0, 255, 225]),
+        10: dict(link=('left_hip', 'left_knee'), id=10, color=[0, 170, 255]),
+        11: dict(link=('left_knee', 'left_ankle'), id=11, color=[0, 85, 255]),
+        12: dict(link=('neck', 'nose'), id=12, color=[0, 0, 255]),
+        13: dict(link=('nose', 'right_eye'), id=13, color=[255, 0, 170]),
+        14: dict(link=('right_eye', 'right_ear'), id=14, color=[170, 0, 255]),
+        15: dict(link=('nose', 'left_eye'), id=15, color=[255, 0, 255]),
+        16: dict(link=('left_eye', 'left_ear'), id=16, color=[255, 0, 170]),
+    },
+    joint_weights=[1.] * 18,
+    sigmas=[
+        0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
+        0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089, 0.082
+    ])

configs/_base_/datasets/coco_wholebody.py

@@ -0,0 +1,1160 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='coco_wholebody',
+    paper_info=dict(
+        author='Jin, Sheng and Xu, Lumin and Xu, Jin and '
+        'Wang, Can and Liu, Wentao and '
+        'Qian, Chen and Ouyang, Wanli and Luo, Ping',
+        title='Whole-Body Human Pose Estimation in the Wild',
+        container='Proceedings of the European '
+        'Conference on Computer Vision (ECCV)',
+        year='2020',
+        homepage='https://github.com/jin-s13/COCO-WholeBody/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
+        1:
+        dict(
+            name='left_eye',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_eye'),
+        2:
+        dict(
+            name='right_eye',
+            id=2,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_eye'),
+        3:
+        dict(
+            name='left_ear',
+            id=3,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_ear'),
+        4:
+        dict(
+            name='right_ear',
+            id=4,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_ear'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='right_shoulder',
+            id=6,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        7:
+        dict(
+            name='left_elbow',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        8:
+        dict(
+            name='right_elbow',
+            id=8,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        9:
+        dict(
+            name='left_wrist',
+            id=9,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='right_hip',
+            id=12,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        13:
+        dict(
+            name='left_knee',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        14:
+        dict(
+            name='right_knee',
+            id=14,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        15:
+        dict(
+            name='left_ankle',
+            id=15,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        16:
+        dict(
+            name='right_ankle',
+            id=16,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle'),
+        17:
+        dict(
+            name='left_big_toe',
+            id=17,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_big_toe'),
+        18:
+        dict(
+            name='left_small_toe',
+            id=18,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_small_toe'),
+        19:
+        dict(
+            name='left_heel',
+            id=19,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_heel'),
+        20:
+        dict(
+            name='right_big_toe',
+            id=20,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_big_toe'),
+        21:
+        dict(
+            name='right_small_toe',
+            id=21,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_small_toe'),
+        22:
+        dict(
+            name='right_heel',
+            id=22,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_heel'),
+        23:
+        dict(
+            name='face-0',
+            id=23,
+            color=[255, 255, 255],
+            type='',
+            swap='face-16'),
+        24:
+        dict(
+            name='face-1',
+            id=24,
+            color=[255, 255, 255],
+            type='',
+            swap='face-15'),
+        25:
+        dict(
+            name='face-2',
+            id=25,
+            color=[255, 255, 255],
+            type='',
+            swap='face-14'),
+        26:
+        dict(
+            name='face-3',
+            id=26,
+            color=[255, 255, 255],
+            type='',
+            swap='face-13'),
+        27:
+        dict(
+            name='face-4',
+            id=27,
+            color=[255, 255, 255],
+            type='',
+            swap='face-12'),
+        28:
+        dict(
+            name='face-5',
+            id=28,
+            color=[255, 255, 255],
+            type='',
+            swap='face-11'),
+        29:
+        dict(
+            name='face-6',
+            id=29,
+            color=[255, 255, 255],
+            type='',
+            swap='face-10'),
+        30:
+        dict(
+            name='face-7',
+            id=30,
+            color=[255, 255, 255],
+            type='',
+            swap='face-9'),
+        31:
+        dict(name='face-8', id=31, color=[255, 255, 255], type='', swap=''),
+        32:
+        dict(
+            name='face-9',
+            id=32,
+            color=[255, 255, 255],
+            type='',
+            swap='face-7'),
+        33:
+        dict(
+            name='face-10',
+            id=33,
+            color=[255, 255, 255],
+            type='',
+            swap='face-6'),
+        34:
+        dict(
+            name='face-11',
+            id=34,
+            color=[255, 255, 255],
+            type='',
+            swap='face-5'),
+        35:
+        dict(
+            name='face-12',
+            id=35,
+            color=[255, 255, 255],
+            type='',
+            swap='face-4'),
+        36:
+        dict(
+            name='face-13',
+            id=36,
+            color=[255, 255, 255],
+            type='',
+            swap='face-3'),
+        37:
+        dict(
+            name='face-14',
+            id=37,
+            color=[255, 255, 255],
+            type='',
+            swap='face-2'),
+        38:
+        dict(
+            name='face-15',
+            id=38,
+            color=[255, 255, 255],
+            type='',
+            swap='face-1'),
+        39:
+        dict(
+            name='face-16',
+            id=39,
+            color=[255, 255, 255],
+            type='',
+            swap='face-0'),
+        40:
+        dict(
+            name='face-17',
+            id=40,
+            color=[255, 255, 255],
+            type='',
+            swap='face-26'),
+        41:
+        dict(
+            name='face-18',
+            id=41,
+            color=[255, 255, 255],
+            type='',
+            swap='face-25'),
+        42:
+        dict(
+            name='face-19',
+            id=42,
+            color=[255, 255, 255],
+            type='',
+            swap='face-24'),
+        43:
+        dict(
+            name='face-20',
+            id=43,
+            color=[255, 255, 255],
+            type='',
+            swap='face-23'),
+        44:
+        dict(
+            name='face-21',
+            id=44,
+            color=[255, 255, 255],
+            type='',
+            swap='face-22'),
+        45:
+        dict(
+            name='face-22',
+            id=45,
+            color=[255, 255, 255],
+            type='',
+            swap='face-21'),
+        46:
+        dict(
+            name='face-23',
+            id=46,
+            color=[255, 255, 255],
+            type='',
+            swap='face-20'),
+        47:
+        dict(
+            name='face-24',
+            id=47,
+            color=[255, 255, 255],
+            type='',
+            swap='face-19'),
+        48:
+        dict(
+            name='face-25',
+            id=48,
+            color=[255, 255, 255],
+            type='',
+            swap='face-18'),
+        49:
+        dict(
+            name='face-26',
+            id=49,
+            color=[255, 255, 255],
+            type='',
+            swap='face-17'),
+        50:
+        dict(name='face-27', id=50, color=[255, 255, 255], type='', swap=''),
+        51:
+        dict(name='face-28', id=51, color=[255, 255, 255], type='', swap=''),
+        52:
+        dict(name='face-29', id=52, color=[255, 255, 255], type='', swap=''),
+        53:
+        dict(name='face-30', id=53, color=[255, 255, 255], type='', swap=''),
+        54:
+        dict(
+            name='face-31',
+            id=54,
+            color=[255, 255, 255],
+            type='',
+            swap='face-35'),
+        55:
+        dict(
+            name='face-32',
+            id=55,
+            color=[255, 255, 255],
+            type='',
+            swap='face-34'),
+        56:
+        dict(name='face-33', id=56, color=[255, 255, 255], type='', swap=''),
+        57:
+        dict(
+            name='face-34',
+            id=57,
+            color=[255, 255, 255],
+            type='',
+            swap='face-32'),
+        58:
+        dict(
+            name='face-35',
+            id=58,
+            color=[255, 255, 255],
+            type='',
+            swap='face-31'),
+        59:
+        dict(
+            name='face-36',
+            id=59,
+            color=[255, 255, 255],
+            type='',
+            swap='face-45'),
+        60:
+        dict(
+            name='face-37',
+            id=60,
+            color=[255, 255, 255],
+            type='',
+            swap='face-44'),
+        61:
+        dict(
+            name='face-38',
+            id=61,
+            color=[255, 255, 255],
+            type='',
+            swap='face-43'),
+        62:
+        dict(
+            name='face-39',
+            id=62,
+            color=[255, 255, 255],
+            type='',
+            swap='face-42'),
+        63:
+        dict(
+            name='face-40',
+            id=63,
+            color=[255, 255, 255],
+            type='',
+            swap='face-47'),
+        64:
+        dict(
+            name='face-41',
+            id=64,
+            color=[255, 255, 255],
+            type='',
+            swap='face-46'),
+        65:
+        dict(
+            name='face-42',
+            id=65,
+            color=[255, 255, 255],
+            type='',
+            swap='face-39'),
+        66:
+        dict(
+            name='face-43',
+            id=66,
+            color=[255, 255, 255],
+            type='',
+            swap='face-38'),
+        67:
+        dict(
+            name='face-44',
+            id=67,
+            color=[255, 255, 255],
+            type='',
+            swap='face-37'),
+        68:
+        dict(
+            name='face-45',
+            id=68,
+            color=[255, 255, 255],
+            type='',
+            swap='face-36'),
+        69:
+        dict(
+            name='face-46',
+            id=69,
+            color=[255, 255, 255],
+            type='',
+            swap='face-41'),
+        70:
+        dict(
+            name='face-47',
+            id=70,
+            color=[255, 255, 255],
+            type='',
+            swap='face-40'),
+        71:
+        dict(
+            name='face-48',
+            id=71,
+            color=[255, 255, 255],
+            type='',
+            swap='face-54'),
+        72:
+        dict(
+            name='face-49',
+            id=72,
+            color=[255, 255, 255],
+            type='',
+            swap='face-53'),
+        73:
+        dict(
+            name='face-50',
+            id=73,
+            color=[255, 255, 255],
+            type='',
+            swap='face-52'),
+        74:
+        dict(name='face-51', id=74, color=[255, 255, 255], type='', swap=''),
+        75:
+        dict(
+            name='face-52',
+            id=75,
+            color=[255, 255, 255],
+            type='',
+            swap='face-50'),
+        76:
+        dict(
+            name='face-53',
+            id=76,
+            color=[255, 255, 255],
+            type='',
+            swap='face-49'),
+        77:
+        dict(
+            name='face-54',
+            id=77,
+            color=[255, 255, 255],
+            type='',
+            swap='face-48'),
+        78:
+        dict(
+            name='face-55',
+            id=78,
+            color=[255, 255, 255],
+            type='',
+            swap='face-59'),
+        79:
+        dict(
+            name='face-56',
+            id=79,
+            color=[255, 255, 255],
+            type='',
+            swap='face-58'),
+        80:
+        dict(name='face-57', id=80, color=[255, 255, 255], type='', swap=''),
+        81:
+        dict(
+            name='face-58',
+            id=81,
+            color=[255, 255, 255],
+            type='',
+            swap='face-56'),
+        82:
+        dict(
+            name='face-59',
+            id=82,
+            color=[255, 255, 255],
+            type='',
+            swap='face-55'),
+        83:
+        dict(
+            name='face-60',
+            id=83,
+            color=[255, 255, 255],
+            type='',
+            swap='face-64'),
+        84:
+        dict(
+            name='face-61',
+            id=84,
+            color=[255, 255, 255],
+            type='',
+            swap='face-63'),
+        85:
+        dict(name='face-62', id=85, color=[255, 255, 255], type='', swap=''),
+        86:
+        dict(
+            name='face-63',
+            id=86,
+            color=[255, 255, 255],
+            type='',
+            swap='face-61'),
+        87:
+        dict(
+            name='face-64',
+            id=87,
+            color=[255, 255, 255],
+            type='',
+            swap='face-60'),
+        88:
+        dict(
+            name='face-65',
+            id=88,
+            color=[255, 255, 255],
+            type='',
+            swap='face-67'),
+        89:
+        dict(name='face-66', id=89, color=[255, 255, 255], type='', swap=''),
+        90:
+        dict(
+            name='face-67',
+            id=90,
+            color=[255, 255, 255],
+            type='',
+            swap='face-65'),
+        91:
+        dict(
+            name='left_hand_root',
+            id=91,
+            color=[255, 255, 255],
+            type='',
+            swap='right_hand_root'),
+        92:
+        dict(
+            name='left_thumb1',
+            id=92,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb1'),
+        93:
+        dict(
+            name='left_thumb2',
+            id=93,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb2'),
+        94:
+        dict(
+            name='left_thumb3',
+            id=94,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb3'),
+        95:
+        dict(
+            name='left_thumb4',
+            id=95,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb4'),
+        96:
+        dict(
+            name='left_forefinger1',
+            id=96,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger1'),
+        97:
+        dict(
+            name='left_forefinger2',
+            id=97,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger2'),
+        98:
+        dict(
+            name='left_forefinger3',
+            id=98,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger3'),
+        99:
+        dict(
+            name='left_forefinger4',
+            id=99,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger4'),
+        100:
+        dict(
+            name='left_middle_finger1',
+            id=100,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger1'),
+        101:
+        dict(
+            name='left_middle_finger2',
+            id=101,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger2'),
+        102:
+        dict(
+            name='left_middle_finger3',
+            id=102,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger3'),
+        103:
+        dict(
+            name='left_middle_finger4',
+            id=103,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger4'),
+        104:
+        dict(
+            name='left_ring_finger1',
+            id=104,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger1'),
+        105:
+        dict(
+            name='left_ring_finger2',
+            id=105,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger2'),
+        106:
+        dict(
+            name='left_ring_finger3',
+            id=106,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger3'),
+        107:
+        dict(
+            name='left_ring_finger4',
+            id=107,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger4'),
+        108:
+        dict(
+            name='left_pinky_finger1',
+            id=108,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger1'),
+        109:
+        dict(
+            name='left_pinky_finger2',
+            id=109,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger2'),
+        110:
+        dict(
+            name='left_pinky_finger3',
+            id=110,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger3'),
+        111:
+        dict(
+            name='left_pinky_finger4',
+            id=111,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger4'),
+        112:
+        dict(
+            name='right_hand_root',
+            id=112,
+            color=[255, 255, 255],
+            type='',
+            swap='left_hand_root'),
+        113:
+        dict(
+            name='right_thumb1',
+            id=113,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb1'),
+        114:
+        dict(
+            name='right_thumb2',
+            id=114,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb2'),
+        115:
+        dict(
+            name='right_thumb3',
+            id=115,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb3'),
+        116:
+        dict(
+            name='right_thumb4',
+            id=116,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb4'),
+        117:
+        dict(
+            name='right_forefinger1',
+            id=117,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger1'),
+        118:
+        dict(
+            name='right_forefinger2',
+            id=118,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger2'),
+        119:
+        dict(
+            name='right_forefinger3',
+            id=119,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger3'),
+        120:
+        dict(
+            name='right_forefinger4',
+            id=120,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger4'),
+        121:
+        dict(
+            name='right_middle_finger1',
+            id=121,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger1'),
+        122:
+        dict(
+            name='right_middle_finger2',
+            id=122,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger2'),
+        123:
+        dict(
+            name='right_middle_finger3',
+            id=123,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger3'),
+        124:
+        dict(
+            name='right_middle_finger4',
+            id=124,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger4'),
+        125:
+        dict(
+            name='right_ring_finger1',
+            id=125,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger1'),
+        126:
+        dict(
+            name='right_ring_finger2',
+            id=126,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger2'),
+        127:
+        dict(
+            name='right_ring_finger3',
+            id=127,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger3'),
+        128:
+        dict(
+            name='right_ring_finger4',
+            id=128,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger4'),
+        129:
+        dict(
+            name='right_pinky_finger1',
+            id=129,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger1'),
+        130:
+        dict(
+            name='right_pinky_finger2',
+            id=130,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger2'),
+        131:
+        dict(
+            name='right_pinky_finger3',
+            id=131,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger3'),
+        132:
+        dict(
+            name='right_pinky_finger4',
+            id=132,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger4')
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
+        5:
+        dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
+        6:
+        dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
+        7:
+        dict(
+            link=('left_shoulder', 'right_shoulder'),
+            id=7,
+            color=[51, 153, 255]),
+        8:
+        dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
+        9:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
+        13:
+        dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
+        14:
+        dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
+        15:
+        dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
+        16:
+        dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
+        17:
+        dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
+        18:
+        dict(
+            link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255]),
+        19:
+        dict(link=('left_ankle', 'left_big_toe'), id=19, color=[0, 255, 0]),
+        20:
+        dict(link=('left_ankle', 'left_small_toe'), id=20, color=[0, 255, 0]),
+        21:
+        dict(link=('left_ankle', 'left_heel'), id=21, color=[0, 255, 0]),
+        22:
+        dict(
+            link=('right_ankle', 'right_big_toe'), id=22, color=[255, 128, 0]),
+        23:
+        dict(
+            link=('right_ankle', 'right_small_toe'),
+            id=23,
+            color=[255, 128, 0]),
+        24:
+        dict(link=('right_ankle', 'right_heel'), id=24, color=[255, 128, 0]),
+        25:
+        dict(
+            link=('left_hand_root', 'left_thumb1'), id=25, color=[255, 128,
+                                                                  0]),
+        26:
+        dict(link=('left_thumb1', 'left_thumb2'), id=26, color=[255, 128, 0]),
+        27:
+        dict(link=('left_thumb2', 'left_thumb3'), id=27, color=[255, 128, 0]),
+        28:
+        dict(link=('left_thumb3', 'left_thumb4'), id=28, color=[255, 128, 0]),
+        29:
+        dict(
+            link=('left_hand_root', 'left_forefinger1'),
+            id=29,
+            color=[255, 153, 255]),
+        30:
+        dict(
+            link=('left_forefinger1', 'left_forefinger2'),
+            id=30,
+            color=[255, 153, 255]),
+        31:
+        dict(
+            link=('left_forefinger2', 'left_forefinger3'),
+            id=31,
+            color=[255, 153, 255]),
+        32:
+        dict(
+            link=('left_forefinger3', 'left_forefinger4'),
+            id=32,
+            color=[255, 153, 255]),
+        33:
+        dict(
+            link=('left_hand_root', 'left_middle_finger1'),
+            id=33,
+            color=[102, 178, 255]),
+        34:
+        dict(
+            link=('left_middle_finger1', 'left_middle_finger2'),
+            id=34,
+            color=[102, 178, 255]),
+        35:
+        dict(
+            link=('left_middle_finger2', 'left_middle_finger3'),
+            id=35,
+            color=[102, 178, 255]),
+        36:
+        dict(
+            link=('left_middle_finger3', 'left_middle_finger4'),
+            id=36,
+            color=[102, 178, 255]),
+        37:
+        dict(
+            link=('left_hand_root', 'left_ring_finger1'),
+            id=37,
+            color=[255, 51, 51]),
+        38:
+        dict(
+            link=('left_ring_finger1', 'left_ring_finger2'),
+            id=38,
+            color=[255, 51, 51]),
+        39:
+        dict(
+            link=('left_ring_finger2', 'left_ring_finger3'),
+            id=39,
+            color=[255, 51, 51]),
+        40:
+        dict(
+            link=('left_ring_finger3', 'left_ring_finger4'),
+            id=40,
+            color=[255, 51, 51]),
+        41:
+        dict(
+            link=('left_hand_root', 'left_pinky_finger1'),
+            id=41,
+            color=[0, 255, 0]),
+        42:
+        dict(
+            link=('left_pinky_finger1', 'left_pinky_finger2'),
+            id=42,
+            color=[0, 255, 0]),
+        43:
+        dict(
+            link=('left_pinky_finger2', 'left_pinky_finger3'),
+            id=43,
+            color=[0, 255, 0]),
+        44:
+        dict(
+            link=('left_pinky_finger3', 'left_pinky_finger4'),
+            id=44,
+            color=[0, 255, 0]),
+        45:
+        dict(
+            link=('right_hand_root', 'right_thumb1'),
+            id=45,
+            color=[255, 128, 0]),
+        46:
+        dict(
+            link=('right_thumb1', 'right_thumb2'), id=46, color=[255, 128, 0]),
+        47:
+        dict(
+            link=('right_thumb2', 'right_thumb3'), id=47, color=[255, 128, 0]),
+        48:
+        dict(
+            link=('right_thumb3', 'right_thumb4'), id=48, color=[255, 128, 0]),
+        49:
+        dict(
+            link=('right_hand_root', 'right_forefinger1'),
+            id=49,
+            color=[255, 153, 255]),
+        50:
+        dict(
+            link=('right_forefinger1', 'right_forefinger2'),
+            id=50,
+            color=[255, 153, 255]),
+        51:
+        dict(
+            link=('right_forefinger2', 'right_forefinger3'),
+            id=51,
+            color=[255, 153, 255]),
+        52:
+        dict(
+            link=('right_forefinger3', 'right_forefinger4'),
+            id=52,
+            color=[255, 153, 255]),
+        53:
+        dict(
+            link=('right_hand_root', 'right_middle_finger1'),
+            id=53,
+            color=[102, 178, 255]),
+        54:
+        dict(
+            link=('right_middle_finger1', 'right_middle_finger2'),
+            id=54,
+            color=[102, 178, 255]),
+        55:
+        dict(
+            link=('right_middle_finger2', 'right_middle_finger3'),
+            id=55,
+            color=[102, 178, 255]),
+        56:
+        dict(
+            link=('right_middle_finger3', 'right_middle_finger4'),
+            id=56,
+            color=[102, 178, 255]),
+        57:
+        dict(
+            link=('right_hand_root', 'right_ring_finger1'),
+            id=57,
+            color=[255, 51, 51]),
+        58:
+        dict(
+            link=('right_ring_finger1', 'right_ring_finger2'),
+            id=58,
+            color=[255, 51, 51]),
+        59:
+        dict(
+            link=('right_ring_finger2', 'right_ring_finger3'),
+            id=59,
+            color=[255, 51, 51]),
+        60:
+        dict(
+            link=('right_ring_finger3', 'right_ring_finger4'),
+            id=60,
+            color=[255, 51, 51]),
+        61:
+        dict(
+            link=('right_hand_root', 'right_pinky_finger1'),
+            id=61,
+            color=[0, 255, 0]),
+        62:
+        dict(
+            link=('right_pinky_finger1', 'right_pinky_finger2'),
+            id=62,
+            color=[0, 255, 0]),
+        63:
+        dict(
+            link=('right_pinky_finger2', 'right_pinky_finger3'),
+            id=63,
+            color=[0, 255, 0]),
+        64:
+        dict(
+            link=('right_pinky_finger3', 'right_pinky_finger4'),
+            id=64,
+            color=[0, 255, 0])
+    },
+    joint_weights=[1.] * 133,
+    # 'https://github.com/jin-s13/COCO-WholeBody/blob/master/'
+    # 'evaluation/myeval_wholebody.py#L175'
+    sigmas=[
+        0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
+        0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089, 0.068, 0.066, 0.066,
+        0.092, 0.094, 0.094, 0.042, 0.043, 0.044, 0.043, 0.040, 0.035, 0.031,
+        0.025, 0.020, 0.023, 0.029, 0.032, 0.037, 0.038, 0.043, 0.041, 0.045,
+        0.013, 0.012, 0.011, 0.011, 0.012, 0.012, 0.011, 0.011, 0.013, 0.015,
+        0.009, 0.007, 0.007, 0.007, 0.012, 0.009, 0.008, 0.016, 0.010, 0.017,
+        0.011, 0.009, 0.011, 0.009, 0.007, 0.013, 0.008, 0.011, 0.012, 0.010,
+        0.034, 0.008, 0.008, 0.009, 0.008, 0.008, 0.007, 0.010, 0.008, 0.009,
+        0.009, 0.009, 0.007, 0.007, 0.008, 0.011, 0.008, 0.008, 0.008, 0.01,
+        0.008, 0.029, 0.022, 0.035, 0.037, 0.047, 0.026, 0.025, 0.024, 0.035,
+        0.018, 0.024, 0.022, 0.026, 0.017, 0.021, 0.021, 0.032, 0.02, 0.019,
+        0.022, 0.031, 0.029, 0.022, 0.035, 0.037, 0.047, 0.026, 0.025, 0.024,
+        0.035, 0.018, 0.024, 0.022, 0.026, 0.017, 0.021, 0.021, 0.032, 0.02,
+        0.019, 0.022, 0.031
+    ])

configs/_base_/datasets/coco_wholebody2goliath.py

@@ -0,0 +1,11 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+from configs._base_.datasets.goliath import dataset_info as goliath_info
+
+goliath_info = goliath_info.build()
+dataset_info = goliath_info.copy()
+dataset_info['dataset_name'] = 'coco_wholebody2goliath'

configs/_base_/datasets/coco_wholebody2goliath3d.py

@@ -0,0 +1,11 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+from configs._base_.datasets.goliath3d import dataset_info as goliath_info
+
+goliath_info = goliath_info.build()
+dataset_info = goliath_info.copy()
+dataset_info['dataset_name'] = 'coco_wholebody2goliath3d'

configs/_base_/datasets/coco_wholebody_face.py

@@ -0,0 +1,160 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='coco_wholebody_face',
+    paper_info=dict(
+        author='Jin, Sheng and Xu, Lumin and Xu, Jin and '
+        'Wang, Can and Liu, Wentao and '
+        'Qian, Chen and Ouyang, Wanli and Luo, Ping',
+        title='Whole-Body Human Pose Estimation in the Wild',
+        container='Proceedings of the European '
+        'Conference on Computer Vision (ECCV)',
+        year='2020',
+        homepage='https://github.com/jin-s13/COCO-WholeBody/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='face-0', id=0, color=[255, 0, 0], type='', swap='face-16'),
+        1:
+        dict(name='face-1', id=1, color=[255, 0, 0], type='', swap='face-15'),
+        2:
+        dict(name='face-2', id=2, color=[255, 0, 0], type='', swap='face-14'),
+        3:
+        dict(name='face-3', id=3, color=[255, 0, 0], type='', swap='face-13'),
+        4:
+        dict(name='face-4', id=4, color=[255, 0, 0], type='', swap='face-12'),
+        5:
+        dict(name='face-5', id=5, color=[255, 0, 0], type='', swap='face-11'),
+        6:
+        dict(name='face-6', id=6, color=[255, 0, 0], type='', swap='face-10'),
+        7:
+        dict(name='face-7', id=7, color=[255, 0, 0], type='', swap='face-9'),
+        8: dict(name='face-8', id=8, color=[255, 0, 0], type='', swap=''),
+        9:
+        dict(name='face-9', id=9, color=[255, 0, 0], type='', swap='face-7'),
+        10:
+        dict(name='face-10', id=10, color=[255, 0, 0], type='', swap='face-6'),
+        11:
+        dict(name='face-11', id=11, color=[255, 0, 0], type='', swap='face-5'),
+        12:
+        dict(name='face-12', id=12, color=[255, 0, 0], type='', swap='face-4'),
+        13:
+        dict(name='face-13', id=13, color=[255, 0, 0], type='', swap='face-3'),
+        14:
+        dict(name='face-14', id=14, color=[255, 0, 0], type='', swap='face-2'),
+        15:
+        dict(name='face-15', id=15, color=[255, 0, 0], type='', swap='face-1'),
+        16:
+        dict(name='face-16', id=16, color=[255, 0, 0], type='', swap='face-0'),
+        17: dict(
+            name='face-17', id=17, color=[255, 0, 0], type='', swap='face-26'),
+        18: dict(
+            name='face-18', id=18, color=[255, 0, 0], type='', swap='face-25'),
+        19: dict(
+            name='face-19', id=19, color=[255, 0, 0], type='', swap='face-24'),
+        20: dict(
+            name='face-20', id=20, color=[255, 0, 0], type='', swap='face-23'),
+        21: dict(
+            name='face-21', id=21, color=[255, 0, 0], type='', swap='face-22'),
+        22: dict(
+            name='face-22', id=22, color=[255, 0, 0], type='', swap='face-21'),
+        23: dict(
+            name='face-23', id=23, color=[255, 0, 0], type='', swap='face-20'),
+        24: dict(
+            name='face-24', id=24, color=[255, 0, 0], type='', swap='face-19'),
+        25: dict(
+            name='face-25', id=25, color=[255, 0, 0], type='', swap='face-18'),
+        26: dict(
+            name='face-26', id=26, color=[255, 0, 0], type='', swap='face-17'),
+        27: dict(name='face-27', id=27, color=[255, 0, 0], type='', swap=''),
+        28: dict(name='face-28', id=28, color=[255, 0, 0], type='', swap=''),
+        29: dict(name='face-29', id=29, color=[255, 0, 0], type='', swap=''),
+        30: dict(name='face-30', id=30, color=[255, 0, 0], type='', swap=''),
+        31: dict(
+            name='face-31', id=31, color=[255, 0, 0], type='', swap='face-35'),
+        32: dict(
+            name='face-32', id=32, color=[255, 0, 0], type='', swap='face-34'),
+        33: dict(name='face-33', id=33, color=[255, 0, 0], type='', swap=''),
+        34: dict(
+            name='face-34', id=34, color=[255, 0, 0], type='', swap='face-32'),
+        35: dict(
+            name='face-35', id=35, color=[255, 0, 0], type='', swap='face-31'),
+        36: dict(
+            name='face-36', id=36, color=[255, 0, 0], type='', swap='face-45'),
+        37: dict(
+            name='face-37', id=37, color=[255, 0, 0], type='', swap='face-44'),
+        38: dict(
+            name='face-38', id=38, color=[255, 0, 0], type='', swap='face-43'),
+        39: dict(
+            name='face-39', id=39, color=[255, 0, 0], type='', swap='face-42'),
+        40: dict(
+            name='face-40', id=40, color=[255, 0, 0], type='', swap='face-47'),
+        41: dict(
+            name='face-41', id=41, color=[255, 0, 0], type='', swap='face-46'),
+        42: dict(
+            name='face-42', id=42, color=[255, 0, 0], type='', swap='face-39'),
+        43: dict(
+            name='face-43', id=43, color=[255, 0, 0], type='', swap='face-38'),
+        44: dict(
+            name='face-44', id=44, color=[255, 0, 0], type='', swap='face-37'),
+        45: dict(
+            name='face-45', id=45, color=[255, 0, 0], type='', swap='face-36'),
+        46: dict(
+            name='face-46', id=46, color=[255, 0, 0], type='', swap='face-41'),
+        47: dict(
+            name='face-47', id=47, color=[255, 0, 0], type='', swap='face-40'),
+        48: dict(
+            name='face-48', id=48, color=[255, 0, 0], type='', swap='face-54'),
+        49: dict(
+            name='face-49', id=49, color=[255, 0, 0], type='', swap='face-53'),
+        50: dict(
+            name='face-50', id=50, color=[255, 0, 0], type='', swap='face-52'),
+        51: dict(name='face-51', id=52, color=[255, 0, 0], type='', swap=''),
+        52: dict(
+            name='face-52', id=52, color=[255, 0, 0], type='', swap='face-50'),
+        53: dict(
+            name='face-53', id=53, color=[255, 0, 0], type='', swap='face-49'),
+        54: dict(
+            name='face-54', id=54, color=[255, 0, 0], type='', swap='face-48'),
+        55: dict(
+            name='face-55', id=55, color=[255, 0, 0], type='', swap='face-59'),
+        56: dict(
+            name='face-56', id=56, color=[255, 0, 0], type='', swap='face-58'),
+        57: dict(name='face-57', id=57, color=[255, 0, 0], type='', swap=''),
+        58: dict(
+            name='face-58', id=58, color=[255, 0, 0], type='', swap='face-56'),
+        59: dict(
+            name='face-59', id=59, color=[255, 0, 0], type='', swap='face-55'),
+        60: dict(
+            name='face-60', id=60, color=[255, 0, 0], type='', swap='face-64'),
+        61: dict(
+            name='face-61', id=61, color=[255, 0, 0], type='', swap='face-63'),
+        62: dict(name='face-62', id=62, color=[255, 0, 0], type='', swap=''),
+        63: dict(
+            name='face-63', id=63, color=[255, 0, 0], type='', swap='face-61'),
+        64: dict(
+            name='face-64', id=64, color=[255, 0, 0], type='', swap='face-60'),
+        65: dict(
+            name='face-65', id=65, color=[255, 0, 0], type='', swap='face-67'),
+        66: dict(name='face-66', id=66, color=[255, 0, 0], type='', swap=''),
+        67: dict(
+            name='face-67', id=67, color=[255, 0, 0], type='', swap='face-65')
+    },
+    skeleton_info={},
+    joint_weights=[1.] * 68,
+
+    # 'https://github.com/jin-s13/COCO-WholeBody/blob/master/'
+    # 'evaluation/myeval_wholebody.py#L177'
+    sigmas=[
+        0.042, 0.043, 0.044, 0.043, 0.040, 0.035, 0.031, 0.025, 0.020, 0.023,
+        0.029, 0.032, 0.037, 0.038, 0.043, 0.041, 0.045, 0.013, 0.012, 0.011,
+        0.011, 0.012, 0.012, 0.011, 0.011, 0.013, 0.015, 0.009, 0.007, 0.007,
+        0.007, 0.012, 0.009, 0.008, 0.016, 0.010, 0.017, 0.011, 0.009, 0.011,
+        0.009, 0.007, 0.013, 0.008, 0.011, 0.012, 0.010, 0.034, 0.008, 0.008,
+        0.009, 0.008, 0.008, 0.007, 0.010, 0.008, 0.009, 0.009, 0.009, 0.007,
+        0.007, 0.008, 0.011, 0.008, 0.008, 0.008, 0.01, 0.008
+    ])

configs/_base_/datasets/coco_wholebody_hand.py

@@ -0,0 +1,153 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='coco_wholebody_hand',
+    paper_info=dict(
+        author='Jin, Sheng and Xu, Lumin and Xu, Jin and '
+        'Wang, Can and Liu, Wentao and '
+        'Qian, Chen and Ouyang, Wanli and Luo, Ping',
+        title='Whole-Body Human Pose Estimation in the Wild',
+        container='Proceedings of the European '
+        'Conference on Computer Vision (ECCV)',
+        year='2020',
+        homepage='https://github.com/jin-s13/COCO-WholeBody/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='wrist', id=0, color=[255, 255, 255], type='', swap=''),
+        1:
+        dict(name='thumb1', id=1, color=[255, 128, 0], type='', swap=''),
+        2:
+        dict(name='thumb2', id=2, color=[255, 128, 0], type='', swap=''),
+        3:
+        dict(name='thumb3', id=3, color=[255, 128, 0], type='', swap=''),
+        4:
+        dict(name='thumb4', id=4, color=[255, 128, 0], type='', swap=''),
+        5:
+        dict(
+            name='forefinger1', id=5, color=[255, 153, 255], type='', swap=''),
+        6:
+        dict(
+            name='forefinger2', id=6, color=[255, 153, 255], type='', swap=''),
+        7:
+        dict(
+            name='forefinger3', id=7, color=[255, 153, 255], type='', swap=''),
+        8:
+        dict(
+            name='forefinger4', id=8, color=[255, 153, 255], type='', swap=''),
+        9:
+        dict(
+            name='middle_finger1',
+            id=9,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        10:
+        dict(
+            name='middle_finger2',
+            id=10,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        11:
+        dict(
+            name='middle_finger3',
+            id=11,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        12:
+        dict(
+            name='middle_finger4',
+            id=12,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        13:
+        dict(
+            name='ring_finger1', id=13, color=[255, 51, 51], type='', swap=''),
+        14:
+        dict(
+            name='ring_finger2', id=14, color=[255, 51, 51], type='', swap=''),
+        15:
+        dict(
+            name='ring_finger3', id=15, color=[255, 51, 51], type='', swap=''),
+        16:
+        dict(
+            name='ring_finger4', id=16, color=[255, 51, 51], type='', swap=''),
+        17:
+        dict(name='pinky_finger1', id=17, color=[0, 255, 0], type='', swap=''),
+        18:
+        dict(name='pinky_finger2', id=18, color=[0, 255, 0], type='', swap=''),
+        19:
+        dict(name='pinky_finger3', id=19, color=[0, 255, 0], type='', swap=''),
+        20:
+        dict(name='pinky_finger4', id=20, color=[0, 255, 0], type='', swap='')
+    },
+    skeleton_info={
+        0:
+        dict(link=('wrist', 'thumb1'), id=0, color=[255, 128, 0]),
+        1:
+        dict(link=('thumb1', 'thumb2'), id=1, color=[255, 128, 0]),
+        2:
+        dict(link=('thumb2', 'thumb3'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('thumb3', 'thumb4'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('wrist', 'forefinger1'), id=4, color=[255, 153, 255]),
+        5:
+        dict(link=('forefinger1', 'forefinger2'), id=5, color=[255, 153, 255]),
+        6:
+        dict(link=('forefinger2', 'forefinger3'), id=6, color=[255, 153, 255]),
+        7:
+        dict(link=('forefinger3', 'forefinger4'), id=7, color=[255, 153, 255]),
+        8:
+        dict(link=('wrist', 'middle_finger1'), id=8, color=[102, 178, 255]),
+        9:
+        dict(
+            link=('middle_finger1', 'middle_finger2'),
+            id=9,
+            color=[102, 178, 255]),
+        10:
+        dict(
+            link=('middle_finger2', 'middle_finger3'),
+            id=10,
+            color=[102, 178, 255]),
+        11:
+        dict(
+            link=('middle_finger3', 'middle_finger4'),
+            id=11,
+            color=[102, 178, 255]),
+        12:
+        dict(link=('wrist', 'ring_finger1'), id=12, color=[255, 51, 51]),
+        13:
+        dict(
+            link=('ring_finger1', 'ring_finger2'), id=13, color=[255, 51, 51]),
+        14:
+        dict(
+            link=('ring_finger2', 'ring_finger3'), id=14, color=[255, 51, 51]),
+        15:
+        dict(
+            link=('ring_finger3', 'ring_finger4'), id=15, color=[255, 51, 51]),
+        16:
+        dict(link=('wrist', 'pinky_finger1'), id=16, color=[0, 255, 0]),
+        17:
+        dict(
+            link=('pinky_finger1', 'pinky_finger2'), id=17, color=[0, 255, 0]),
+        18:
+        dict(
+            link=('pinky_finger2', 'pinky_finger3'), id=18, color=[0, 255, 0]),
+        19:
+        dict(
+            link=('pinky_finger3', 'pinky_finger4'), id=19, color=[0, 255, 0])
+    },
+    joint_weights=[1.] * 21,
+    sigmas=[
+        0.029, 0.022, 0.035, 0.037, 0.047, 0.026, 0.025, 0.024, 0.035, 0.018,
+        0.024, 0.022, 0.026, 0.017, 0.021, 0.021, 0.032, 0.02, 0.019, 0.022,
+        0.031
+    ])

configs/_base_/datasets/coco_wholebody_openpose.py

@@ -0,0 +1,1034 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='coco_wholebody_openpose',
+    paper_info=dict(
+        author='Jin, Sheng and Xu, Lumin and Xu, Jin and '
+        'Wang, Can and Liu, Wentao and '
+        'Qian, Chen and Ouyang, Wanli and Luo, Ping',
+        title='Whole-Body Human Pose Estimation in the Wild',
+        container='Proceedings of the European '
+        'Conference on Computer Vision (ECCV)',
+        year='2020',
+        homepage='https://github.com/jin-s13/COCO-WholeBody/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[255, 0, 0], type='upper', swap=''),
+        1:
+        dict(name='neck', id=1, color=[255, 85, 0], type='upper', swap=''),
+        2:
+        dict(
+            name='right_shoulder',
+            id=2,
+            color=[255, 170, 0],
+            type='upper',
+            swap='left_shoulder'),
+        3:
+        dict(
+            name='right_elbow',
+            id=3,
+            color=[255, 255, 0],
+            type='upper',
+            swap='left_elbow'),
+        4:
+        dict(
+            name='right_wrist',
+            id=4,
+            color=[170, 255, 0],
+            type='upper',
+            swap='left_wrist'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[85, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='left_elbow',
+            id=6,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        7:
+        dict(
+            name='left_wrist',
+            id=7,
+            color=[0, 255, 85],
+            type='upper',
+            swap='right_wrist'),
+        8:
+        dict(
+            name='right_hip',
+            id=8,
+            color=[0, 255, 170],
+            type='lower',
+            swap='left_hip'),
+        9:
+        dict(
+            name='right_knee',
+            id=9,
+            color=[0, 255, 255],
+            type='lower',
+            swap='left_knee'),
+        10:
+        dict(
+            name='right_ankle',
+            id=10,
+            color=[0, 170, 255],
+            type='lower',
+            swap='left_ankle'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 85, 255],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='left_knee',
+            id=12,
+            color=[0, 0, 255],
+            type='lower',
+            swap='right_knee'),
+        13:
+        dict(
+            name='left_ankle',
+            id=13,
+            color=[85, 0, 255],
+            type='lower',
+            swap='right_ankle'),
+        14:
+        dict(
+            name='right_eye',
+            id=14,
+            color=[170, 0, 255],
+            type='upper',
+            swap='left_eye'),
+        15:
+        dict(
+            name='left_eye',
+            id=15,
+            color=[255, 0, 255],
+            type='upper',
+            swap='right_eye'),
+        16:
+        dict(
+            name='right_ear',
+            id=16,
+            color=[255, 0, 170],
+            type='upper',
+            swap='left_ear'),
+        17:
+        dict(
+            name='left_ear',
+            id=17,
+            color=[255, 0, 85],
+            type='upper',
+            swap='right_ear'),
+        18:
+            dict(
+                name='left_big_toe',
+                id=18,
+                color=[255, 128, 0],
+                type='lower',
+                swap='right_big_toe'),
+        19:
+            dict(
+                name='left_small_toe',
+                id=19,
+                color=[255, 128, 0],
+                type='lower',
+                swap='right_small_toe'),
+        20:
+            dict(
+                name='left_heel',
+                id=20,
+                color=[255, 128, 0],
+                type='lower',
+                swap='right_heel'),
+        21:
+            dict(
+                name='right_big_toe',
+                id=21,
+                color=[255, 128, 0],
+                type='lower',
+                swap='left_big_toe'),
+        22:
+            dict(
+                name='right_small_toe',
+                id=22,
+                color=[255, 128, 0],
+                type='lower',
+                swap='left_small_toe'),
+        23:
+            dict(
+                name='right_heel',
+                id=23,
+                color=[255, 128, 0],
+                type='lower',
+                swap='left_heel'),
+        24:
+            dict(
+                name='face-0',
+                id=24,
+                color=[255, 255, 255],
+                type='',
+                swap='face-16'),
+        25:
+            dict(
+                name='face-1',
+                id=25,
+                color=[255, 255, 255],
+                type='',
+                swap='face-15'),
+        26:
+            dict(
+                name='face-2',
+                id=26,
+                color=[255, 255, 255],
+                type='',
+                swap='face-14'),
+        27:
+            dict(
+                name='face-3',
+                id=27,
+                color=[255, 255, 255],
+                type='',
+                swap='face-13'),
+        28:
+            dict(
+                name='face-4',
+                id=28,
+                color=[255, 255, 255],
+                type='',
+                swap='face-12'),
+        29:
+            dict(
+                name='face-5',
+                id=29,
+                color=[255, 255, 255],
+                type='',
+                swap='face-11'),
+        30:
+            dict(
+                name='face-6',
+                id=30,
+                color=[255, 255, 255],
+                type='',
+                swap='face-10'),
+        31:
+            dict(
+                name='face-7',
+                id=31,
+                color=[255, 255, 255],
+                type='',
+                swap='face-9'),
+        32:
+            dict(name='face-8', id=32, color=[255, 255, 255], type='', swap=''),
+        33:
+            dict(
+                name='face-9',
+                id=33,
+                color=[255, 255, 255],
+                type='',
+                swap='face-7'),
+        34:
+            dict(
+                name='face-10',
+                id=34,
+                color=[255, 255, 255],
+                type='',
+                swap='face-6'),
+        35:
+            dict(
+                name='face-11',
+                id=35,
+                color=[255, 255, 255],
+                type='',
+                swap='face-5'),
+        36:
+            dict(
+                name='face-12',
+                id=36,
+                color=[255, 255, 255],
+                type='',
+                swap='face-4'),
+        37:
+            dict(
+                name='face-13',
+                id=37,
+                color=[255, 255, 255],
+                type='',
+                swap='face-3'),
+        38:
+            dict(
+                name='face-14',
+                id=38,
+                color=[255, 255, 255],
+                type='',
+                swap='face-2'),
+        39:
+            dict(
+                name='face-15',
+                id=39,
+                color=[255, 255, 255],
+                type='',
+                swap='face-1'),
+        40:
+            dict(
+                name='face-16',
+                id=40,
+                color=[255, 255, 255],
+                type='',
+                swap='face-0'),
+        41:
+            dict(
+                name='face-17',
+                id=41,
+                color=[255, 255, 255],
+                type='',
+                swap='face-26'),
+        42:
+            dict(
+                name='face-18',
+                id=42,
+                color=[255, 255, 255],
+                type='',
+                swap='face-25'),
+        43:
+            dict(
+                name='face-19',
+                id=43,
+                color=[255, 255, 255],
+                type='',
+                swap='face-24'),
+        44:
+            dict(
+                name='face-20',
+                id=44,
+                color=[255, 255, 255],
+                type='',
+                swap='face-23'),
+        45:
+            dict(
+                name='face-21',
+                id=45,
+                color=[255, 255, 255],
+                type='',
+                swap='face-22'),
+        46:
+            dict(
+                name='face-22',
+                id=46,
+                color=[255, 255, 255],
+                type='',
+                swap='face-21'),
+        47:
+            dict(
+                name='face-23',
+                id=47,
+                color=[255, 255, 255],
+                type='',
+                swap='face-20'),
+        48:
+            dict(
+                name='face-24',
+                id=48,
+                color=[255, 255, 255],
+                type='',
+                swap='face-19'),
+        49:
+            dict(
+                name='face-25',
+                id=49,
+                color=[255, 255, 255],
+                type='',
+                swap='face-18'),
+        50:
+            dict(
+                name='face-26',
+                id=50,
+                color=[255, 255, 255],
+                type='',
+                swap='face-17'),
+        51:
+            dict(name='face-27', id=51, color=[255, 255, 255], type='', swap=''),
+        52:
+            dict(name='face-28', id=52, color=[255, 255, 255], type='', swap=''),
+        53:
+            dict(name='face-29', id=53, color=[255, 255, 255], type='', swap=''),
+        54:
+            dict(name='face-30', id=54, color=[255, 255, 255], type='', swap=''),
+        55:
+            dict(
+                name='face-31',
+                id=55,
+                color=[255, 255, 255],
+                type='',
+                swap='face-35'),
+        56:
+            dict(
+                name='face-32',
+                id=56,
+                color=[255, 255, 255],
+                type='',
+                swap='face-34'),
+        57:
+            dict(name='face-33', id=57, color=[255, 255, 255], type='', swap=''),
+        58:
+            dict(
+                name='face-34',
+                id=58,
+                color=[255, 255, 255],
+                type='',
+                swap='face-32'),
+        59:
+            dict(
+                name='face-35',
+                id=59,
+                color=[255, 255, 255],
+                type='',
+                swap='face-31'),
+        60:
+            dict(
+                name='face-36',
+                id=60,
+                color=[255, 255, 255],
+                type='',
+                swap='face-45'),
+        61:
+            dict(
+                name='face-37',
+                id=61,
+                color=[255, 255, 255],
+                type='',
+                swap='face-44'),
+        62:
+            dict(
+                name='face-38',
+                id=62,
+                color=[255, 255, 255],
+                type='',
+                swap='face-43'),
+        63:
+            dict(
+                name='face-39',
+                id=63,
+                color=[255, 255, 255],
+                type='',
+                swap='face-42'),
+        64:
+            dict(
+                name='face-40',
+                id=64,
+                color=[255, 255, 255],
+                type='',
+                swap='face-47'),
+        65:
+            dict(
+                name='face-41',
+                id=65,
+                color=[255, 255, 255],
+                type='',
+                swap='face-46'),
+        66:
+            dict(
+                name='face-42',
+                id=66,
+                color=[255, 255, 255],
+                type='',
+                swap='face-39'),
+        67:
+            dict(
+                name='face-43',
+                id=67,
+                color=[255, 255, 255],
+                type='',
+                swap='face-38'),
+        68:
+            dict(
+                name='face-44',
+                id=68,
+                color=[255, 255, 255],
+                type='',
+                swap='face-37'),
+        69:
+            dict(
+                name='face-45',
+                id=69,
+                color=[255, 255, 255],
+                type='',
+                swap='face-36'),
+        70:
+            dict(
+                name='face-46',
+                id=70,
+                color=[255, 255, 255],
+                type='',
+                swap='face-41'),
+        71:
+            dict(
+                name='face-47',
+                id=71,
+                color=[255, 255, 255],
+                type='',
+                swap='face-40'),
+        72:
+            dict(
+                name='face-48',
+                id=72,
+                color=[255, 255, 255],
+                type='',
+                swap='face-54'),
+        73:
+            dict(
+                name='face-49',
+                id=73,
+                color=[255, 255, 255],
+                type='',
+                swap='face-53'),
+        74:
+            dict(
+                name='face-50',
+                id=74,
+                color=[255, 255, 255],
+                type='',
+                swap='face-52'),
+        75:
+            dict(name='face-51', id=75, color=[255, 255, 255], type='', swap=''),
+        76:
+            dict(
+                name='face-52',
+                id=76,
+                color=[255, 255, 255],
+                type='',
+                swap='face-50'),
+        77:
+            dict(
+                name='face-53',
+                id=77,
+                color=[255, 255, 255],
+                type='',
+                swap='face-49'),
+        78:
+            dict(
+                name='face-54',
+                id=78,
+                color=[255, 255, 255],
+                type='',
+                swap='face-48'),
+        79:
+            dict(
+                name='face-55',
+                id=79,
+                color=[255, 255, 255],
+                type='',
+                swap='face-59'),
+        80:
+            dict(
+                name='face-56',
+                id=80,
+                color=[255, 255, 255],
+                type='',
+                swap='face-58'),
+        81:
+            dict(name='face-57', id=81, color=[255, 255, 255], type='', swap=''),
+        82:
+            dict(
+                name='face-58',
+                id=82,
+                color=[255, 255, 255],
+                type='',
+                swap='face-56'),
+        83:
+            dict(
+                name='face-59',
+                id=83,
+                color=[255, 255, 255],
+                type='',
+                swap='face-55'),
+        84:
+            dict(
+                name='face-60',
+                id=84,
+                color=[255, 255, 255],
+                type='',
+                swap='face-64'),
+        85:
+            dict(
+                name='face-61',
+                id=85,
+                color=[255, 255, 255],
+                type='',
+                swap='face-63'),
+        86:
+            dict(name='face-62', id=86, color=[255, 255, 255], type='', swap=''),
+        87:
+            dict(
+                name='face-63',
+                id=87,
+                color=[255, 255, 255],
+                type='',
+                swap='face-61'),
+        88:
+            dict(
+                name='face-64',
+                id=88,
+                color=[255, 255, 255],
+                type='',
+                swap='face-60'),
+        89:
+            dict(
+                name='face-65',
+                id=89,
+                color=[255, 255, 255],
+                type='',
+                swap='face-67'),
+        90:
+            dict(name='face-66', id=90, color=[255, 255, 255], type='', swap=''),
+        91:
+            dict(
+                name='face-67',
+                id=91,
+                color=[255, 255, 255],
+                type='',
+                swap='face-65'),
+        92:
+            dict(
+                name='left_hand_root',
+                id=92,
+                color=[255, 255, 255],
+                type='',
+                swap='right_hand_root'),
+        93:
+            dict(
+                name='left_thumb1',
+                id=93,
+                color=[255, 128, 0],
+                type='',
+                swap='right_thumb1'),
+        94:
+            dict(
+                name='left_thumb2',
+                id=94,
+                color=[255, 128, 0],
+                type='',
+                swap='right_thumb2'),
+        95:
+            dict(
+                name='left_thumb3',
+                id=95,
+                color=[255, 128, 0],
+                type='',
+                swap='right_thumb3'),
+        96:
+            dict(
+                name='left_thumb4',
+                id=96,
+                color=[255, 128, 0],
+                type='',
+                swap='right_thumb4'),
+        97:
+            dict(
+                name='left_forefinger1',
+                id=97,
+                color=[255, 153, 255],
+                type='',
+                swap='right_forefinger1'),
+        98:
+            dict(
+                name='left_forefinger2',
+                id=98,
+                color=[255, 153, 255],
+                type='',
+                swap='right_forefinger2'),
+        99:
+            dict(
+                name='left_forefinger3',
+                id=99,
+                color=[255, 153, 255],
+                type='',
+                swap='right_forefinger3'),
+        100:
+            dict(
+                name='left_forefinger4',
+                id=100,
+                color=[255, 153, 255],
+                type='',
+                swap='right_forefinger4'),
+        101:
+            dict(
+                name='left_middle_finger1',
+                id=101,
+                color=[102, 178, 255],
+                type='',
+                swap='right_middle_finger1'),
+        102:
+            dict(
+                name='left_middle_finger2',
+                id=102,
+                color=[102, 178, 255],
+                type='',
+                swap='right_middle_finger2'),
+        103:
+            dict(
+                name='left_middle_finger3',
+                id=103,
+                color=[102, 178, 255],
+                type='',
+                swap='right_middle_finger3'),
+        104:
+            dict(
+                name='left_middle_finger4',
+                id=104,
+                color=[102, 178, 255],
+                type='',
+                swap='right_middle_finger4'),
+        105:
+            dict(
+                name='left_ring_finger1',
+                id=105,
+                color=[255, 51, 51],
+                type='',
+                swap='right_ring_finger1'),
+        106:
+            dict(
+                name='left_ring_finger2',
+                id=106,
+                color=[255, 51, 51],
+                type='',
+                swap='right_ring_finger2'),
+        107:
+            dict(
+                name='left_ring_finger3',
+                id=107,
+                color=[255, 51, 51],
+                type='',
+                swap='right_ring_finger3'),
+        108:
+            dict(
+                name='left_ring_finger4',
+                id=108,
+                color=[255, 51, 51],
+                type='',
+                swap='right_ring_finger4'),
+        109:
+            dict(
+                name='left_pinky_finger1',
+                id=109,
+                color=[0, 255, 0],
+                type='',
+                swap='right_pinky_finger1'),
+        110:
+            dict(
+                name='left_pinky_finger2',
+                id=110,
+                color=[0, 255, 0],
+                type='',
+                swap='right_pinky_finger2'),
+        111:
+            dict(
+                name='left_pinky_finger3',
+                id=111,
+                color=[0, 255, 0],
+                type='',
+                swap='right_pinky_finger3'),
+
+        112:
+            dict(
+                name='left_pinky_finger4',
+                id=112,
+                color=[0, 255, 0],
+                type='',
+                swap='right_pinky_finger4'),
+        113:
+            dict(
+                name='right_hand_root',
+                id=113,
+                color=[255, 255, 255],
+                type='',
+                swap='left_hand_root'),
+        114:
+            dict(
+                name='right_thumb1',
+                id=114,
+                color=[255, 128, 0],
+                type='',
+                swap='left_thumb1'),
+        115:
+            dict(
+                name='right_thumb2',
+                id=115,
+                color=[255, 128, 0],
+                type='',
+                swap='left_thumb2'),
+        116:
+            dict(
+                name='right_thumb3',
+                id=116,
+                color=[255, 128, 0],
+                type='',
+                swap='left_thumb3'),
+        117:
+            dict(
+                name='right_thumb4',
+                id=117,
+                color=[255, 128, 0],
+                type='',
+                swap='left_thumb4'),
+        118:
+            dict(
+                name='right_forefinger1',
+                id=118,
+                color=[255, 153, 255],
+                type='',
+                swap='left_forefinger1'),
+        119:
+            dict(
+                name='right_forefinger2',
+                id=119,
+                color=[255, 153, 255],
+                type='',
+                swap='left_forefinger2'),
+        120:
+            dict(
+                name='right_forefinger3',
+                id=120,
+                color=[255, 153, 255],
+                type='',
+                swap='left_forefinger3'),
+        121:
+            dict(
+                name='right_forefinger4',
+                id=121,
+                color=[255, 153, 255],
+                type='',
+                swap='left_forefinger4'),
+        122:
+            dict(
+                name='right_middle_finger1',
+                id=122,
+                color=[102, 178, 255],
+                type='',
+                swap='left_middle_finger1'),
+        123:
+            dict(
+                name='right_middle_finger2',
+                id=123,
+                color=[102, 178, 255],
+                type='',
+                swap='left_middle_finger2'),
+        124:
+            dict(
+                name='right_middle_finger3',
+                id=124,
+                color=[102, 178, 255],
+                type='',
+                swap='left_middle_finger3'),
+        125:
+            dict(
+                name='right_middle_finger4',
+                id=125,
+                color=[102, 178, 255],
+                type='',
+                swap='left_middle_finger4'),
+        126:
+            dict(
+                name='right_ring_finger1',
+                id=126,
+                color=[255, 51, 51],
+                type='',
+                swap='left_ring_finger1'),
+        127:
+            dict(
+                name='right_ring_finger2',
+                id=127,
+                color=[255, 51, 51],
+                type='',
+                swap='left_ring_finger2'),
+        128:
+            dict(
+                name='right_ring_finger3',
+                id=128,
+                color=[255, 51, 51],
+                type='',
+                swap='left_ring_finger3'),
+        129:
+            dict(
+                name='right_ring_finger4',
+                id=129,
+                color=[255, 51, 51],
+                type='',
+                swap='left_ring_finger4'),
+        130:
+            dict(
+                name='right_pinky_finger1',
+                id=130,
+                color=[0, 255, 0],
+                type='',
+                swap='left_pinky_finger1'),
+        131:
+            dict(
+                name='right_pinky_finger2',
+                id=131,
+                color=[0, 255, 0],
+                type='',
+                swap='left_pinky_finger2'),
+        132:
+            dict(
+                name='right_pinky_finger3',
+                id=132,
+                color=[0, 255, 0],
+                type='',
+                swap='left_pinky_finger3'),
+        133:
+            dict(
+                name='right_pinky_finger4',
+                id=133,
+                color=[0, 255, 0],
+                type='',
+                swap='left_pinky_finger4'),
+    },
+    skeleton_info={
+        0: dict(link=('neck', 'right_shoulder'), id=0, color=[255, 0, 0]),
+        1: dict(link=('neck', 'left_shoulder'), id=1, color=[255, 85, 0]),
+        2: dict(
+            link=('right_shoulder', 'right_elbow'), id=2, color=[255, 170, 0]),
+        3:
+        dict(link=('right_elbow', 'right_wrist'), id=3, color=[255, 255, 0]),
+        4:
+        dict(link=('left_shoulder', 'left_elbow'), id=4, color=[170, 255, 0]),
+        5: dict(link=('left_elbow', 'left_wrist'), id=5, color=[85, 255, 0]),
+        6: dict(link=('neck', 'right_hip'), id=6, color=[0, 255, 0]),
+        7: dict(link=('right_hip', 'right_knee'), id=7, color=[0, 255, 85]),
+        8: dict(link=('right_knee', 'right_ankle'), id=8, color=[0, 255, 170]),
+        9: dict(link=('neck', 'left_hip'), id=9, color=[0, 255, 225]),
+        10: dict(link=('left_hip', 'left_knee'), id=10, color=[0, 170, 255]),
+        11: dict(link=('left_knee', 'left_ankle'), id=11, color=[0, 85, 255]),
+        12: dict(link=('neck', 'nose'), id=12, color=[0, 0, 255]),
+        13: dict(link=('nose', 'right_eye'), id=13, color=[255, 0, 170]),
+        14: dict(link=('right_eye', 'right_ear'), id=14, color=[170, 0, 255]),
+        15: dict(link=('nose', 'left_eye'), id=15, color=[255, 0, 255]),
+        16: dict(link=('left_eye', 'left_ear'), id=16, color=[255, 0, 170]),
+        17:
+            dict(link=('left_ankle', 'left_big_toe'), id=17, color=[0, 255, 0]),
+        18:
+            dict(link=('left_ankle', 'left_small_toe'), id=18, color=[0, 255, 0]),
+        19:
+            dict(link=('left_ankle', 'left_heel'), id=19, color=[0, 255, 0]),
+        20:
+            dict(link=('right_ankle', 'right_big_toe'), id=20, color=[255, 128, 0]),
+        21:
+            dict(link=('right_ankle', 'right_small_toe'), id=21, color=[255, 128, 0]),
+        22:
+            dict(link=('right_ankle', 'right_heel'), id=22, color=[255, 128, 0]),
+        23:
+            dict(link=('left_hand_root', 'left_thumb1'), id=23, color=[255, 128, 0]),
+        24:
+            dict(link=('left_thumb1', 'left_thumb2'), id=24, color=[255, 128, 0]),
+        25:
+            dict(link=('left_thumb2', 'left_thumb3'), id=25, color=[255, 128, 0]),
+        26:
+            dict(link=('left_thumb3', 'left_thumb4'), id=26, color=[255, 128, 0]),
+        27:
+            dict(link=('left_hand_root', 'left_forefinger1'), id=27, color=[255, 153, 255]),
+        28:
+            dict(link=('left_forefinger1', 'left_forefinger2'), id=28, color=[255, 153, 255]),
+        29:
+            dict(link=('left_forefinger2', 'left_forefinger3'), id=29, color=[255, 153, 255]),
+        30:
+            dict(link=('left_forefinger3', 'left_forefinger4'), id=30, color=[255, 153, 255]),
+        31:
+            dict(link=('left_hand_root', 'left_middle_finger1'), id=31, color=[102, 178, 255]),
+        32:
+            dict(link=('left_middle_finger1', 'left_middle_finger2'), id=32, color=[102, 178, 255]),
+        33:
+            dict(link=('left_middle_finger2', 'left_middle_finger3'), id=33, color=[102, 178, 255]),
+        34:
+            dict(link=('left_middle_finger3', 'left_middle_finger4'), id=34, color=[102, 178, 255]),
+        35:
+            dict(link=('left_hand_root', 'left_ring_finger1'), id=35, color=[255, 51, 51]),
+        36:
+            dict(link=('left_ring_finger1', 'left_ring_finger2'), id=36, color=[255, 51, 51]),
+        37:
+            dict(link=('left_ring_finger2', 'left_ring_finger3'), id=37, color=[255, 51, 51]),
+        38:
+            dict(link=('left_ring_finger3', 'left_ring_finger4'), id=38, color=[255, 51, 51]),
+        39:
+            dict(link=('left_hand_root', 'left_pinky_finger1'), id=39, color=[0, 255, 0]),
+        40:
+            dict(link=('left_pinky_finger1', 'left_pinky_finger2'), id=40, color=[0, 255, 0]),
+        41:
+            dict(link=('left_pinky_finger2', 'left_pinky_finger3'), id=41, color=[0, 255, 0]),
+        42:
+            dict(link=('left_pinky_finger3', 'left_pinky_finger4'), id=42, color=[0, 255, 0]),
+        43:
+            dict(link=('right_hand_root', 'right_thumb1'), id=43, color=[255, 128, 0]),
+        44:
+            dict(link=('right_thumb1', 'right_thumb2'), id=44, color=[255, 128, 0]),
+        45:
+            dict(link=('right_thumb2', 'right_thumb3'), id=45, color=[255, 128, 0]),
+        46:
+            dict(link=('right_thumb3', 'right_thumb4'), id=46, color=[255, 128, 0]),
+        47:
+            dict(link=('right_hand_root', 'right_forefinger1'), id=47, color=[255, 153, 255]),
+        48:
+            dict(link=('right_forefinger1', 'right_forefinger2'), id=48, color=[255, 153, 255]),
+        49:
+            dict(link=('right_forefinger2', 'right_forefinger3'), id=49, color=[255, 153, 255]),
+        50:
+            dict(link=('right_forefinger3', 'right_forefinger4'), id=50, color=[255, 153, 255]),
+        51:
+            dict(link=('right_hand_root', 'right_middle_finger1'), id=51, color=[102, 178, 255]),
+        52:
+            dict(link=('right_middle_finger1', 'right_middle_finger2'), id=52, color=[102, 178, 255]),
+        53:
+            dict(link=('right_middle_finger2', 'right_middle_finger3'), id=53, color=[102, 178, 255]),
+        54:
+            dict(link=('right_middle_finger3', 'right_middle_finger4'), id=54, color=[102, 178, 255]),
+        55:
+            dict(link=('right_hand_root', 'right_ring_finger1'), id=55, color=[255, 51, 51]),
+        56:
+            dict(link=('right_ring_finger1', 'right_ring_finger2'), id=56, color=[255, 51, 51]),
+        57:
+            dict(link=('right_ring_finger2', 'right_ring_finger3'), id=57, color=[255, 51, 51]),
+        58:
+            dict(link=('right_ring_finger3', 'right_ring_finger4'), id=58, color=[255, 51, 51]),
+        59:
+            dict(link=('right_hand_root', 'right_pinky_finger1'), id=59, color=[0, 255, 0]),
+        60:
+            dict(link=('right_pinky_finger1', 'right_pinky_finger2'), id=60, color=[0, 255, 0]),
+        61:
+            dict(link=('right_pinky_finger2', 'right_pinky_finger3'), id=61, color=[0, 255, 0]),
+        62:
+            dict(link=('right_pinky_finger3', 'right_pinky_finger4'), id=62, color=[0, 255, 0])
+
+    },
+    joint_weights=[1.] * 134,
+    # 'https://github.com/jin-s13/COCO-WholeBody/blob/master/'
+    # 'evaluation/myeval_wholebody.py#L175'
+    sigmas=[
+        0.025,
+        0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
+        0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089, 0.068, 0.066, 0.066,
+        0.092, 0.094, 0.094, 0.042, 0.043, 0.044, 0.043, 0.040, 0.035, 0.031,
+        0.025, 0.020, 0.023, 0.029, 0.032, 0.037, 0.038, 0.043, 0.041, 0.045,
+        0.013, 0.012, 0.011, 0.011, 0.012, 0.012, 0.011, 0.011, 0.013, 0.015,
+        0.009, 0.007, 0.007, 0.007, 0.012, 0.009, 0.008, 0.016, 0.010, 0.017,
+        0.011, 0.009, 0.011, 0.009, 0.007, 0.013, 0.008, 0.011, 0.012, 0.010,
+        0.034, 0.008, 0.008, 0.009, 0.008, 0.008, 0.007, 0.010, 0.008, 0.009,
+        0.009, 0.009, 0.007, 0.007, 0.008, 0.011, 0.008, 0.008, 0.008, 0.01,
+        0.008, 0.029, 0.022, 0.035, 0.037, 0.047, 0.026, 0.025, 0.024, 0.035,
+        0.018, 0.024, 0.022, 0.026, 0.017, 0.021, 0.021, 0.032, 0.02, 0.019,
+        0.022, 0.031, 0.029, 0.022, 0.035, 0.037, 0.047, 0.026, 0.025, 0.024,
+        0.035, 0.018, 0.024, 0.022, 0.026, 0.017, 0.021, 0.021, 0.032, 0.02,
+        0.019, 0.022, 0.031
+    ])

configs/_base_/datasets/cofw.py

@@ -0,0 +1,63 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='cofw',
+    paper_info=dict(
+        author='Burgos-Artizzu, Xavier P and Perona, '
+        r'Pietro and Doll{\'a}r, Piotr',
+        title='Robust face landmark estimation under occlusion',
+        container='Proceedings of the IEEE international '
+        'conference on computer vision',
+        year='2013',
+        homepage='http://www.vision.caltech.edu/xpburgos/ICCV13/',
+    ),
+    keypoint_info={
+        0: dict(name='kpt-0', id=0, color=[255, 0, 0], type='', swap='kpt-1'),
+        1: dict(name='kpt-1', id=1, color=[255, 0, 0], type='', swap='kpt-0'),
+        2: dict(name='kpt-2', id=2, color=[255, 0, 0], type='', swap='kpt-3'),
+        3: dict(name='kpt-3', id=3, color=[255, 0, 0], type='', swap='kpt-2'),
+        4: dict(name='kpt-4', id=4, color=[255, 0, 0], type='', swap='kpt-6'),
+        5: dict(name='kpt-5', id=5, color=[255, 0, 0], type='', swap='kpt-7'),
+        6: dict(name='kpt-6', id=6, color=[255, 0, 0], type='', swap='kpt-4'),
+        7: dict(name='kpt-7', id=7, color=[255, 0, 0], type='', swap='kpt-5'),
+        8: dict(name='kpt-8', id=8, color=[255, 0, 0], type='', swap='kpt-9'),
+        9: dict(name='kpt-9', id=9, color=[255, 0, 0], type='', swap='kpt-8'),
+        10:
+        dict(name='kpt-10', id=10, color=[255, 0, 0], type='', swap='kpt-11'),
+        11:
+        dict(name='kpt-11', id=11, color=[255, 0, 0], type='', swap='kpt-10'),
+        12:
+        dict(name='kpt-12', id=12, color=[255, 0, 0], type='', swap='kpt-14'),
+        13:
+        dict(name='kpt-13', id=13, color=[255, 0, 0], type='', swap='kpt-15'),
+        14:
+        dict(name='kpt-14', id=14, color=[255, 0, 0], type='', swap='kpt-12'),
+        15:
+        dict(name='kpt-15', id=15, color=[255, 0, 0], type='', swap='kpt-13'),
+        16:
+        dict(name='kpt-16', id=16, color=[255, 0, 0], type='', swap='kpt-17'),
+        17:
+        dict(name='kpt-17', id=17, color=[255, 0, 0], type='', swap='kpt-16'),
+        18:
+        dict(name='kpt-18', id=18, color=[255, 0, 0], type='', swap='kpt-19'),
+        19:
+        dict(name='kpt-19', id=19, color=[255, 0, 0], type='', swap='kpt-18'),
+        20: dict(name='kpt-20', id=20, color=[255, 0, 0], type='', swap=''),
+        21: dict(name='kpt-21', id=21, color=[255, 0, 0], type='', swap=''),
+        22:
+        dict(name='kpt-22', id=22, color=[255, 0, 0], type='', swap='kpt-23'),
+        23:
+        dict(name='kpt-23', id=23, color=[255, 0, 0], type='', swap='kpt-22'),
+        24: dict(name='kpt-24', id=24, color=[255, 0, 0], type='', swap=''),
+        25: dict(name='kpt-25', id=25, color=[255, 0, 0], type='', swap=''),
+        26: dict(name='kpt-26', id=26, color=[255, 0, 0], type='', swap=''),
+        27: dict(name='kpt-27', id=27, color=[255, 0, 0], type='', swap=''),
+        28: dict(name='kpt-28', id=28, color=[255, 0, 0], type='', swap='')
+    },
+    skeleton_info={},
+    joint_weights=[1.] * 29,
+    sigmas=[])

configs/_base_/datasets/crowdpose.py

@@ -0,0 +1,153 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='crowdpose',
+    paper_info=dict(
+        author='Li, Jiefeng and Wang, Can and Zhu, Hao and '
+        'Mao, Yihuan and Fang, Hao-Shu and Lu, Cewu',
+        title='CrowdPose: Efficient Crowded Scenes Pose Estimation '
+        'and A New Benchmark',
+        container='Proceedings of IEEE Conference on Computer '
+        'Vision and Pattern Recognition (CVPR)',
+        year='2019',
+        homepage='https://github.com/Jeff-sjtu/CrowdPose',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='left_shoulder',
+            id=0,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_shoulder'),
+        1:
+        dict(
+            name='right_shoulder',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_shoulder'),
+        2:
+        dict(
+            name='left_elbow',
+            id=2,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_elbow'),
+        3:
+        dict(
+            name='right_elbow',
+            id=3,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_elbow'),
+        4:
+        dict(
+            name='left_wrist',
+            id=4,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_wrist'),
+        5:
+        dict(
+            name='right_wrist',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='left_wrist'),
+        6:
+        dict(
+            name='left_hip',
+            id=6,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_hip'),
+        7:
+        dict(
+            name='right_hip',
+            id=7,
+            color=[0, 255, 0],
+            type='lower',
+            swap='left_hip'),
+        8:
+        dict(
+            name='left_knee',
+            id=8,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_knee'),
+        9:
+        dict(
+            name='right_knee',
+            id=9,
+            color=[0, 255, 0],
+            type='lower',
+            swap='left_knee'),
+        10:
+        dict(
+            name='left_ankle',
+            id=10,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_ankle'),
+        11:
+        dict(
+            name='right_ankle',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='left_ankle'),
+        12:
+        dict(
+            name='top_head', id=12, color=[255, 128, 0], type='upper',
+            swap=''),
+        13:
+        dict(name='neck', id=13, color=[0, 255, 0], type='upper', swap='')
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
+        5:
+        dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
+        6:
+        dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
+        7:
+        dict(
+            link=('left_shoulder', 'right_shoulder'),
+            id=7,
+            color=[51, 153, 255]),
+        8:
+        dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
+        9:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('top_head', 'neck'), id=12, color=[51, 153, 255]),
+        13:
+        dict(link=('right_shoulder', 'neck'), id=13, color=[51, 153, 255]),
+        14:
+        dict(link=('left_shoulder', 'neck'), id=14, color=[51, 153, 255])
+    },
+    joint_weights=[
+        0.2, 0.2, 0.2, 1.3, 1.5, 0.2, 1.3, 1.5, 0.2, 0.2, 0.5, 0.2, 0.2, 0.5
+    ],
+    sigmas=[
+        0.079, 0.079, 0.072, 0.072, 0.062, 0.062, 0.107, 0.107, 0.087, 0.087,
+        0.089, 0.089, 0.079, 0.079
+    ])

configs/_base_/datasets/crowdpose2coco.py

@@ -0,0 +1,187 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='crowdpose2coco',
+    paper_info=dict(
+        author='Lin, Tsung-Yi and Maire, Michael and '
+        'Belongie, Serge and Hays, James and '
+        'Perona, Pietro and Ramanan, Deva and '
+        r'Doll{\'a}r, Piotr and Zitnick, C Lawrence',
+        title='Microsoft coco: Common objects in context',
+        container='European conference on computer vision',
+        year='2014',
+        homepage='http://cocodataset.org/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
+        1:
+        dict(
+            name='left_eye',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_eye'),
+        2:
+        dict(
+            name='right_eye',
+            id=2,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_eye'),
+        3:
+        dict(
+            name='left_ear',
+            id=3,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_ear'),
+        4:
+        dict(
+            name='right_ear',
+            id=4,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_ear'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='right_shoulder',
+            id=6,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        7:
+        dict(
+            name='left_elbow',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        8:
+        dict(
+            name='right_elbow',
+            id=8,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        9:
+        dict(
+            name='left_wrist',
+            id=9,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='right_hip',
+            id=12,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        13:
+        dict(
+            name='left_knee',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        14:
+        dict(
+            name='right_knee',
+            id=14,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        15:
+        dict(
+            name='left_ankle',
+            id=15,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        16:
+        dict(
+            name='right_ankle',
+            id=16,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle')
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
+        5:
+        dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
+        6:
+        dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
+        7:
+        dict(
+            link=('left_shoulder', 'right_shoulder'),
+            id=7,
+            color=[51, 153, 255]),
+        8:
+        dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
+        9:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
+        13:
+        dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
+        14:
+        dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
+        15:
+        dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
+        16:
+        dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
+        17:
+        dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
+        18:
+        dict(
+            link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255])
+    },
+    joint_weights=[
+        1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
+        1.5
+    ],
+    sigmas=[
+        0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
+        0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089
+    ])

configs/_base_/datasets/crowdpose2goliath.py

@@ -0,0 +1,11 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+from configs._base_.datasets.goliath import dataset_info as goliath_info
+
+goliath_info = goliath_info.build()
+dataset_info = goliath_info.copy()
+dataset_info['dataset_name'] = 'crowdpose2goliath'

configs/_base_/datasets/crowdpose2goliath3d.py

@@ -0,0 +1,11 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+from configs._base_.datasets.goliath3d import dataset_info as goliath_info
+
+goliath_info = goliath_info.build()
+dataset_info = goliath_info.copy()
+dataset_info['dataset_name'] = 'crowdpose2goliath3d'

configs/_base_/datasets/deepfashion2.py

@@ -0,0 +1,2666 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+colors = dict(
+    sss=[255, 128, 0],  # short_sleeve_shirt
+    lss=[255, 0, 128],  # long_sleeved_shirt
+    sso=[128, 0, 255],  # short_sleeved_outwear
+    lso=[0, 128, 255],  # long_sleeved_outwear
+    vest=[0, 128, 128],  # vest
+    sling=[0, 0, 128],  # sling
+    shorts=[128, 128, 128],  # shorts
+    trousers=[128, 0, 128],  # trousers
+    skirt=[64, 128, 128],  # skirt
+    ssd=[64, 64, 128],  # short_sleeved_dress
+    lsd=[128, 64, 0],  # long_sleeved_dress
+    vd=[128, 64, 255],  # vest_dress
+    sd=[128, 64, 0],  # sling_dress
+)
+dataset_info = dict(
+    dataset_name='deepfashion2',
+    paper_info=dict(
+        author='Yuying Ge and Ruimao Zhang and Lingyun Wu '
+        'and Xiaogang Wang and Xiaoou Tang and Ping Luo',
+        title='DeepFashion2: A Versatile Benchmark for '
+        'Detection, Pose Estimation, Segmentation and '
+        'Re-Identification of Clothing Images',
+        container='Proceedings of IEEE Conference on Computer '
+        'Vision and Pattern Recognition (CVPR)',
+        year='2019',
+        homepage='https://github.com/switchablenorms/DeepFashion2',
+    ),
+    keypoint_info={
+        # short_sleeved_shirt
+        0:
+        dict(name='sss_kpt1', id=0, color=colors['sss'], type='', swap=''),
+        1:
+        dict(
+            name='sss_kpt2',
+            id=1,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt6'),
+        2:
+        dict(
+            name='sss_kpt3',
+            id=2,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt5'),
+        3:
+        dict(name='sss_kpt4', id=3, color=colors['sss'], type='', swap=''),
+        4:
+        dict(
+            name='sss_kpt5',
+            id=4,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt3'),
+        5:
+        dict(
+            name='sss_kpt6',
+            id=5,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt2'),
+        6:
+        dict(
+            name='sss_kpt7',
+            id=6,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt25'),
+        7:
+        dict(
+            name='sss_kpt8',
+            id=7,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt24'),
+        8:
+        dict(
+            name='sss_kpt9',
+            id=8,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt23'),
+        9:
+        dict(
+            name='sss_kpt10',
+            id=9,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt22'),
+        10:
+        dict(
+            name='sss_kpt11',
+            id=10,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt21'),
+        11:
+        dict(
+            name='sss_kpt12',
+            id=11,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt20'),
+        12:
+        dict(
+            name='sss_kpt13',
+            id=12,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt19'),
+        13:
+        dict(
+            name='sss_kpt14',
+            id=13,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt18'),
+        14:
+        dict(
+            name='sss_kpt15',
+            id=14,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt17'),
+        15:
+        dict(name='sss_kpt16', id=15, color=colors['sss'], type='', swap=''),
+        16:
+        dict(
+            name='sss_kpt17',
+            id=16,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt15'),
+        17:
+        dict(
+            name='sss_kpt18',
+            id=17,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt14'),
+        18:
+        dict(
+            name='sss_kpt19',
+            id=18,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt13'),
+        19:
+        dict(
+            name='sss_kpt20',
+            id=19,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt12'),
+        20:
+        dict(
+            name='sss_kpt21',
+            id=20,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt11'),
+        21:
+        dict(
+            name='sss_kpt22',
+            id=21,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt10'),
+        22:
+        dict(
+            name='sss_kpt23',
+            id=22,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt9'),
+        23:
+        dict(
+            name='sss_kpt24',
+            id=23,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt8'),
+        24:
+        dict(
+            name='sss_kpt25',
+            id=24,
+            color=colors['sss'],
+            type='',
+            swap='sss_kpt7'),
+        # long_sleeved_shirt
+        25:
+        dict(name='lss_kpt1', id=25, color=colors['lss'], type='', swap=''),
+        26:
+        dict(
+            name='lss_kpt2',
+            id=26,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt6'),
+        27:
+        dict(
+            name='lss_kpt3',
+            id=27,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt5'),
+        28:
+        dict(name='lss_kpt4', id=28, color=colors['lss'], type='', swap=''),
+        29:
+        dict(
+            name='lss_kpt5',
+            id=29,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt3'),
+        30:
+        dict(
+            name='lss_kpt6',
+            id=30,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt2'),
+        31:
+        dict(
+            name='lss_kpt7',
+            id=31,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt33'),
+        32:
+        dict(
+            name='lss_kpt8',
+            id=32,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt32'),
+        33:
+        dict(
+            name='lss_kpt9',
+            id=33,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt31'),
+        34:
+        dict(
+            name='lss_kpt10',
+            id=34,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt30'),
+        35:
+        dict(
+            name='lss_kpt11',
+            id=35,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt29'),
+        36:
+        dict(
+            name='lss_kpt12',
+            id=36,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt28'),
+        37:
+        dict(
+            name='lss_kpt13',
+            id=37,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt27'),
+        38:
+        dict(
+            name='lss_kpt14',
+            id=38,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt26'),
+        39:
+        dict(
+            name='lss_kpt15',
+            id=39,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt25'),
+        40:
+        dict(
+            name='lss_kpt16',
+            id=40,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt24'),
+        41:
+        dict(
+            name='lss_kpt17',
+            id=41,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt23'),
+        42:
+        dict(
+            name='lss_kpt18',
+            id=42,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt22'),
+        43:
+        dict(
+            name='lss_kpt19',
+            id=43,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt21'),
+        44:
+        dict(name='lss_kpt20', id=44, color=colors['lss'], type='', swap=''),
+        45:
+        dict(
+            name='lss_kpt21',
+            id=45,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt19'),
+        46:
+        dict(
+            name='lss_kpt22',
+            id=46,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt18'),
+        47:
+        dict(
+            name='lss_kpt23',
+            id=47,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt17'),
+        48:
+        dict(
+            name='lss_kpt24',
+            id=48,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt16'),
+        49:
+        dict(
+            name='lss_kpt25',
+            id=49,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt15'),
+        50:
+        dict(
+            name='lss_kpt26',
+            id=50,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt14'),
+        51:
+        dict(
+            name='lss_kpt27',
+            id=51,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt13'),
+        52:
+        dict(
+            name='lss_kpt28',
+            id=52,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt12'),
+        53:
+        dict(
+            name='lss_kpt29',
+            id=53,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt11'),
+        54:
+        dict(
+            name='lss_kpt30',
+            id=54,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt10'),
+        55:
+        dict(
+            name='lss_kpt31',
+            id=55,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt9'),
+        56:
+        dict(
+            name='lss_kpt32',
+            id=56,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt8'),
+        57:
+        dict(
+            name='lss_kpt33',
+            id=57,
+            color=colors['lss'],
+            type='',
+            swap='lss_kpt7'),
+        # short_sleeved_outwear
+        58:
+        dict(name='sso_kpt1', id=58, color=colors['sso'], type='', swap=''),
+        59:
+        dict(
+            name='sso_kpt2',
+            id=59,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt26'),
+        60:
+        dict(
+            name='sso_kpt3',
+            id=60,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt5'),
+        61:
+        dict(
+            name='sso_kpt4',
+            id=61,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt6'),
+        62:
+        dict(
+            name='sso_kpt5',
+            id=62,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt3'),
+        63:
+        dict(
+            name='sso_kpt6',
+            id=63,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt4'),
+        64:
+        dict(
+            name='sso_kpt7',
+            id=64,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt25'),
+        65:
+        dict(
+            name='sso_kpt8',
+            id=65,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt24'),
+        66:
+        dict(
+            name='sso_kpt9',
+            id=66,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt23'),
+        67:
+        dict(
+            name='sso_kpt10',
+            id=67,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt22'),
+        68:
+        dict(
+            name='sso_kpt11',
+            id=68,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt21'),
+        69:
+        dict(
+            name='sso_kpt12',
+            id=69,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt20'),
+        70:
+        dict(
+            name='sso_kpt13',
+            id=70,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt19'),
+        71:
+        dict(
+            name='sso_kpt14',
+            id=71,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt18'),
+        72:
+        dict(
+            name='sso_kpt15',
+            id=72,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt17'),
+        73:
+        dict(
+            name='sso_kpt16',
+            id=73,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt29'),
+        74:
+        dict(
+            name='sso_kpt17',
+            id=74,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt15'),
+        75:
+        dict(
+            name='sso_kpt18',
+            id=75,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt14'),
+        76:
+        dict(
+            name='sso_kpt19',
+            id=76,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt13'),
+        77:
+        dict(
+            name='sso_kpt20',
+            id=77,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt12'),
+        78:
+        dict(
+            name='sso_kpt21',
+            id=78,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt11'),
+        79:
+        dict(
+            name='sso_kpt22',
+            id=79,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt10'),
+        80:
+        dict(
+            name='sso_kpt23',
+            id=80,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt9'),
+        81:
+        dict(
+            name='sso_kpt24',
+            id=81,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt8'),
+        82:
+        dict(
+            name='sso_kpt25',
+            id=82,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt7'),
+        83:
+        dict(
+            name='sso_kpt26',
+            id=83,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt2'),
+        84:
+        dict(
+            name='sso_kpt27',
+            id=84,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt30'),
+        85:
+        dict(
+            name='sso_kpt28',
+            id=85,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt31'),
+        86:
+        dict(
+            name='sso_kpt29',
+            id=86,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt16'),
+        87:
+        dict(
+            name='sso_kpt30',
+            id=87,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt27'),
+        88:
+        dict(
+            name='sso_kpt31',
+            id=88,
+            color=colors['sso'],
+            type='',
+            swap='sso_kpt28'),
+        # long_sleeved_outwear
+        89:
+        dict(name='lso_kpt1', id=89, color=colors['lso'], type='', swap=''),
+        90:
+        dict(
+            name='lso_kpt2',
+            id=90,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt6'),
+        91:
+        dict(
+            name='lso_kpt3',
+            id=91,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt5'),
+        92:
+        dict(
+            name='lso_kpt4',
+            id=92,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt34'),
+        93:
+        dict(
+            name='lso_kpt5',
+            id=93,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt3'),
+        94:
+        dict(
+            name='lso_kpt6',
+            id=94,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt2'),
+        95:
+        dict(
+            name='lso_kpt7',
+            id=95,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt33'),
+        96:
+        dict(
+            name='lso_kpt8',
+            id=96,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt32'),
+        97:
+        dict(
+            name='lso_kpt9',
+            id=97,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt31'),
+        98:
+        dict(
+            name='lso_kpt10',
+            id=98,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt30'),
+        99:
+        dict(
+            name='lso_kpt11',
+            id=99,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt29'),
+        100:
+        dict(
+            name='lso_kpt12',
+            id=100,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt28'),
+        101:
+        dict(
+            name='lso_kpt13',
+            id=101,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt27'),
+        102:
+        dict(
+            name='lso_kpt14',
+            id=102,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt26'),
+        103:
+        dict(
+            name='lso_kpt15',
+            id=103,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt25'),
+        104:
+        dict(
+            name='lso_kpt16',
+            id=104,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt24'),
+        105:
+        dict(
+            name='lso_kpt17',
+            id=105,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt23'),
+        106:
+        dict(
+            name='lso_kpt18',
+            id=106,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt22'),
+        107:
+        dict(
+            name='lso_kpt19',
+            id=107,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt21'),
+        108:
+        dict(
+            name='lso_kpt20',
+            id=108,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt37'),
+        109:
+        dict(
+            name='lso_kpt21',
+            id=109,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt19'),
+        110:
+        dict(
+            name='lso_kpt22',
+            id=110,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt18'),
+        111:
+        dict(
+            name='lso_kpt23',
+            id=111,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt17'),
+        112:
+        dict(
+            name='lso_kpt24',
+            id=112,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt16'),
+        113:
+        dict(
+            name='lso_kpt25',
+            id=113,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt15'),
+        114:
+        dict(
+            name='lso_kpt26',
+            id=114,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt14'),
+        115:
+        dict(
+            name='lso_kpt27',
+            id=115,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt13'),
+        116:
+        dict(
+            name='lso_kpt28',
+            id=116,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt12'),
+        117:
+        dict(
+            name='lso_kpt29',
+            id=117,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt11'),
+        118:
+        dict(
+            name='lso_kpt30',
+            id=118,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt10'),
+        119:
+        dict(
+            name='lso_kpt31',
+            id=119,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt9'),
+        120:
+        dict(
+            name='lso_kpt32',
+            id=120,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt8'),
+        121:
+        dict(
+            name='lso_kpt33',
+            id=121,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt7'),
+        122:
+        dict(
+            name='lso_kpt34',
+            id=122,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt4'),
+        123:
+        dict(
+            name='lso_kpt35',
+            id=123,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt38'),
+        124:
+        dict(
+            name='lso_kpt36',
+            id=124,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt39'),
+        125:
+        dict(
+            name='lso_kpt37',
+            id=125,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt20'),
+        126:
+        dict(
+            name='lso_kpt38',
+            id=126,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt35'),
+        127:
+        dict(
+            name='lso_kpt39',
+            id=127,
+            color=colors['lso'],
+            type='',
+            swap='lso_kpt36'),
+        # vest
+        128:
+        dict(name='vest_kpt1', id=128, color=colors['vest'], type='', swap=''),
+        129:
+        dict(
+            name='vest_kpt2',
+            id=129,
+            color=colors['vest'],
+            type='',
+            swap='vest_kpt6'),
+        130:
+        dict(
+            name='vest_kpt3',
+            id=130,
+            color=colors['vest'],
+            type='',
+            swap='vest_kpt5'),
+        131:
+        dict(name='vest_kpt4', id=131, color=colors['vest'], type='', swap=''),
+        132:
+        dict(
+            name='vest_kpt5',
+            id=132,
+            color=colors['vest'],
+            type='',
+            swap='vest_kpt3'),
+        133:
+        dict(
+            name='vest_kpt6',
+            id=133,
+            color=colors['vest'],
+            type='',
+            swap='vest_kpt2'),
+        134:
+        dict(
+            name='vest_kpt7',
+            id=134,
+            color=colors['vest'],
+            type='',
+            swap='vest_kpt15'),
+        135:
+        dict(
+            name='vest_kpt8',
+            id=135,
+            color=colors['vest'],
+            type='',
+            swap='vest_kpt14'),
+        136:
+        dict(
+            name='vest_kpt9',
+            id=136,
+            color=colors['vest'],
+            type='',
+            swap='vest_kpt13'),
+        137:
+        dict(
+            name='vest_kpt10',
+            id=137,
+            color=colors['vest'],
+            type='',
+            swap='vest_kpt12'),
+        138:
+        dict(
+            name='vest_kpt11', id=138, color=colors['vest'], type='', swap=''),
+        139:
+        dict(
+            name='vest_kpt12',
+            id=139,
+            color=colors['vest'],
+            type='',
+            swap='vest_kpt10'),
+        140:
+        dict(
+            name='vest_kpt13', id=140, color=colors['vest'], type='', swap=''),
+        141:
+        dict(
+            name='vest_kpt14',
+            id=141,
+            color=colors['vest'],
+            type='',
+            swap='vest_kpt8'),
+        142:
+        dict(
+            name='vest_kpt15',
+            id=142,
+            color=colors['vest'],
+            type='',
+            swap='vest_kpt7'),
+        # sling
+        143:
+        dict(
+            name='sling_kpt1', id=143, color=colors['sling'], type='',
+            swap=''),
+        144:
+        dict(
+            name='sling_kpt2',
+            id=144,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt6'),
+        145:
+        dict(
+            name='sling_kpt3',
+            id=145,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt5'),
+        146:
+        dict(
+            name='sling_kpt4', id=146, color=colors['sling'], type='',
+            swap=''),
+        147:
+        dict(
+            name='sling_kpt5',
+            id=147,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt3'),
+        148:
+        dict(
+            name='sling_kpt6',
+            id=148,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt2'),
+        149:
+        dict(
+            name='sling_kpt7',
+            id=149,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt15'),
+        150:
+        dict(
+            name='sling_kpt8',
+            id=150,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt14'),
+        151:
+        dict(
+            name='sling_kpt9',
+            id=151,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt13'),
+        152:
+        dict(
+            name='sling_kpt10',
+            id=152,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt12'),
+        153:
+        dict(
+            name='sling_kpt11',
+            id=153,
+            color=colors['sling'],
+            type='',
+            swap=''),
+        154:
+        dict(
+            name='sling_kpt12',
+            id=154,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt10'),
+        155:
+        dict(
+            name='sling_kpt13',
+            id=155,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt9'),
+        156:
+        dict(
+            name='sling_kpt14',
+            id=156,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt8'),
+        157:
+        dict(
+            name='sling_kpt15',
+            id=157,
+            color=colors['sling'],
+            type='',
+            swap='sling_kpt7'),
+        # shorts
+        158:
+        dict(
+            name='shorts_kpt1',
+            id=158,
+            color=colors['shorts'],
+            type='',
+            swap='shorts_kpt3'),
+        159:
+        dict(
+            name='shorts_kpt2',
+            id=159,
+            color=colors['shorts'],
+            type='',
+            swap=''),
+        160:
+        dict(
+            name='shorts_kpt3',
+            id=160,
+            color=colors['shorts'],
+            type='',
+            swap='shorts_kpt1'),
+        161:
+        dict(
+            name='shorts_kpt4',
+            id=161,
+            color=colors['shorts'],
+            type='',
+            swap='shorts_kpt10'),
+        162:
+        dict(
+            name='shorts_kpt5',
+            id=162,
+            color=colors['shorts'],
+            type='',
+            swap='shorts_kpt9'),
+        163:
+        dict(
+            name='shorts_kpt6',
+            id=163,
+            color=colors['shorts'],
+            type='',
+            swap='shorts_kpt8'),
+        164:
+        dict(
+            name='shorts_kpt7',
+            id=164,
+            color=colors['shorts'],
+            type='',
+            swap=''),
+        165:
+        dict(
+            name='shorts_kpt8',
+            id=165,
+            color=colors['shorts'],
+            type='',
+            swap='shorts_kpt6'),
+        166:
+        dict(
+            name='shorts_kpt9',
+            id=166,
+            color=colors['shorts'],
+            type='',
+            swap='shorts_kpt5'),
+        167:
+        dict(
+            name='shorts_kpt10',
+            id=167,
+            color=colors['shorts'],
+            type='',
+            swap='shorts_kpt4'),
+        # trousers
+        168:
+        dict(
+            name='trousers_kpt1',
+            id=168,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt3'),
+        169:
+        dict(
+            name='trousers_kpt2',
+            id=169,
+            color=colors['trousers'],
+            type='',
+            swap=''),
+        170:
+        dict(
+            name='trousers_kpt3',
+            id=170,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt1'),
+        171:
+        dict(
+            name='trousers_kpt4',
+            id=171,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt14'),
+        172:
+        dict(
+            name='trousers_kpt5',
+            id=172,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt13'),
+        173:
+        dict(
+            name='trousers_kpt6',
+            id=173,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt12'),
+        174:
+        dict(
+            name='trousers_kpt7',
+            id=174,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt11'),
+        175:
+        dict(
+            name='trousers_kpt8',
+            id=175,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt10'),
+        176:
+        dict(
+            name='trousers_kpt9',
+            id=176,
+            color=colors['trousers'],
+            type='',
+            swap=''),
+        177:
+        dict(
+            name='trousers_kpt10',
+            id=177,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt8'),
+        178:
+        dict(
+            name='trousers_kpt11',
+            id=178,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt7'),
+        179:
+        dict(
+            name='trousers_kpt12',
+            id=179,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt6'),
+        180:
+        dict(
+            name='trousers_kpt13',
+            id=180,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt5'),
+        181:
+        dict(
+            name='trousers_kpt14',
+            id=181,
+            color=colors['trousers'],
+            type='',
+            swap='trousers_kpt4'),
+        # skirt
+        182:
+        dict(
+            name='skirt_kpt1',
+            id=182,
+            color=colors['skirt'],
+            type='',
+            swap='skirt_kpt3'),
+        183:
+        dict(
+            name='skirt_kpt2', id=183, color=colors['skirt'], type='',
+            swap=''),
+        184:
+        dict(
+            name='skirt_kpt3',
+            id=184,
+            color=colors['skirt'],
+            type='',
+            swap='skirt_kpt1'),
+        185:
+        dict(
+            name='skirt_kpt4',
+            id=185,
+            color=colors['skirt'],
+            type='',
+            swap='skirt_kpt8'),
+        186:
+        dict(
+            name='skirt_kpt5',
+            id=186,
+            color=colors['skirt'],
+            type='',
+            swap='skirt_kpt7'),
+        187:
+        dict(
+            name='skirt_kpt6', id=187, color=colors['skirt'], type='',
+            swap=''),
+        188:
+        dict(
+            name='skirt_kpt7',
+            id=188,
+            color=colors['skirt'],
+            type='',
+            swap='skirt_kpt5'),
+        189:
+        dict(
+            name='skirt_kpt8',
+            id=189,
+            color=colors['skirt'],
+            type='',
+            swap='skirt_kpt4'),
+        # short_sleeved_dress
+        190:
+        dict(name='ssd_kpt1', id=190, color=colors['ssd'], type='', swap=''),
+        191:
+        dict(
+            name='ssd_kpt2',
+            id=191,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt6'),
+        192:
+        dict(
+            name='ssd_kpt3',
+            id=192,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt5'),
+        193:
+        dict(name='ssd_kpt4', id=193, color=colors['ssd'], type='', swap=''),
+        194:
+        dict(
+            name='ssd_kpt5',
+            id=194,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt3'),
+        195:
+        dict(
+            name='ssd_kpt6',
+            id=195,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt2'),
+        196:
+        dict(
+            name='ssd_kpt7',
+            id=196,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt29'),
+        197:
+        dict(
+            name='ssd_kpt8',
+            id=197,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt28'),
+        198:
+        dict(
+            name='ssd_kpt9',
+            id=198,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt27'),
+        199:
+        dict(
+            name='ssd_kpt10',
+            id=199,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt26'),
+        200:
+        dict(
+            name='ssd_kpt11',
+            id=200,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt25'),
+        201:
+        dict(
+            name='ssd_kpt12',
+            id=201,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt24'),
+        202:
+        dict(
+            name='ssd_kpt13',
+            id=202,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt23'),
+        203:
+        dict(
+            name='ssd_kpt14',
+            id=203,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt22'),
+        204:
+        dict(
+            name='ssd_kpt15',
+            id=204,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt21'),
+        205:
+        dict(
+            name='ssd_kpt16',
+            id=205,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt20'),
+        206:
+        dict(
+            name='ssd_kpt17',
+            id=206,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt19'),
+        207:
+        dict(name='ssd_kpt18', id=207, color=colors['ssd'], type='', swap=''),
+        208:
+        dict(
+            name='ssd_kpt19',
+            id=208,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt17'),
+        209:
+        dict(
+            name='ssd_kpt20',
+            id=209,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt16'),
+        210:
+        dict(
+            name='ssd_kpt21',
+            id=210,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt15'),
+        211:
+        dict(
+            name='ssd_kpt22',
+            id=211,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt14'),
+        212:
+        dict(
+            name='ssd_kpt23',
+            id=212,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt13'),
+        213:
+        dict(
+            name='ssd_kpt24',
+            id=213,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt12'),
+        214:
+        dict(
+            name='ssd_kpt25',
+            id=214,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt11'),
+        215:
+        dict(
+            name='ssd_kpt26',
+            id=215,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt10'),
+        216:
+        dict(
+            name='ssd_kpt27',
+            id=216,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt9'),
+        217:
+        dict(
+            name='ssd_kpt28',
+            id=217,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt8'),
+        218:
+        dict(
+            name='ssd_kpt29',
+            id=218,
+            color=colors['ssd'],
+            type='',
+            swap='ssd_kpt7'),
+        # long_sleeved_dress
+        219:
+        dict(name='lsd_kpt1', id=219, color=colors['lsd'], type='', swap=''),
+        220:
+        dict(
+            name='lsd_kpt2',
+            id=220,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt6'),
+        221:
+        dict(
+            name='lsd_kpt3',
+            id=221,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt5'),
+        222:
+        dict(name='lsd_kpt4', id=222, color=colors['lsd'], type='', swap=''),
+        223:
+        dict(
+            name='lsd_kpt5',
+            id=223,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt3'),
+        224:
+        dict(
+            name='lsd_kpt6',
+            id=224,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt2'),
+        225:
+        dict(
+            name='lsd_kpt7',
+            id=225,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt37'),
+        226:
+        dict(
+            name='lsd_kpt8',
+            id=226,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt36'),
+        227:
+        dict(
+            name='lsd_kpt9',
+            id=227,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt35'),
+        228:
+        dict(
+            name='lsd_kpt10',
+            id=228,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt34'),
+        229:
+        dict(
+            name='lsd_kpt11',
+            id=229,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt33'),
+        230:
+        dict(
+            name='lsd_kpt12',
+            id=230,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt32'),
+        231:
+        dict(
+            name='lsd_kpt13',
+            id=231,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt31'),
+        232:
+        dict(
+            name='lsd_kpt14',
+            id=232,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt30'),
+        233:
+        dict(
+            name='lsd_kpt15',
+            id=233,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt29'),
+        234:
+        dict(
+            name='lsd_kpt16',
+            id=234,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt28'),
+        235:
+        dict(
+            name='lsd_kpt17',
+            id=235,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt27'),
+        236:
+        dict(
+            name='lsd_kpt18',
+            id=236,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt26'),
+        237:
+        dict(
+            name='lsd_kpt19',
+            id=237,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt25'),
+        238:
+        dict(
+            name='lsd_kpt20',
+            id=238,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt24'),
+        239:
+        dict(
+            name='lsd_kpt21',
+            id=239,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt23'),
+        240:
+        dict(name='lsd_kpt22', id=240, color=colors['lsd'], type='', swap=''),
+        241:
+        dict(
+            name='lsd_kpt23',
+            id=241,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt21'),
+        242:
+        dict(
+            name='lsd_kpt24',
+            id=242,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt20'),
+        243:
+        dict(
+            name='lsd_kpt25',
+            id=243,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt19'),
+        244:
+        dict(
+            name='lsd_kpt26',
+            id=244,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt18'),
+        245:
+        dict(
+            name='lsd_kpt27',
+            id=245,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt17'),
+        246:
+        dict(
+            name='lsd_kpt28',
+            id=246,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt16'),
+        247:
+        dict(
+            name='lsd_kpt29',
+            id=247,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt15'),
+        248:
+        dict(
+            name='lsd_kpt30',
+            id=248,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt14'),
+        249:
+        dict(
+            name='lsd_kpt31',
+            id=249,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt13'),
+        250:
+        dict(
+            name='lsd_kpt32',
+            id=250,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt12'),
+        251:
+        dict(
+            name='lsd_kpt33',
+            id=251,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt11'),
+        252:
+        dict(
+            name='lsd_kpt34',
+            id=252,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt10'),
+        253:
+        dict(
+            name='lsd_kpt35',
+            id=253,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt9'),
+        254:
+        dict(
+            name='lsd_kpt36',
+            id=254,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt8'),
+        255:
+        dict(
+            name='lsd_kpt37',
+            id=255,
+            color=colors['lsd'],
+            type='',
+            swap='lsd_kpt7'),
+        # vest_dress
+        256:
+        dict(name='vd_kpt1', id=256, color=colors['vd'], type='', swap=''),
+        257:
+        dict(
+            name='vd_kpt2',
+            id=257,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt6'),
+        258:
+        dict(
+            name='vd_kpt3',
+            id=258,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt5'),
+        259:
+        dict(name='vd_kpt4', id=259, color=colors['vd'], type='', swap=''),
+        260:
+        dict(
+            name='vd_kpt5',
+            id=260,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt3'),
+        261:
+        dict(
+            name='vd_kpt6',
+            id=261,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt2'),
+        262:
+        dict(
+            name='vd_kpt7',
+            id=262,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt19'),
+        263:
+        dict(
+            name='vd_kpt8',
+            id=263,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt18'),
+        264:
+        dict(
+            name='vd_kpt9',
+            id=264,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt17'),
+        265:
+        dict(
+            name='vd_kpt10',
+            id=265,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt16'),
+        266:
+        dict(
+            name='vd_kpt11',
+            id=266,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt15'),
+        267:
+        dict(
+            name='vd_kpt12',
+            id=267,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt14'),
+        268:
+        dict(name='vd_kpt13', id=268, color=colors['vd'], type='', swap=''),
+        269:
+        dict(
+            name='vd_kpt14',
+            id=269,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt12'),
+        270:
+        dict(
+            name='vd_kpt15',
+            id=270,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt11'),
+        271:
+        dict(
+            name='vd_kpt16',
+            id=271,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt10'),
+        272:
+        dict(
+            name='vd_kpt17',
+            id=272,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt9'),
+        273:
+        dict(
+            name='vd_kpt18',
+            id=273,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt8'),
+        274:
+        dict(
+            name='vd_kpt19',
+            id=274,
+            color=colors['vd'],
+            type='',
+            swap='vd_kpt7'),
+        # sling_dress
+        275:
+        dict(name='sd_kpt1', id=275, color=colors['sd'], type='', swap=''),
+        276:
+        dict(
+            name='sd_kpt2',
+            id=276,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt6'),
+        277:
+        dict(
+            name='sd_kpt3',
+            id=277,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt5'),
+        278:
+        dict(name='sd_kpt4', id=278, color=colors['sd'], type='', swap=''),
+        279:
+        dict(
+            name='sd_kpt5',
+            id=279,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt3'),
+        280:
+        dict(
+            name='sd_kpt6',
+            id=280,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt2'),
+        281:
+        dict(
+            name='sd_kpt7',
+            id=281,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt19'),
+        282:
+        dict(
+            name='sd_kpt8',
+            id=282,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt18'),
+        283:
+        dict(
+            name='sd_kpt9',
+            id=283,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt17'),
+        284:
+        dict(
+            name='sd_kpt10',
+            id=284,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt16'),
+        285:
+        dict(
+            name='sd_kpt11',
+            id=285,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt15'),
+        286:
+        dict(
+            name='sd_kpt12',
+            id=286,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt14'),
+        287:
+        dict(name='sd_kpt13', id=287, color=colors['sd'], type='', swap=''),
+        288:
+        dict(
+            name='sd_kpt14',
+            id=288,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt12'),
+        289:
+        dict(
+            name='sd_kpt15',
+            id=289,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt11'),
+        290:
+        dict(
+            name='sd_kpt16',
+            id=290,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt10'),
+        291:
+        dict(
+            name='sd_kpt17',
+            id=291,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt9'),
+        292:
+        dict(
+            name='sd_kpt18',
+            id=292,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt8'),
+        293:
+        dict(
+            name='sd_kpt19',
+            id=293,
+            color=colors['sd'],
+            type='',
+            swap='sd_kpt7'),
+    },
+    skeleton_info={
+        # short_sleeved_shirt
+        0:
+        dict(link=('sss_kpt1', 'sss_kpt2'), id=0, color=[255, 128, 0]),
+        1:
+        dict(link=('sss_kpt2', 'sss_kpt7'), id=1, color=[255, 128, 0]),
+        2:
+        dict(link=('sss_kpt7', 'sss_kpt8'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('sss_kpt8', 'sss_kpt9'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('sss_kpt9', 'sss_kpt10'), id=4, color=[255, 128, 0]),
+        5:
+        dict(link=('sss_kpt10', 'sss_kpt11'), id=5, color=[255, 128, 0]),
+        6:
+        dict(link=('sss_kpt11', 'sss_kpt12'), id=6, color=[255, 128, 0]),
+        7:
+        dict(link=('sss_kpt12', 'sss_kpt13'), id=7, color=[255, 128, 0]),
+        8:
+        dict(link=('sss_kpt13', 'sss_kpt14'), id=8, color=[255, 128, 0]),
+        9:
+        dict(link=('sss_kpt14', 'sss_kpt15'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('sss_kpt15', 'sss_kpt16'), id=10, color=[255, 128, 0]),
+        11:
+        dict(link=('sss_kpt16', 'sss_kpt17'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('sss_kpt17', 'sss_kpt18'), id=12, color=[255, 128, 0]),
+        13:
+        dict(link=('sss_kpt18', 'sss_kpt19'), id=13, color=[255, 128, 0]),
+        14:
+        dict(link=('sss_kpt19', 'sss_kpt20'), id=14, color=[255, 128, 0]),
+        15:
+        dict(link=('sss_kpt20', 'sss_kpt21'), id=15, color=[255, 128, 0]),
+        16:
+        dict(link=('sss_kpt21', 'sss_kpt22'), id=16, color=[255, 128, 0]),
+        17:
+        dict(link=('sss_kpt22', 'sss_kpt23'), id=17, color=[255, 128, 0]),
+        18:
+        dict(link=('sss_kpt23', 'sss_kpt24'), id=18, color=[255, 128, 0]),
+        19:
+        dict(link=('sss_kpt24', 'sss_kpt25'), id=19, color=[255, 128, 0]),
+        20:
+        dict(link=('sss_kpt25', 'sss_kpt6'), id=20, color=[255, 128, 0]),
+        21:
+        dict(link=('sss_kpt6', 'sss_kpt1'), id=21, color=[255, 128, 0]),
+        22:
+        dict(link=('sss_kpt2', 'sss_kpt3'), id=22, color=[255, 128, 0]),
+        23:
+        dict(link=('sss_kpt3', 'sss_kpt4'), id=23, color=[255, 128, 0]),
+        24:
+        dict(link=('sss_kpt4', 'sss_kpt5'), id=24, color=[255, 128, 0]),
+        25:
+        dict(link=('sss_kpt5', 'sss_kpt6'), id=25, color=[255, 128, 0]),
+        # long_sleeve_shirt
+        26:
+        dict(link=('lss_kpt1', 'lss_kpt2'), id=26, color=[255, 0, 128]),
+        27:
+        dict(link=('lss_kpt2', 'lss_kpt7'), id=27, color=[255, 0, 128]),
+        28:
+        dict(link=('lss_kpt7', 'lss_kpt8'), id=28, color=[255, 0, 128]),
+        29:
+        dict(link=('lss_kpt8', 'lss_kpt9'), id=29, color=[255, 0, 128]),
+        30:
+        dict(link=('lss_kpt9', 'lss_kpt10'), id=30, color=[255, 0, 128]),
+        31:
+        dict(link=('lss_kpt10', 'lss_kpt11'), id=31, color=[255, 0, 128]),
+        32:
+        dict(link=('lss_kpt11', 'lss_kpt12'), id=32, color=[255, 0, 128]),
+        33:
+        dict(link=('lss_kpt12', 'lss_kpt13'), id=33, color=[255, 0, 128]),
+        34:
+        dict(link=('lss_kpt13', 'lss_kpt14'), id=34, color=[255, 0, 128]),
+        35:
+        dict(link=('lss_kpt14', 'lss_kpt15'), id=35, color=[255, 0, 128]),
+        36:
+        dict(link=('lss_kpt15', 'lss_kpt16'), id=36, color=[255, 0, 128]),
+        37:
+        dict(link=('lss_kpt16', 'lss_kpt17'), id=37, color=[255, 0, 128]),
+        38:
+        dict(link=('lss_kpt17', 'lss_kpt18'), id=38, color=[255, 0, 128]),
+        39:
+        dict(link=('lss_kpt18', 'lss_kpt19'), id=39, color=[255, 0, 128]),
+        40:
+        dict(link=('lss_kpt19', 'lss_kpt20'), id=40, color=[255, 0, 128]),
+        41:
+        dict(link=('lss_kpt20', 'lss_kpt21'), id=41, color=[255, 0, 128]),
+        42:
+        dict(link=('lss_kpt21', 'lss_kpt22'), id=42, color=[255, 0, 128]),
+        43:
+        dict(link=('lss_kpt22', 'lss_kpt23'), id=43, color=[255, 0, 128]),
+        44:
+        dict(link=('lss_kpt23', 'lss_kpt24'), id=44, color=[255, 0, 128]),
+        45:
+        dict(link=('lss_kpt24', 'lss_kpt25'), id=45, color=[255, 0, 128]),
+        46:
+        dict(link=('lss_kpt25', 'lss_kpt26'), id=46, color=[255, 0, 128]),
+        47:
+        dict(link=('lss_kpt26', 'lss_kpt27'), id=47, color=[255, 0, 128]),
+        48:
+        dict(link=('lss_kpt27', 'lss_kpt28'), id=48, color=[255, 0, 128]),
+        49:
+        dict(link=('lss_kpt28', 'lss_kpt29'), id=49, color=[255, 0, 128]),
+        50:
+        dict(link=('lss_kpt29', 'lss_kpt30'), id=50, color=[255, 0, 128]),
+        51:
+        dict(link=('lss_kpt30', 'lss_kpt31'), id=51, color=[255, 0, 128]),
+        52:
+        dict(link=('lss_kpt31', 'lss_kpt32'), id=52, color=[255, 0, 128]),
+        53:
+        dict(link=('lss_kpt32', 'lss_kpt33'), id=53, color=[255, 0, 128]),
+        54:
+        dict(link=('lss_kpt33', 'lss_kpt6'), id=54, color=[255, 0, 128]),
+        55:
+        dict(link=('lss_kpt6', 'lss_kpt5'), id=55, color=[255, 0, 128]),
+        56:
+        dict(link=('lss_kpt5', 'lss_kpt4'), id=56, color=[255, 0, 128]),
+        57:
+        dict(link=('lss_kpt4', 'lss_kpt3'), id=57, color=[255, 0, 128]),
+        58:
+        dict(link=('lss_kpt3', 'lss_kpt2'), id=58, color=[255, 0, 128]),
+        59:
+        dict(link=('lss_kpt6', 'lss_kpt1'), id=59, color=[255, 0, 128]),
+        # short_sleeved_outwear
+        60:
+        dict(link=('sso_kpt1', 'sso_kpt4'), id=60, color=[128, 0, 255]),
+        61:
+        dict(link=('sso_kpt4', 'sso_kpt7'), id=61, color=[128, 0, 255]),
+        62:
+        dict(link=('sso_kpt7', 'sso_kpt8'), id=62, color=[128, 0, 255]),
+        63:
+        dict(link=('sso_kpt8', 'sso_kpt9'), id=63, color=[128, 0, 255]),
+        64:
+        dict(link=('sso_kpt9', 'sso_kpt10'), id=64, color=[128, 0, 255]),
+        65:
+        dict(link=('sso_kpt10', 'sso_kpt11'), id=65, color=[128, 0, 255]),
+        66:
+        dict(link=('sso_kpt11', 'sso_kpt12'), id=66, color=[128, 0, 255]),
+        67:
+        dict(link=('sso_kpt12', 'sso_kpt13'), id=67, color=[128, 0, 255]),
+        68:
+        dict(link=('sso_kpt13', 'sso_kpt14'), id=68, color=[128, 0, 255]),
+        69:
+        dict(link=('sso_kpt14', 'sso_kpt15'), id=69, color=[128, 0, 255]),
+        70:
+        dict(link=('sso_kpt15', 'sso_kpt16'), id=70, color=[128, 0, 255]),
+        71:
+        dict(link=('sso_kpt16', 'sso_kpt31'), id=71, color=[128, 0, 255]),
+        72:
+        dict(link=('sso_kpt31', 'sso_kpt30'), id=72, color=[128, 0, 255]),
+        73:
+        dict(link=('sso_kpt30', 'sso_kpt2'), id=73, color=[128, 0, 255]),
+        74:
+        dict(link=('sso_kpt2', 'sso_kpt3'), id=74, color=[128, 0, 255]),
+        75:
+        dict(link=('sso_kpt3', 'sso_kpt4'), id=75, color=[128, 0, 255]),
+        76:
+        dict(link=('sso_kpt1', 'sso_kpt6'), id=76, color=[128, 0, 255]),
+        77:
+        dict(link=('sso_kpt6', 'sso_kpt25'), id=77, color=[128, 0, 255]),
+        78:
+        dict(link=('sso_kpt25', 'sso_kpt24'), id=78, color=[128, 0, 255]),
+        79:
+        dict(link=('sso_kpt24', 'sso_kpt23'), id=79, color=[128, 0, 255]),
+        80:
+        dict(link=('sso_kpt23', 'sso_kpt22'), id=80, color=[128, 0, 255]),
+        81:
+        dict(link=('sso_kpt22', 'sso_kpt21'), id=81, color=[128, 0, 255]),
+        82:
+        dict(link=('sso_kpt21', 'sso_kpt20'), id=82, color=[128, 0, 255]),
+        83:
+        dict(link=('sso_kpt20', 'sso_kpt19'), id=83, color=[128, 0, 255]),
+        84:
+        dict(link=('sso_kpt19', 'sso_kpt18'), id=84, color=[128, 0, 255]),
+        85:
+        dict(link=('sso_kpt18', 'sso_kpt17'), id=85, color=[128, 0, 255]),
+        86:
+        dict(link=('sso_kpt17', 'sso_kpt29'), id=86, color=[128, 0, 255]),
+        87:
+        dict(link=('sso_kpt29', 'sso_kpt28'), id=87, color=[128, 0, 255]),
+        88:
+        dict(link=('sso_kpt28', 'sso_kpt27'), id=88, color=[128, 0, 255]),
+        89:
+        dict(link=('sso_kpt27', 'sso_kpt26'), id=89, color=[128, 0, 255]),
+        90:
+        dict(link=('sso_kpt26', 'sso_kpt5'), id=90, color=[128, 0, 255]),
+        91:
+        dict(link=('sso_kpt5', 'sso_kpt6'), id=91, color=[128, 0, 255]),
+        # long_sleeved_outwear
+        92:
+        dict(link=('lso_kpt1', 'lso_kpt2'), id=92, color=[0, 128, 255]),
+        93:
+        dict(link=('lso_kpt2', 'lso_kpt7'), id=93, color=[0, 128, 255]),
+        94:
+        dict(link=('lso_kpt7', 'lso_kpt8'), id=94, color=[0, 128, 255]),
+        95:
+        dict(link=('lso_kpt8', 'lso_kpt9'), id=95, color=[0, 128, 255]),
+        96:
+        dict(link=('lso_kpt9', 'lso_kpt10'), id=96, color=[0, 128, 255]),
+        97:
+        dict(link=('lso_kpt10', 'lso_kpt11'), id=97, color=[0, 128, 255]),
+        98:
+        dict(link=('lso_kpt11', 'lso_kpt12'), id=98, color=[0, 128, 255]),
+        99:
+        dict(link=('lso_kpt12', 'lso_kpt13'), id=99, color=[0, 128, 255]),
+        100:
+        dict(link=('lso_kpt13', 'lso_kpt14'), id=100, color=[0, 128, 255]),
+        101:
+        dict(link=('lso_kpt14', 'lso_kpt15'), id=101, color=[0, 128, 255]),
+        102:
+        dict(link=('lso_kpt15', 'lso_kpt16'), id=102, color=[0, 128, 255]),
+        103:
+        dict(link=('lso_kpt16', 'lso_kpt17'), id=103, color=[0, 128, 255]),
+        104:
+        dict(link=('lso_kpt17', 'lso_kpt18'), id=104, color=[0, 128, 255]),
+        105:
+        dict(link=('lso_kpt18', 'lso_kpt19'), id=105, color=[0, 128, 255]),
+        106:
+        dict(link=('lso_kpt19', 'lso_kpt20'), id=106, color=[0, 128, 255]),
+        107:
+        dict(link=('lso_kpt20', 'lso_kpt39'), id=107, color=[0, 128, 255]),
+        108:
+        dict(link=('lso_kpt39', 'lso_kpt38'), id=108, color=[0, 128, 255]),
+        109:
+        dict(link=('lso_kpt38', 'lso_kpt4'), id=109, color=[0, 128, 255]),
+        110:
+        dict(link=('lso_kpt4', 'lso_kpt3'), id=110, color=[0, 128, 255]),
+        111:
+        dict(link=('lso_kpt3', 'lso_kpt2'), id=111, color=[0, 128, 255]),
+        112:
+        dict(link=('lso_kpt1', 'lso_kpt6'), id=112, color=[0, 128, 255]),
+        113:
+        dict(link=('lso_kpt6', 'lso_kpt33'), id=113, color=[0, 128, 255]),
+        114:
+        dict(link=('lso_kpt33', 'lso_kpt32'), id=114, color=[0, 128, 255]),
+        115:
+        dict(link=('lso_kpt32', 'lso_kpt31'), id=115, color=[0, 128, 255]),
+        116:
+        dict(link=('lso_kpt31', 'lso_kpt30'), id=116, color=[0, 128, 255]),
+        117:
+        dict(link=('lso_kpt30', 'lso_kpt29'), id=117, color=[0, 128, 255]),
+        118:
+        dict(link=('lso_kpt29', 'lso_kpt28'), id=118, color=[0, 128, 255]),
+        119:
+        dict(link=('lso_kpt28', 'lso_kpt27'), id=119, color=[0, 128, 255]),
+        120:
+        dict(link=('lso_kpt27', 'lso_kpt26'), id=120, color=[0, 128, 255]),
+        121:
+        dict(link=('lso_kpt26', 'lso_kpt25'), id=121, color=[0, 128, 255]),
+        122:
+        dict(link=('lso_kpt25', 'lso_kpt24'), id=122, color=[0, 128, 255]),
+        123:
+        dict(link=('lso_kpt24', 'lso_kpt23'), id=123, color=[0, 128, 255]),
+        124:
+        dict(link=('lso_kpt23', 'lso_kpt22'), id=124, color=[0, 128, 255]),
+        125:
+        dict(link=('lso_kpt22', 'lso_kpt21'), id=125, color=[0, 128, 255]),
+        126:
+        dict(link=('lso_kpt21', 'lso_kpt37'), id=126, color=[0, 128, 255]),
+        127:
+        dict(link=('lso_kpt37', 'lso_kpt36'), id=127, color=[0, 128, 255]),
+        128:
+        dict(link=('lso_kpt36', 'lso_kpt35'), id=128, color=[0, 128, 255]),
+        129:
+        dict(link=('lso_kpt35', 'lso_kpt34'), id=129, color=[0, 128, 255]),
+        130:
+        dict(link=('lso_kpt34', 'lso_kpt5'), id=130, color=[0, 128, 255]),
+        131:
+        dict(link=('lso_kpt5', 'lso_kpt6'), id=131, color=[0, 128, 255]),
+        # vest
+        132:
+        dict(link=('vest_kpt1', 'vest_kpt2'), id=132, color=[0, 128, 128]),
+        133:
+        dict(link=('vest_kpt2', 'vest_kpt7'), id=133, color=[0, 128, 128]),
+        134:
+        dict(link=('vest_kpt7', 'vest_kpt8'), id=134, color=[0, 128, 128]),
+        135:
+        dict(link=('vest_kpt8', 'vest_kpt9'), id=135, color=[0, 128, 128]),
+        136:
+        dict(link=('vest_kpt9', 'vest_kpt10'), id=136, color=[0, 128, 128]),
+        137:
+        dict(link=('vest_kpt10', 'vest_kpt11'), id=137, color=[0, 128, 128]),
+        138:
+        dict(link=('vest_kpt11', 'vest_kpt12'), id=138, color=[0, 128, 128]),
+        139:
+        dict(link=('vest_kpt12', 'vest_kpt13'), id=139, color=[0, 128, 128]),
+        140:
+        dict(link=('vest_kpt13', 'vest_kpt14'), id=140, color=[0, 128, 128]),
+        141:
+        dict(link=('vest_kpt14', 'vest_kpt15'), id=141, color=[0, 128, 128]),
+        142:
+        dict(link=('vest_kpt15', 'vest_kpt6'), id=142, color=[0, 128, 128]),
+        143:
+        dict(link=('vest_kpt6', 'vest_kpt1'), id=143, color=[0, 128, 128]),
+        144:
+        dict(link=('vest_kpt2', 'vest_kpt3'), id=144, color=[0, 128, 128]),
+        145:
+        dict(link=('vest_kpt3', 'vest_kpt4'), id=145, color=[0, 128, 128]),
+        146:
+        dict(link=('vest_kpt4', 'vest_kpt5'), id=146, color=[0, 128, 128]),
+        147:
+        dict(link=('vest_kpt5', 'vest_kpt6'), id=147, color=[0, 128, 128]),
+        # sling
+        148:
+        dict(link=('sling_kpt1', 'sling_kpt2'), id=148, color=[0, 0, 128]),
+        149:
+        dict(link=('sling_kpt2', 'sling_kpt8'), id=149, color=[0, 0, 128]),
+        150:
+        dict(link=('sling_kpt8', 'sling_kpt9'), id=150, color=[0, 0, 128]),
+        151:
+        dict(link=('sling_kpt9', 'sling_kpt10'), id=151, color=[0, 0, 128]),
+        152:
+        dict(link=('sling_kpt10', 'sling_kpt11'), id=152, color=[0, 0, 128]),
+        153:
+        dict(link=('sling_kpt11', 'sling_kpt12'), id=153, color=[0, 0, 128]),
+        154:
+        dict(link=('sling_kpt12', 'sling_kpt13'), id=154, color=[0, 0, 128]),
+        155:
+        dict(link=('sling_kpt13', 'sling_kpt14'), id=155, color=[0, 0, 128]),
+        156:
+        dict(link=('sling_kpt14', 'sling_kpt6'), id=156, color=[0, 0, 128]),
+        157:
+        dict(link=('sling_kpt2', 'sling_kpt7'), id=157, color=[0, 0, 128]),
+        158:
+        dict(link=('sling_kpt6', 'sling_kpt15'), id=158, color=[0, 0, 128]),
+        159:
+        dict(link=('sling_kpt2', 'sling_kpt3'), id=159, color=[0, 0, 128]),
+        160:
+        dict(link=('sling_kpt3', 'sling_kpt4'), id=160, color=[0, 0, 128]),
+        161:
+        dict(link=('sling_kpt4', 'sling_kpt5'), id=161, color=[0, 0, 128]),
+        162:
+        dict(link=('sling_kpt5', 'sling_kpt6'), id=162, color=[0, 0, 128]),
+        163:
+        dict(link=('sling_kpt1', 'sling_kpt6'), id=163, color=[0, 0, 128]),
+        # shorts
+        164:
+        dict(
+            link=('shorts_kpt1', 'shorts_kpt4'), id=164, color=[128, 128,
+                                                                128]),
+        165:
+        dict(
+            link=('shorts_kpt4', 'shorts_kpt5'), id=165, color=[128, 128,
+                                                                128]),
+        166:
+        dict(
+            link=('shorts_kpt5', 'shorts_kpt6'), id=166, color=[128, 128,
+                                                                128]),
+        167:
+        dict(
+            link=('shorts_kpt6', 'shorts_kpt7'), id=167, color=[128, 128,
+                                                                128]),
+        168:
+        dict(
+            link=('shorts_kpt7', 'shorts_kpt8'), id=168, color=[128, 128,
+                                                                128]),
+        169:
+        dict(
+            link=('shorts_kpt8', 'shorts_kpt9'), id=169, color=[128, 128,
+                                                                128]),
+        170:
+        dict(
+            link=('shorts_kpt9', 'shorts_kpt10'),
+            id=170,
+            color=[128, 128, 128]),
+        171:
+        dict(
+            link=('shorts_kpt10', 'shorts_kpt3'),
+            id=171,
+            color=[128, 128, 128]),
+        172:
+        dict(
+            link=('shorts_kpt3', 'shorts_kpt2'), id=172, color=[128, 128,
+                                                                128]),
+        173:
+        dict(
+            link=('shorts_kpt2', 'shorts_kpt1'), id=173, color=[128, 128,
+                                                                128]),
+        # trousers
+        174:
+        dict(
+            link=('trousers_kpt1', 'trousers_kpt4'),
+            id=174,
+            color=[128, 0, 128]),
+        175:
+        dict(
+            link=('trousers_kpt4', 'trousers_kpt5'),
+            id=175,
+            color=[128, 0, 128]),
+        176:
+        dict(
+            link=('trousers_kpt5', 'trousers_kpt6'),
+            id=176,
+            color=[128, 0, 128]),
+        177:
+        dict(
+            link=('trousers_kpt6', 'trousers_kpt7'),
+            id=177,
+            color=[128, 0, 128]),
+        178:
+        dict(
+            link=('trousers_kpt7', 'trousers_kpt8'),
+            id=178,
+            color=[128, 0, 128]),
+        179:
+        dict(
+            link=('trousers_kpt8', 'trousers_kpt9'),
+            id=179,
+            color=[128, 0, 128]),
+        180:
+        dict(
+            link=('trousers_kpt9', 'trousers_kpt10'),
+            id=180,
+            color=[128, 0, 128]),
+        181:
+        dict(
+            link=('trousers_kpt10', 'trousers_kpt11'),
+            id=181,
+            color=[128, 0, 128]),
+        182:
+        dict(
+            link=('trousers_kpt11', 'trousers_kpt12'),
+            id=182,
+            color=[128, 0, 128]),
+        183:
+        dict(
+            link=('trousers_kpt12', 'trousers_kpt13'),
+            id=183,
+            color=[128, 0, 128]),
+        184:
+        dict(
+            link=('trousers_kpt13', 'trousers_kpt14'),
+            id=184,
+            color=[128, 0, 128]),
+        185:
+        dict(
+            link=('trousers_kpt14', 'trousers_kpt3'),
+            id=185,
+            color=[128, 0, 128]),
+        186:
+        dict(
+            link=('trousers_kpt3', 'trousers_kpt2'),
+            id=186,
+            color=[128, 0, 128]),
+        187:
+        dict(
+            link=('trousers_kpt2', 'trousers_kpt1'),
+            id=187,
+            color=[128, 0, 128]),
+        # skirt
+        188:
+        dict(link=('skirt_kpt1', 'skirt_kpt4'), id=188, color=[64, 128, 128]),
+        189:
+        dict(link=('skirt_kpt4', 'skirt_kpt5'), id=189, color=[64, 128, 128]),
+        190:
+        dict(link=('skirt_kpt5', 'skirt_kpt6'), id=190, color=[64, 128, 128]),
+        191:
+        dict(link=('skirt_kpt6', 'skirt_kpt7'), id=191, color=[64, 128, 128]),
+        192:
+        dict(link=('skirt_kpt7', 'skirt_kpt8'), id=192, color=[64, 128, 128]),
+        193:
+        dict(link=('skirt_kpt8', 'skirt_kpt3'), id=193, color=[64, 128, 128]),
+        194:
+        dict(link=('skirt_kpt3', 'skirt_kpt2'), id=194, color=[64, 128, 128]),
+        195:
+        dict(link=('skirt_kpt2', 'skirt_kpt1'), id=195, color=[64, 128, 128]),
+        # short_sleeved_dress
+        196:
+        dict(link=('ssd_kpt1', 'ssd_kpt2'), id=196, color=[64, 64, 128]),
+        197:
+        dict(link=('ssd_kpt2', 'ssd_kpt7'), id=197, color=[64, 64, 128]),
+        198:
+        dict(link=('ssd_kpt7', 'ssd_kpt8'), id=198, color=[64, 64, 128]),
+        199:
+        dict(link=('ssd_kpt8', 'ssd_kpt9'), id=199, color=[64, 64, 128]),
+        200:
+        dict(link=('ssd_kpt9', 'ssd_kpt10'), id=200, color=[64, 64, 128]),
+        201:
+        dict(link=('ssd_kpt10', 'ssd_kpt11'), id=201, color=[64, 64, 128]),
+        202:
+        dict(link=('ssd_kpt11', 'ssd_kpt12'), id=202, color=[64, 64, 128]),
+        203:
+        dict(link=('ssd_kpt12', 'ssd_kpt13'), id=203, color=[64, 64, 128]),
+        204:
+        dict(link=('ssd_kpt13', 'ssd_kpt14'), id=204, color=[64, 64, 128]),
+        205:
+        dict(link=('ssd_kpt14', 'ssd_kpt15'), id=205, color=[64, 64, 128]),
+        206:
+        dict(link=('ssd_kpt15', 'ssd_kpt16'), id=206, color=[64, 64, 128]),
+        207:
+        dict(link=('ssd_kpt16', 'ssd_kpt17'), id=207, color=[64, 64, 128]),
+        208:
+        dict(link=('ssd_kpt17', 'ssd_kpt18'), id=208, color=[64, 64, 128]),
+        209:
+        dict(link=('ssd_kpt18', 'ssd_kpt19'), id=209, color=[64, 64, 128]),
+        210:
+        dict(link=('ssd_kpt19', 'ssd_kpt20'), id=210, color=[64, 64, 128]),
+        211:
+        dict(link=('ssd_kpt20', 'ssd_kpt21'), id=211, color=[64, 64, 128]),
+        212:
+        dict(link=('ssd_kpt21', 'ssd_kpt22'), id=212, color=[64, 64, 128]),
+        213:
+        dict(link=('ssd_kpt22', 'ssd_kpt23'), id=213, color=[64, 64, 128]),
+        214:
+        dict(link=('ssd_kpt23', 'ssd_kpt24'), id=214, color=[64, 64, 128]),
+        215:
+        dict(link=('ssd_kpt24', 'ssd_kpt25'), id=215, color=[64, 64, 128]),
+        216:
+        dict(link=('ssd_kpt25', 'ssd_kpt26'), id=216, color=[64, 64, 128]),
+        217:
+        dict(link=('ssd_kpt26', 'ssd_kpt27'), id=217, color=[64, 64, 128]),
+        218:
+        dict(link=('ssd_kpt27', 'ssd_kpt28'), id=218, color=[64, 64, 128]),
+        219:
+        dict(link=('ssd_kpt28', 'ssd_kpt29'), id=219, color=[64, 64, 128]),
+        220:
+        dict(link=('ssd_kpt29', 'ssd_kpt6'), id=220, color=[64, 64, 128]),
+        221:
+        dict(link=('ssd_kpt6', 'ssd_kpt5'), id=221, color=[64, 64, 128]),
+        222:
+        dict(link=('ssd_kpt5', 'ssd_kpt4'), id=222, color=[64, 64, 128]),
+        223:
+        dict(link=('ssd_kpt4', 'ssd_kpt3'), id=223, color=[64, 64, 128]),
+        224:
+        dict(link=('ssd_kpt3', 'ssd_kpt2'), id=224, color=[64, 64, 128]),
+        225:
+        dict(link=('ssd_kpt6', 'ssd_kpt1'), id=225, color=[64, 64, 128]),
+        # long_sleeved_dress
+        226:
+        dict(link=('lsd_kpt1', 'lsd_kpt2'), id=226, color=[128, 64, 0]),
+        227:
+        dict(link=('lsd_kpt2', 'lsd_kpt7'), id=228, color=[128, 64, 0]),
+        228:
+        dict(link=('lsd_kpt7', 'lsd_kpt8'), id=228, color=[128, 64, 0]),
+        229:
+        dict(link=('lsd_kpt8', 'lsd_kpt9'), id=229, color=[128, 64, 0]),
+        230:
+        dict(link=('lsd_kpt9', 'lsd_kpt10'), id=230, color=[128, 64, 0]),
+        231:
+        dict(link=('lsd_kpt10', 'lsd_kpt11'), id=231, color=[128, 64, 0]),
+        232:
+        dict(link=('lsd_kpt11', 'lsd_kpt12'), id=232, color=[128, 64, 0]),
+        233:
+        dict(link=('lsd_kpt12', 'lsd_kpt13'), id=233, color=[128, 64, 0]),
+        234:
+        dict(link=('lsd_kpt13', 'lsd_kpt14'), id=234, color=[128, 64, 0]),
+        235:
+        dict(link=('lsd_kpt14', 'lsd_kpt15'), id=235, color=[128, 64, 0]),
+        236:
+        dict(link=('lsd_kpt15', 'lsd_kpt16'), id=236, color=[128, 64, 0]),
+        237:
+        dict(link=('lsd_kpt16', 'lsd_kpt17'), id=237, color=[128, 64, 0]),
+        238:
+        dict(link=('lsd_kpt17', 'lsd_kpt18'), id=238, color=[128, 64, 0]),
+        239:
+        dict(link=('lsd_kpt18', 'lsd_kpt19'), id=239, color=[128, 64, 0]),
+        240:
+        dict(link=('lsd_kpt19', 'lsd_kpt20'), id=240, color=[128, 64, 0]),
+        241:
+        dict(link=('lsd_kpt20', 'lsd_kpt21'), id=241, color=[128, 64, 0]),
+        242:
+        dict(link=('lsd_kpt21', 'lsd_kpt22'), id=242, color=[128, 64, 0]),
+        243:
+        dict(link=('lsd_kpt22', 'lsd_kpt23'), id=243, color=[128, 64, 0]),
+        244:
+        dict(link=('lsd_kpt23', 'lsd_kpt24'), id=244, color=[128, 64, 0]),
+        245:
+        dict(link=('lsd_kpt24', 'lsd_kpt25'), id=245, color=[128, 64, 0]),
+        246:
+        dict(link=('lsd_kpt25', 'lsd_kpt26'), id=246, color=[128, 64, 0]),
+        247:
+        dict(link=('lsd_kpt26', 'lsd_kpt27'), id=247, color=[128, 64, 0]),
+        248:
+        dict(link=('lsd_kpt27', 'lsd_kpt28'), id=248, color=[128, 64, 0]),
+        249:
+        dict(link=('lsd_kpt28', 'lsd_kpt29'), id=249, color=[128, 64, 0]),
+        250:
+        dict(link=('lsd_kpt29', 'lsd_kpt30'), id=250, color=[128, 64, 0]),
+        251:
+        dict(link=('lsd_kpt30', 'lsd_kpt31'), id=251, color=[128, 64, 0]),
+        252:
+        dict(link=('lsd_kpt31', 'lsd_kpt32'), id=252, color=[128, 64, 0]),
+        253:
+        dict(link=('lsd_kpt32', 'lsd_kpt33'), id=253, color=[128, 64, 0]),
+        254:
+        dict(link=('lsd_kpt33', 'lsd_kpt34'), id=254, color=[128, 64, 0]),
+        255:
+        dict(link=('lsd_kpt34', 'lsd_kpt35'), id=255, color=[128, 64, 0]),
+        256:
+        dict(link=('lsd_kpt35', 'lsd_kpt36'), id=256, color=[128, 64, 0]),
+        257:
+        dict(link=('lsd_kpt36', 'lsd_kpt37'), id=257, color=[128, 64, 0]),
+        258:
+        dict(link=('lsd_kpt37', 'lsd_kpt6'), id=258, color=[128, 64, 0]),
+        259:
+        dict(link=('lsd_kpt6', 'lsd_kpt5'), id=259, color=[128, 64, 0]),
+        260:
+        dict(link=('lsd_kpt5', 'lsd_kpt4'), id=260, color=[128, 64, 0]),
+        261:
+        dict(link=('lsd_kpt4', 'lsd_kpt3'), id=261, color=[128, 64, 0]),
+        262:
+        dict(link=('lsd_kpt3', 'lsd_kpt2'), id=262, color=[128, 64, 0]),
+        263:
+        dict(link=('lsd_kpt6', 'lsd_kpt1'), id=263, color=[128, 64, 0]),
+        # vest_dress
+        264:
+        dict(link=('vd_kpt1', 'vd_kpt2'), id=264, color=[128, 64, 255]),
+        265:
+        dict(link=('vd_kpt2', 'vd_kpt7'), id=265, color=[128, 64, 255]),
+        266:
+        dict(link=('vd_kpt7', 'vd_kpt8'), id=266, color=[128, 64, 255]),
+        267:
+        dict(link=('vd_kpt8', 'vd_kpt9'), id=267, color=[128, 64, 255]),
+        268:
+        dict(link=('vd_kpt9', 'vd_kpt10'), id=268, color=[128, 64, 255]),
+        269:
+        dict(link=('vd_kpt10', 'vd_kpt11'), id=269, color=[128, 64, 255]),
+        270:
+        dict(link=('vd_kpt11', 'vd_kpt12'), id=270, color=[128, 64, 255]),
+        271:
+        dict(link=('vd_kpt12', 'vd_kpt13'), id=271, color=[128, 64, 255]),
+        272:
+        dict(link=('vd_kpt13', 'vd_kpt14'), id=272, color=[128, 64, 255]),
+        273:
+        dict(link=('vd_kpt14', 'vd_kpt15'), id=273, color=[128, 64, 255]),
+        274:
+        dict(link=('vd_kpt15', 'vd_kpt16'), id=274, color=[128, 64, 255]),
+        275:
+        dict(link=('vd_kpt16', 'vd_kpt17'), id=275, color=[128, 64, 255]),
+        276:
+        dict(link=('vd_kpt17', 'vd_kpt18'), id=276, color=[128, 64, 255]),
+        277:
+        dict(link=('vd_kpt18', 'vd_kpt19'), id=277, color=[128, 64, 255]),
+        278:
+        dict(link=('vd_kpt19', 'vd_kpt6'), id=278, color=[128, 64, 255]),
+        279:
+        dict(link=('vd_kpt6', 'vd_kpt5'), id=279, color=[128, 64, 255]),
+        280:
+        dict(link=('vd_kpt5', 'vd_kpt4'), id=280, color=[128, 64, 255]),
+        281:
+        dict(link=('vd_kpt4', 'vd_kpt3'), id=281, color=[128, 64, 255]),
+        282:
+        dict(link=('vd_kpt3', 'vd_kpt2'), id=282, color=[128, 64, 255]),
+        283:
+        dict(link=('vd_kpt6', 'vd_kpt1'), id=283, color=[128, 64, 255]),
+        # sling_dress
+        284:
+        dict(link=('sd_kpt1', 'sd_kpt2'), id=284, color=[128, 64, 0]),
+        285:
+        dict(link=('sd_kpt2', 'sd_kpt8'), id=285, color=[128, 64, 0]),
+        286:
+        dict(link=('sd_kpt8', 'sd_kpt9'), id=286, color=[128, 64, 0]),
+        287:
+        dict(link=('sd_kpt9', 'sd_kpt10'), id=287, color=[128, 64, 0]),
+        288:
+        dict(link=('sd_kpt10', 'sd_kpt11'), id=288, color=[128, 64, 0]),
+        289:
+        dict(link=('sd_kpt11', 'sd_kpt12'), id=289, color=[128, 64, 0]),
+        290:
+        dict(link=('sd_kpt12', 'sd_kpt13'), id=290, color=[128, 64, 0]),
+        291:
+        dict(link=('sd_kpt13', 'sd_kpt14'), id=291, color=[128, 64, 0]),
+        292:
+        dict(link=('sd_kpt14', 'sd_kpt15'), id=292, color=[128, 64, 0]),
+        293:
+        dict(link=('sd_kpt15', 'sd_kpt16'), id=293, color=[128, 64, 0]),
+        294:
+        dict(link=('sd_kpt16', 'sd_kpt17'), id=294, color=[128, 64, 0]),
+        295:
+        dict(link=('sd_kpt17', 'sd_kpt18'), id=295, color=[128, 64, 0]),
+        296:
+        dict(link=('sd_kpt18', 'sd_kpt6'), id=296, color=[128, 64, 0]),
+        297:
+        dict(link=('sd_kpt6', 'sd_kpt5'), id=297, color=[128, 64, 0]),
+        298:
+        dict(link=('sd_kpt5', 'sd_kpt4'), id=298, color=[128, 64, 0]),
+        299:
+        dict(link=('sd_kpt4', 'sd_kpt3'), id=299, color=[128, 64, 0]),
+        300:
+        dict(link=('sd_kpt3', 'sd_kpt2'), id=300, color=[128, 64, 0]),
+        301:
+        dict(link=('sd_kpt2', 'sd_kpt7'), id=301, color=[128, 64, 0]),
+        302:
+        dict(link=('sd_kpt6', 'sd_kpt19'), id=302, color=[128, 64, 0]),
+        303:
+        dict(link=('sd_kpt6', 'sd_kpt1'), id=303, color=[128, 64, 0]),
+    },
+    joint_weights=[1.] * 294,
+    sigmas=[])

configs/_base_/datasets/deepfashion_full.py

@@ -0,0 +1,80 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='deepfashion_full',
+    paper_info=dict(
+        author='Liu, Ziwei and Luo, Ping and Qiu, Shi '
+        'and Wang, Xiaogang and Tang, Xiaoou',
+        title='DeepFashion: Powering Robust Clothes Recognition '
+        'and Retrieval with Rich Annotations',
+        container='Proceedings of IEEE Conference on Computer '
+        'Vision and Pattern Recognition (CVPR)',
+        year='2016',
+        homepage='http://mmlab.ie.cuhk.edu.hk/projects/'
+        'DeepFashion/LandmarkDetection.html',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='left collar',
+            id=0,
+            color=[255, 255, 255],
+            type='',
+            swap='right collar'),
+        1:
+        dict(
+            name='right collar',
+            id=1,
+            color=[255, 255, 255],
+            type='',
+            swap='left collar'),
+        2:
+        dict(
+            name='left sleeve',
+            id=2,
+            color=[255, 255, 255],
+            type='',
+            swap='right sleeve'),
+        3:
+        dict(
+            name='right sleeve',
+            id=3,
+            color=[255, 255, 255],
+            type='',
+            swap='left sleeve'),
+        4:
+        dict(
+            name='left waistline',
+            id=0,
+            color=[255, 255, 255],
+            type='',
+            swap='right waistline'),
+        5:
+        dict(
+            name='right waistline',
+            id=1,
+            color=[255, 255, 255],
+            type='',
+            swap='left waistline'),
+        6:
+        dict(
+            name='left hem',
+            id=2,
+            color=[255, 255, 255],
+            type='',
+            swap='right hem'),
+        7:
+        dict(
+            name='right hem',
+            id=3,
+            color=[255, 255, 255],
+            type='',
+            swap='left hem'),
+    },
+    skeleton_info={},
+    joint_weights=[1.] * 8,
+    sigmas=[])

configs/_base_/datasets/deepfashion_lower.py

@@ -0,0 +1,52 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='deepfashion_lower',
+    paper_info=dict(
+        author='Liu, Ziwei and Luo, Ping and Qiu, Shi '
+        'and Wang, Xiaogang and Tang, Xiaoou',
+        title='DeepFashion: Powering Robust Clothes Recognition '
+        'and Retrieval with Rich Annotations',
+        container='Proceedings of IEEE Conference on Computer '
+        'Vision and Pattern Recognition (CVPR)',
+        year='2016',
+        homepage='http://mmlab.ie.cuhk.edu.hk/projects/'
+        'DeepFashion/LandmarkDetection.html',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='left waistline',
+            id=0,
+            color=[255, 255, 255],
+            type='',
+            swap='right waistline'),
+        1:
+        dict(
+            name='right waistline',
+            id=1,
+            color=[255, 255, 255],
+            type='',
+            swap='left waistline'),
+        2:
+        dict(
+            name='left hem',
+            id=2,
+            color=[255, 255, 255],
+            type='',
+            swap='right hem'),
+        3:
+        dict(
+            name='right hem',
+            id=3,
+            color=[255, 255, 255],
+            type='',
+            swap='left hem'),
+    },
+    skeleton_info={},
+    joint_weights=[1.] * 4,
+    sigmas=[])

configs/_base_/datasets/deepfashion_upper.py

@@ -0,0 +1,66 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='deepfashion_upper',
+    paper_info=dict(
+        author='Liu, Ziwei and Luo, Ping and Qiu, Shi '
+        'and Wang, Xiaogang and Tang, Xiaoou',
+        title='DeepFashion: Powering Robust Clothes Recognition '
+        'and Retrieval with Rich Annotations',
+        container='Proceedings of IEEE Conference on Computer '
+        'Vision and Pattern Recognition (CVPR)',
+        year='2016',
+        homepage='http://mmlab.ie.cuhk.edu.hk/projects/'
+        'DeepFashion/LandmarkDetection.html',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='left collar',
+            id=0,
+            color=[255, 255, 255],
+            type='',
+            swap='right collar'),
+        1:
+        dict(
+            name='right collar',
+            id=1,
+            color=[255, 255, 255],
+            type='',
+            swap='left collar'),
+        2:
+        dict(
+            name='left sleeve',
+            id=2,
+            color=[255, 255, 255],
+            type='',
+            swap='right sleeve'),
+        3:
+        dict(
+            name='right sleeve',
+            id=3,
+            color=[255, 255, 255],
+            type='',
+            swap='left sleeve'),
+        4:
+        dict(
+            name='left hem',
+            id=4,
+            color=[255, 255, 255],
+            type='',
+            swap='right hem'),
+        5:
+        dict(
+            name='right hem',
+            id=5,
+            color=[255, 255, 255],
+            type='',
+            swap='left hem'),
+    },
+    skeleton_info={},
+    joint_weights=[1.] * 6,
+    sigmas=[])

configs/_base_/datasets/fly.py

@@ -0,0 +1,243 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='fly',
+    paper_info=dict(
+        author='Pereira, Talmo D and Aldarondo, Diego E and '
+        'Willmore, Lindsay and Kislin, Mikhail and '
+        'Wang, Samuel S-H and Murthy, Mala and Shaevitz, Joshua W',
+        title='Fast animal pose estimation using deep neural networks',
+        container='Nature methods',
+        year='2019',
+        homepage='https://github.com/jgraving/DeepPoseKit-Data',
+    ),
+    keypoint_info={
+        0:
+        dict(name='head', id=0, color=[255, 255, 255], type='', swap=''),
+        1:
+        dict(name='eyeL', id=1, color=[255, 255, 255], type='', swap='eyeR'),
+        2:
+        dict(name='eyeR', id=2, color=[255, 255, 255], type='', swap='eyeL'),
+        3:
+        dict(name='neck', id=3, color=[255, 255, 255], type='', swap=''),
+        4:
+        dict(name='thorax', id=4, color=[255, 255, 255], type='', swap=''),
+        5:
+        dict(name='abdomen', id=5, color=[255, 255, 255], type='', swap=''),
+        6:
+        dict(
+            name='forelegR1',
+            id=6,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegL1'),
+        7:
+        dict(
+            name='forelegR2',
+            id=7,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegL2'),
+        8:
+        dict(
+            name='forelegR3',
+            id=8,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegL3'),
+        9:
+        dict(
+            name='forelegR4',
+            id=9,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegL4'),
+        10:
+        dict(
+            name='midlegR1',
+            id=10,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegL1'),
+        11:
+        dict(
+            name='midlegR2',
+            id=11,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegL2'),
+        12:
+        dict(
+            name='midlegR3',
+            id=12,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegL3'),
+        13:
+        dict(
+            name='midlegR4',
+            id=13,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegL4'),
+        14:
+        dict(
+            name='hindlegR1',
+            id=14,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegL1'),
+        15:
+        dict(
+            name='hindlegR2',
+            id=15,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegL2'),
+        16:
+        dict(
+            name='hindlegR3',
+            id=16,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegL3'),
+        17:
+        dict(
+            name='hindlegR4',
+            id=17,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegL4'),
+        18:
+        dict(
+            name='forelegL1',
+            id=18,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegR1'),
+        19:
+        dict(
+            name='forelegL2',
+            id=19,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegR2'),
+        20:
+        dict(
+            name='forelegL3',
+            id=20,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegR3'),
+        21:
+        dict(
+            name='forelegL4',
+            id=21,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegR4'),
+        22:
+        dict(
+            name='midlegL1',
+            id=22,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegR1'),
+        23:
+        dict(
+            name='midlegL2',
+            id=23,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegR2'),
+        24:
+        dict(
+            name='midlegL3',
+            id=24,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegR3'),
+        25:
+        dict(
+            name='midlegL4',
+            id=25,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegR4'),
+        26:
+        dict(
+            name='hindlegL1',
+            id=26,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegR1'),
+        27:
+        dict(
+            name='hindlegL2',
+            id=27,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegR2'),
+        28:
+        dict(
+            name='hindlegL3',
+            id=28,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegR3'),
+        29:
+        dict(
+            name='hindlegL4',
+            id=29,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegR4'),
+        30:
+        dict(
+            name='wingL', id=30, color=[255, 255, 255], type='', swap='wingR'),
+        31:
+        dict(
+            name='wingR', id=31, color=[255, 255, 255], type='', swap='wingL'),
+    },
+    skeleton_info={
+        0: dict(link=('eyeL', 'head'), id=0, color=[255, 255, 255]),
+        1: dict(link=('eyeR', 'head'), id=1, color=[255, 255, 255]),
+        2: dict(link=('neck', 'head'), id=2, color=[255, 255, 255]),
+        3: dict(link=('thorax', 'neck'), id=3, color=[255, 255, 255]),
+        4: dict(link=('abdomen', 'thorax'), id=4, color=[255, 255, 255]),
+        5: dict(link=('forelegR2', 'forelegR1'), id=5, color=[255, 255, 255]),
+        6: dict(link=('forelegR3', 'forelegR2'), id=6, color=[255, 255, 255]),
+        7: dict(link=('forelegR4', 'forelegR3'), id=7, color=[255, 255, 255]),
+        8: dict(link=('midlegR2', 'midlegR1'), id=8, color=[255, 255, 255]),
+        9: dict(link=('midlegR3', 'midlegR2'), id=9, color=[255, 255, 255]),
+        10: dict(link=('midlegR4', 'midlegR3'), id=10, color=[255, 255, 255]),
+        11:
+        dict(link=('hindlegR2', 'hindlegR1'), id=11, color=[255, 255, 255]),
+        12:
+        dict(link=('hindlegR3', 'hindlegR2'), id=12, color=[255, 255, 255]),
+        13:
+        dict(link=('hindlegR4', 'hindlegR3'), id=13, color=[255, 255, 255]),
+        14:
+        dict(link=('forelegL2', 'forelegL1'), id=14, color=[255, 255, 255]),
+        15:
+        dict(link=('forelegL3', 'forelegL2'), id=15, color=[255, 255, 255]),
+        16:
+        dict(link=('forelegL4', 'forelegL3'), id=16, color=[255, 255, 255]),
+        17: dict(link=('midlegL2', 'midlegL1'), id=17, color=[255, 255, 255]),
+        18: dict(link=('midlegL3', 'midlegL2'), id=18, color=[255, 255, 255]),
+        19: dict(link=('midlegL4', 'midlegL3'), id=19, color=[255, 255, 255]),
+        20:
+        dict(link=('hindlegL2', 'hindlegL1'), id=20, color=[255, 255, 255]),
+        21:
+        dict(link=('hindlegL3', 'hindlegL2'), id=21, color=[255, 255, 255]),
+        22:
+        dict(link=('hindlegL4', 'hindlegL3'), id=22, color=[255, 255, 255]),
+        23: dict(link=('wingL', 'neck'), id=23, color=[255, 255, 255]),
+        24: dict(link=('wingR', 'neck'), id=24, color=[255, 255, 255])
+    },
+    joint_weights=[1.] * 32,
+    sigmas=[])

configs/_base_/datasets/freihand2d.py

@@ -0,0 +1,150 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='freihand',
+    paper_info=dict(
+        author='Zimmermann, Christian and Ceylan, Duygu and '
+        'Yang, Jimei and Russell, Bryan and '
+        'Argus, Max and Brox, Thomas',
+        title='Freihand: A dataset for markerless capture of hand pose '
+        'and shape from single rgb images',
+        container='Proceedings of the IEEE International '
+        'Conference on Computer Vision',
+        year='2019',
+        homepage='https://lmb.informatik.uni-freiburg.de/projects/freihand/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='wrist', id=0, color=[255, 255, 255], type='', swap=''),
+        1:
+        dict(name='thumb1', id=1, color=[255, 128, 0], type='', swap=''),
+        2:
+        dict(name='thumb2', id=2, color=[255, 128, 0], type='', swap=''),
+        3:
+        dict(name='thumb3', id=3, color=[255, 128, 0], type='', swap=''),
+        4:
+        dict(name='thumb4', id=4, color=[255, 128, 0], type='', swap=''),
+        5:
+        dict(
+            name='forefinger1', id=5, color=[255, 153, 255], type='', swap=''),
+        6:
+        dict(
+            name='forefinger2', id=6, color=[255, 153, 255], type='', swap=''),
+        7:
+        dict(
+            name='forefinger3', id=7, color=[255, 153, 255], type='', swap=''),
+        8:
+        dict(
+            name='forefinger4', id=8, color=[255, 153, 255], type='', swap=''),
+        9:
+        dict(
+            name='middle_finger1',
+            id=9,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        10:
+        dict(
+            name='middle_finger2',
+            id=10,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        11:
+        dict(
+            name='middle_finger3',
+            id=11,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        12:
+        dict(
+            name='middle_finger4',
+            id=12,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        13:
+        dict(
+            name='ring_finger1', id=13, color=[255, 51, 51], type='', swap=''),
+        14:
+        dict(
+            name='ring_finger2', id=14, color=[255, 51, 51], type='', swap=''),
+        15:
+        dict(
+            name='ring_finger3', id=15, color=[255, 51, 51], type='', swap=''),
+        16:
+        dict(
+            name='ring_finger4', id=16, color=[255, 51, 51], type='', swap=''),
+        17:
+        dict(name='pinky_finger1', id=17, color=[0, 255, 0], type='', swap=''),
+        18:
+        dict(name='pinky_finger2', id=18, color=[0, 255, 0], type='', swap=''),
+        19:
+        dict(name='pinky_finger3', id=19, color=[0, 255, 0], type='', swap=''),
+        20:
+        dict(name='pinky_finger4', id=20, color=[0, 255, 0], type='', swap='')
+    },
+    skeleton_info={
+        0:
+        dict(link=('wrist', 'thumb1'), id=0, color=[255, 128, 0]),
+        1:
+        dict(link=('thumb1', 'thumb2'), id=1, color=[255, 128, 0]),
+        2:
+        dict(link=('thumb2', 'thumb3'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('thumb3', 'thumb4'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('wrist', 'forefinger1'), id=4, color=[255, 153, 255]),
+        5:
+        dict(link=('forefinger1', 'forefinger2'), id=5, color=[255, 153, 255]),
+        6:
+        dict(link=('forefinger2', 'forefinger3'), id=6, color=[255, 153, 255]),
+        7:
+        dict(link=('forefinger3', 'forefinger4'), id=7, color=[255, 153, 255]),
+        8:
+        dict(link=('wrist', 'middle_finger1'), id=8, color=[102, 178, 255]),
+        9:
+        dict(
+            link=('middle_finger1', 'middle_finger2'),
+            id=9,
+            color=[102, 178, 255]),
+        10:
+        dict(
+            link=('middle_finger2', 'middle_finger3'),
+            id=10,
+            color=[102, 178, 255]),
+        11:
+        dict(
+            link=('middle_finger3', 'middle_finger4'),
+            id=11,
+            color=[102, 178, 255]),
+        12:
+        dict(link=('wrist', 'ring_finger1'), id=12, color=[255, 51, 51]),
+        13:
+        dict(
+            link=('ring_finger1', 'ring_finger2'), id=13, color=[255, 51, 51]),
+        14:
+        dict(
+            link=('ring_finger2', 'ring_finger3'), id=14, color=[255, 51, 51]),
+        15:
+        dict(
+            link=('ring_finger3', 'ring_finger4'), id=15, color=[255, 51, 51]),
+        16:
+        dict(link=('wrist', 'pinky_finger1'), id=16, color=[0, 255, 0]),
+        17:
+        dict(
+            link=('pinky_finger1', 'pinky_finger2'), id=17, color=[0, 255, 0]),
+        18:
+        dict(
+            link=('pinky_finger2', 'pinky_finger3'), id=18, color=[0, 255, 0]),
+        19:
+        dict(
+            link=('pinky_finger3', 'pinky_finger4'), id=19, color=[0, 255, 0])
+    },
+    joint_weights=[1.] * 21,
+    sigmas=[])

configs/_base_/datasets/goliath3d.py

@@ -0,0 +1,2131 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+coco_wholebody_info = dict(
+    dataset_name='coco_wholebody',
+    paper_info=dict(
+        author='Jin, Sheng and Xu, Lumin and Xu, Jin and '
+        'Wang, Can and Liu, Wentao and '
+        'Qian, Chen and Ouyang, Wanli and Luo, Ping',
+        title='Whole-Body Human Pose Estimation in the Wild',
+        container='Proceedings of the European '
+        'Conference on Computer Vision (ECCV)',
+        year='2020',
+        homepage='https://github.com/jin-s13/COCO-WholeBody/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
+        1:
+        dict(
+            name='left_eye',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_eye'),
+        2:
+        dict(
+            name='right_eye',
+            id=2,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_eye'),
+        3:
+        dict(
+            name='left_ear',
+            id=3,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_ear'),
+        4:
+        dict(
+            name='right_ear',
+            id=4,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_ear'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='right_shoulder',
+            id=6,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        7:
+        dict(
+            name='left_elbow',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        8:
+        dict(
+            name='right_elbow',
+            id=8,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        9:
+        dict(
+            name='left_wrist',
+            id=9,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='right_hip',
+            id=12,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        13:
+        dict(
+            name='left_knee',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        14:
+        dict(
+            name='right_knee',
+            id=14,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        15:
+        dict(
+            name='left_ankle',
+            id=15,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        16:
+        dict(
+            name='right_ankle',
+            id=16,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle'),
+        17:
+        dict(
+            name='left_big_toe',
+            id=17,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_big_toe'),
+        18:
+        dict(
+            name='left_small_toe',
+            id=18,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_small_toe'),
+        19:
+        dict(
+            name='left_heel',
+            id=19,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_heel'),
+        20:
+        dict(
+            name='right_big_toe',
+            id=20,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_big_toe'),
+        21:
+        dict(
+            name='right_small_toe',
+            id=21,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_small_toe'),
+        22:
+        dict(
+            name='right_heel',
+            id=22,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_heel'),
+        23:
+        dict(
+            name='face-0',
+            id=23,
+            color=[255, 255, 255],
+            type='',
+            swap='face-16'),
+        24:
+        dict(
+            name='face-1',
+            id=24,
+            color=[255, 255, 255],
+            type='',
+            swap='face-15'),
+        25:
+        dict(
+            name='face-2',
+            id=25,
+            color=[255, 255, 255],
+            type='',
+            swap='face-14'),
+        26:
+        dict(
+            name='face-3',
+            id=26,
+            color=[255, 255, 255],
+            type='',
+            swap='face-13'),
+        27:
+        dict(
+            name='face-4',
+            id=27,
+            color=[255, 255, 255],
+            type='',
+            swap='face-12'),
+        28:
+        dict(
+            name='face-5',
+            id=28,
+            color=[255, 255, 255],
+            type='',
+            swap='face-11'),
+        29:
+        dict(
+            name='face-6',
+            id=29,
+            color=[255, 255, 255],
+            type='',
+            swap='face-10'),
+        30:
+        dict(
+            name='face-7',
+            id=30,
+            color=[255, 255, 255],
+            type='',
+            swap='face-9'),
+        31:
+        dict(name='face-8', id=31, color=[255, 255, 255], type='', swap=''),
+        32:
+        dict(
+            name='face-9',
+            id=32,
+            color=[255, 255, 255],
+            type='',
+            swap='face-7'),
+        33:
+        dict(
+            name='face-10',
+            id=33,
+            color=[255, 255, 255],
+            type='',
+            swap='face-6'),
+        34:
+        dict(
+            name='face-11',
+            id=34,
+            color=[255, 255, 255],
+            type='',
+            swap='face-5'),
+        35:
+        dict(
+            name='face-12',
+            id=35,
+            color=[255, 255, 255],
+            type='',
+            swap='face-4'),
+        36:
+        dict(
+            name='face-13',
+            id=36,
+            color=[255, 255, 255],
+            type='',
+            swap='face-3'),
+        37:
+        dict(
+            name='face-14',
+            id=37,
+            color=[255, 255, 255],
+            type='',
+            swap='face-2'),
+        38:
+        dict(
+            name='face-15',
+            id=38,
+            color=[255, 255, 255],
+            type='',
+            swap='face-1'),
+        39:
+        dict(
+            name='face-16',
+            id=39,
+            color=[255, 255, 255],
+            type='',
+            swap='face-0'),
+        40:
+        dict(
+            name='face-17',
+            id=40,
+            color=[255, 255, 255],
+            type='',
+            swap='face-26'),
+        41:
+        dict(
+            name='face-18',
+            id=41,
+            color=[255, 255, 255],
+            type='',
+            swap='face-25'),
+        42:
+        dict(
+            name='face-19',
+            id=42,
+            color=[255, 255, 255],
+            type='',
+            swap='face-24'),
+        43:
+        dict(
+            name='face-20',
+            id=43,
+            color=[255, 255, 255],
+            type='',
+            swap='face-23'),
+        44:
+        dict(
+            name='face-21',
+            id=44,
+            color=[255, 255, 255],
+            type='',
+            swap='face-22'),
+        45:
+        dict(
+            name='face-22',
+            id=45,
+            color=[255, 255, 255],
+            type='',
+            swap='face-21'),
+        46:
+        dict(
+            name='face-23',
+            id=46,
+            color=[255, 255, 255],
+            type='',
+            swap='face-20'),
+        47:
+        dict(
+            name='face-24',
+            id=47,
+            color=[255, 255, 255],
+            type='',
+            swap='face-19'),
+        48:
+        dict(
+            name='face-25',
+            id=48,
+            color=[255, 255, 255],
+            type='',
+            swap='face-18'),
+        49:
+        dict(
+            name='face-26',
+            id=49,
+            color=[255, 255, 255],
+            type='',
+            swap='face-17'),
+        50:
+        dict(name='face-27', id=50, color=[255, 255, 255], type='', swap=''),
+        51:
+        dict(name='face-28', id=51, color=[255, 255, 255], type='', swap=''),
+        52:
+        dict(name='face-29', id=52, color=[255, 255, 255], type='', swap=''),
+        53:
+        dict(name='face-30', id=53, color=[255, 255, 255], type='', swap=''),
+        54:
+        dict(
+            name='face-31',
+            id=54,
+            color=[255, 255, 255],
+            type='',
+            swap='face-35'),
+        55:
+        dict(
+            name='face-32',
+            id=55,
+            color=[255, 255, 255],
+            type='',
+            swap='face-34'),
+        56:
+        dict(name='face-33', id=56, color=[255, 255, 255], type='', swap=''),
+        57:
+        dict(
+            name='face-34',
+            id=57,
+            color=[255, 255, 255],
+            type='',
+            swap='face-32'),
+        58:
+        dict(
+            name='face-35',
+            id=58,
+            color=[255, 255, 255],
+            type='',
+            swap='face-31'),
+        59:
+        dict(
+            name='face-36',
+            id=59,
+            color=[255, 255, 255],
+            type='',
+            swap='face-45'),
+        60:
+        dict(
+            name='face-37',
+            id=60,
+            color=[255, 255, 255],
+            type='',
+            swap='face-44'),
+        61:
+        dict(
+            name='face-38',
+            id=61,
+            color=[255, 255, 255],
+            type='',
+            swap='face-43'),
+        62:
+        dict(
+            name='face-39',
+            id=62,
+            color=[255, 255, 255],
+            type='',
+            swap='face-42'),
+        63:
+        dict(
+            name='face-40',
+            id=63,
+            color=[255, 255, 255],
+            type='',
+            swap='face-47'),
+        64:
+        dict(
+            name='face-41',
+            id=64,
+            color=[255, 255, 255],
+            type='',
+            swap='face-46'),
+        65:
+        dict(
+            name='face-42',
+            id=65,
+            color=[255, 255, 255],
+            type='',
+            swap='face-39'),
+        66:
+        dict(
+            name='face-43',
+            id=66,
+            color=[255, 255, 255],
+            type='',
+            swap='face-38'),
+        67:
+        dict(
+            name='face-44',
+            id=67,
+            color=[255, 255, 255],
+            type='',
+            swap='face-37'),
+        68:
+        dict(
+            name='face-45',
+            id=68,
+            color=[255, 255, 255],
+            type='',
+            swap='face-36'),
+        69:
+        dict(
+            name='face-46',
+            id=69,
+            color=[255, 255, 255],
+            type='',
+            swap='face-41'),
+        70:
+        dict(
+            name='face-47',
+            id=70,
+            color=[255, 255, 255],
+            type='',
+            swap='face-40'),
+        71:
+        dict(
+            name='face-48',
+            id=71,
+            color=[255, 255, 255],
+            type='',
+            swap='face-54'),
+        72:
+        dict(
+            name='face-49',
+            id=72,
+            color=[255, 255, 255],
+            type='',
+            swap='face-53'),
+        73:
+        dict(
+            name='face-50',
+            id=73,
+            color=[255, 255, 255],
+            type='',
+            swap='face-52'),
+        74:
+        dict(name='face-51', id=74, color=[255, 255, 255], type='', swap=''),
+        75:
+        dict(
+            name='face-52',
+            id=75,
+            color=[255, 255, 255],
+            type='',
+            swap='face-50'),
+        76:
+        dict(
+            name='face-53',
+            id=76,
+            color=[255, 255, 255],
+            type='',
+            swap='face-49'),
+        77:
+        dict(
+            name='face-54',
+            id=77,
+            color=[255, 255, 255],
+            type='',
+            swap='face-48'),
+        78:
+        dict(
+            name='face-55',
+            id=78,
+            color=[255, 255, 255],
+            type='',
+            swap='face-59'),
+        79:
+        dict(
+            name='face-56',
+            id=79,
+            color=[255, 255, 255],
+            type='',
+            swap='face-58'),
+        80:
+        dict(name='face-57', id=80, color=[255, 255, 255], type='', swap=''),
+        81:
+        dict(
+            name='face-58',
+            id=81,
+            color=[255, 255, 255],
+            type='',
+            swap='face-56'),
+        82:
+        dict(
+            name='face-59',
+            id=82,
+            color=[255, 255, 255],
+            type='',
+            swap='face-55'),
+        83:
+        dict(
+            name='face-60',
+            id=83,
+            color=[255, 255, 255],
+            type='',
+            swap='face-64'),
+        84:
+        dict(
+            name='face-61',
+            id=84,
+            color=[255, 255, 255],
+            type='',
+            swap='face-63'),
+        85:
+        dict(name='face-62', id=85, color=[255, 255, 255], type='', swap=''),
+        86:
+        dict(
+            name='face-63',
+            id=86,
+            color=[255, 255, 255],
+            type='',
+            swap='face-61'),
+        87:
+        dict(
+            name='face-64',
+            id=87,
+            color=[255, 255, 255],
+            type='',
+            swap='face-60'),
+        88:
+        dict(
+            name='face-65',
+            id=88,
+            color=[255, 255, 255],
+            type='',
+            swap='face-67'),
+        89:
+        dict(name='face-66', id=89, color=[255, 255, 255], type='', swap=''),
+        90:
+        dict(
+            name='face-67',
+            id=90,
+            color=[255, 255, 255],
+            type='',
+            swap='face-65'),
+        91:
+        dict(
+            name='left_hand_root',
+            id=91,
+            color=[255, 255, 255],
+            type='',
+            swap='right_hand_root'),
+        92:
+        dict(
+            name='left_thumb1',
+            id=92,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb1'),
+        93:
+        dict(
+            name='left_thumb2',
+            id=93,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb2'),
+        94:
+        dict(
+            name='left_thumb3',
+            id=94,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb3'),
+        95:
+        dict(
+            name='left_thumb4',
+            id=95,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb4'),
+        96:
+        dict(
+            name='left_forefinger1',
+            id=96,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger1'),
+        97:
+        dict(
+            name='left_forefinger2',
+            id=97,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger2'),
+        98:
+        dict(
+            name='left_forefinger3',
+            id=98,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger3'),
+        99:
+        dict(
+            name='left_forefinger4',
+            id=99,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger4'),
+        100:
+        dict(
+            name='left_middle_finger1',
+            id=100,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger1'),
+        101:
+        dict(
+            name='left_middle_finger2',
+            id=101,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger2'),
+        102:
+        dict(
+            name='left_middle_finger3',
+            id=102,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger3'),
+        103:
+        dict(
+            name='left_middle_finger4',
+            id=103,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger4'),
+        104:
+        dict(
+            name='left_ring_finger1',
+            id=104,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger1'),
+        105:
+        dict(
+            name='left_ring_finger2',
+            id=105,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger2'),
+        106:
+        dict(
+            name='left_ring_finger3',
+            id=106,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger3'),
+        107:
+        dict(
+            name='left_ring_finger4',
+            id=107,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger4'),
+        108:
+        dict(
+            name='left_pinky_finger1',
+            id=108,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger1'),
+        109:
+        dict(
+            name='left_pinky_finger2',
+            id=109,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger2'),
+        110:
+        dict(
+            name='left_pinky_finger3',
+            id=110,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger3'),
+        111:
+        dict(
+            name='left_pinky_finger4',
+            id=111,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger4'),
+        112:
+        dict(
+            name='right_hand_root',
+            id=112,
+            color=[255, 255, 255],
+            type='',
+            swap='left_hand_root'),
+        113:
+        dict(
+            name='right_thumb1',
+            id=113,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb1'),
+        114:
+        dict(
+            name='right_thumb2',
+            id=114,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb2'),
+        115:
+        dict(
+            name='right_thumb3',
+            id=115,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb3'),
+        116:
+        dict(
+            name='right_thumb4',
+            id=116,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb4'),
+        117:
+        dict(
+            name='right_forefinger1',
+            id=117,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger1'),
+        118:
+        dict(
+            name='right_forefinger2',
+            id=118,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger2'),
+        119:
+        dict(
+            name='right_forefinger3',
+            id=119,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger3'),
+        120:
+        dict(
+            name='right_forefinger4',
+            id=120,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger4'),
+        121:
+        dict(
+            name='right_middle_finger1',
+            id=121,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger1'),
+        122:
+        dict(
+            name='right_middle_finger2',
+            id=122,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger2'),
+        123:
+        dict(
+            name='right_middle_finger3',
+            id=123,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger3'),
+        124:
+        dict(
+            name='right_middle_finger4',
+            id=124,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger4'),
+        125:
+        dict(
+            name='right_ring_finger1',
+            id=125,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger1'),
+        126:
+        dict(
+            name='right_ring_finger2',
+            id=126,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger2'),
+        127:
+        dict(
+            name='right_ring_finger3',
+            id=127,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger3'),
+        128:
+        dict(
+            name='right_ring_finger4',
+            id=128,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger4'),
+        129:
+        dict(
+            name='right_pinky_finger1',
+            id=129,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger1'),
+        130:
+        dict(
+            name='right_pinky_finger2',
+            id=130,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger2'),
+        131:
+        dict(
+            name='right_pinky_finger3',
+            id=131,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger3'),
+        132:
+        dict(
+            name='right_pinky_finger4',
+            id=132,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger4')
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
+        5:
+        dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
+        6:
+        dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
+        7:
+        dict(
+            link=('left_shoulder', 'right_shoulder'),
+            id=7,
+            color=[51, 153, 255]),
+        8:
+        dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
+        9:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
+        13:
+        dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
+        14:
+        dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
+        15:
+        dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
+        16:
+        dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
+        17:
+        dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
+        18:
+        dict(
+            link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255]),
+        19:
+        dict(link=('left_ankle', 'left_big_toe'), id=19, color=[0, 255, 0]),
+        20:
+        dict(link=('left_ankle', 'left_small_toe'), id=20, color=[0, 255, 0]),
+        21:
+        dict(link=('left_ankle', 'left_heel'), id=21, color=[0, 255, 0]),
+        22:
+        dict(
+            link=('right_ankle', 'right_big_toe'), id=22, color=[255, 128, 0]),
+        23:
+        dict(
+            link=('right_ankle', 'right_small_toe'),
+            id=23,
+            color=[255, 128, 0]),
+        24:
+        dict(link=('right_ankle', 'right_heel'), id=24, color=[255, 128, 0]),
+        25:
+        dict(
+            link=('left_hand_root', 'left_thumb1'), id=25, color=[255, 128,
+                                                                  0]),
+        26:
+        dict(link=('left_thumb1', 'left_thumb2'), id=26, color=[255, 128, 0]),
+        27:
+        dict(link=('left_thumb2', 'left_thumb3'), id=27, color=[255, 128, 0]),
+        28:
+        dict(link=('left_thumb3', 'left_thumb4'), id=28, color=[255, 128, 0]),
+        29:
+        dict(
+            link=('left_hand_root', 'left_forefinger1'),
+            id=29,
+            color=[255, 153, 255]),
+        30:
+        dict(
+            link=('left_forefinger1', 'left_forefinger2'),
+            id=30,
+            color=[255, 153, 255]),
+        31:
+        dict(
+            link=('left_forefinger2', 'left_forefinger3'),
+            id=31,
+            color=[255, 153, 255]),
+        32:
+        dict(
+            link=('left_forefinger3', 'left_forefinger4'),
+            id=32,
+            color=[255, 153, 255]),
+        33:
+        dict(
+            link=('left_hand_root', 'left_middle_finger1'),
+            id=33,
+            color=[102, 178, 255]),
+        34:
+        dict(
+            link=('left_middle_finger1', 'left_middle_finger2'),
+            id=34,
+            color=[102, 178, 255]),
+        35:
+        dict(
+            link=('left_middle_finger2', 'left_middle_finger3'),
+            id=35,
+            color=[102, 178, 255]),
+        36:
+        dict(
+            link=('left_middle_finger3', 'left_middle_finger4'),
+            id=36,
+            color=[102, 178, 255]),
+        37:
+        dict(
+            link=('left_hand_root', 'left_ring_finger1'),
+            id=37,
+            color=[255, 51, 51]),
+        38:
+        dict(
+            link=('left_ring_finger1', 'left_ring_finger2'),
+            id=38,
+            color=[255, 51, 51]),
+        39:
+        dict(
+            link=('left_ring_finger2', 'left_ring_finger3'),
+            id=39,
+            color=[255, 51, 51]),
+        40:
+        dict(
+            link=('left_ring_finger3', 'left_ring_finger4'),
+            id=40,
+            color=[255, 51, 51]),
+        41:
+        dict(
+            link=('left_hand_root', 'left_pinky_finger1'),
+            id=41,
+            color=[0, 255, 0]),
+        42:
+        dict(
+            link=('left_pinky_finger1', 'left_pinky_finger2'),
+            id=42,
+            color=[0, 255, 0]),
+        43:
+        dict(
+            link=('left_pinky_finger2', 'left_pinky_finger3'),
+            id=43,
+            color=[0, 255, 0]),
+        44:
+        dict(
+            link=('left_pinky_finger3', 'left_pinky_finger4'),
+            id=44,
+            color=[0, 255, 0]),
+        45:
+        dict(
+            link=('right_hand_root', 'right_thumb1'),
+            id=45,
+            color=[255, 128, 0]),
+        46:
+        dict(
+            link=('right_thumb1', 'right_thumb2'), id=46, color=[255, 128, 0]),
+        47:
+        dict(
+            link=('right_thumb2', 'right_thumb3'), id=47, color=[255, 128, 0]),
+        48:
+        dict(
+            link=('right_thumb3', 'right_thumb4'), id=48, color=[255, 128, 0]),
+        49:
+        dict(
+            link=('right_hand_root', 'right_forefinger1'),
+            id=49,
+            color=[255, 153, 255]),
+        50:
+        dict(
+            link=('right_forefinger1', 'right_forefinger2'),
+            id=50,
+            color=[255, 153, 255]),
+        51:
+        dict(
+            link=('right_forefinger2', 'right_forefinger3'),
+            id=51,
+            color=[255, 153, 255]),
+        52:
+        dict(
+            link=('right_forefinger3', 'right_forefinger4'),
+            id=52,
+            color=[255, 153, 255]),
+        53:
+        dict(
+            link=('right_hand_root', 'right_middle_finger1'),
+            id=53,
+            color=[102, 178, 255]),
+        54:
+        dict(
+            link=('right_middle_finger1', 'right_middle_finger2'),
+            id=54,
+            color=[102, 178, 255]),
+        55:
+        dict(
+            link=('right_middle_finger2', 'right_middle_finger3'),
+            id=55,
+            color=[102, 178, 255]),
+        56:
+        dict(
+            link=('right_middle_finger3', 'right_middle_finger4'),
+            id=56,
+            color=[102, 178, 255]),
+        57:
+        dict(
+            link=('right_hand_root', 'right_ring_finger1'),
+            id=57,
+            color=[255, 51, 51]),
+        58:
+        dict(
+            link=('right_ring_finger1', 'right_ring_finger2'),
+            id=58,
+            color=[255, 51, 51]),
+        59:
+        dict(
+            link=('right_ring_finger2', 'right_ring_finger3'),
+            id=59,
+            color=[255, 51, 51]),
+        60:
+        dict(
+            link=('right_ring_finger3', 'right_ring_finger4'),
+            id=60,
+            color=[255, 51, 51]),
+        61:
+        dict(
+            link=('right_hand_root', 'right_pinky_finger1'),
+            id=61,
+            color=[0, 255, 0]),
+        62:
+        dict(
+            link=('right_pinky_finger1', 'right_pinky_finger2'),
+            id=62,
+            color=[0, 255, 0]),
+        63:
+        dict(
+            link=('right_pinky_finger2', 'right_pinky_finger3'),
+            id=63,
+            color=[0, 255, 0]),
+        64:
+        dict(
+            link=('right_pinky_finger3', 'right_pinky_finger4'),
+            id=64,
+            color=[0, 255, 0])
+    },
+    joint_weights=[1.] * 133,
+    # 'https://github.com/jin-s13/COCO-WholeBody/blob/master/'
+    # 'evaluation/myeval_wholebody.py#L175'
+    sigmas=[
+        0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
+        0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089, 0.068, 0.066, 0.066,
+        0.092, 0.094, 0.094, 0.042, 0.043, 0.044, 0.043, 0.040, 0.035, 0.031,
+        0.025, 0.020, 0.023, 0.029, 0.032, 0.037, 0.038, 0.043, 0.041, 0.045,
+        0.013, 0.012, 0.011, 0.011, 0.012, 0.012, 0.011, 0.011, 0.013, 0.015,
+        0.009, 0.007, 0.007, 0.007, 0.012, 0.009, 0.008, 0.016, 0.010, 0.017,
+        0.011, 0.009, 0.011, 0.009, 0.007, 0.013, 0.008, 0.011, 0.012, 0.010,
+        0.034, 0.008, 0.008, 0.009, 0.008, 0.008, 0.007, 0.010, 0.008, 0.009,
+        0.009, 0.009, 0.007, 0.007, 0.008, 0.011, 0.008, 0.008, 0.008, 0.01,
+        0.008, 0.029, 0.022, 0.035, 0.037, 0.047, 0.026, 0.025, 0.024, 0.035,
+        0.018, 0.024, 0.022, 0.026, 0.017, 0.021, 0.021, 0.032, 0.02, 0.019,
+        0.022, 0.031, 0.029, 0.022, 0.035, 0.037, 0.047, 0.026, 0.025, 0.024,
+        0.035, 0.018, 0.024, 0.022, 0.026, 0.017, 0.021, 0.021, 0.032, 0.02,
+        0.019, 0.022, 0.031
+    ])
+
+dataset_info = dict(
+    dataset_name='goliath3d',
+    paper_info=dict(
+        author='',
+        year='',
+        homepage='',
+    ),
+    min_visible_keypoints=8,
+    image_height=4096,
+    image_width=2668,
+    original_keypoint_info={
+            0: 'nose',
+            1: 'left_eye',
+            2: 'right_eye',
+            3: 'left_ear',
+            4: 'right_ear',
+            5: 'left_shoulder',
+            6: 'right_shoulder',
+            7: 'left_elbow',
+            8: 'right_elbow',
+            9: 'left_hip',
+            10: 'right_hip',
+            11: 'left_knee',
+            12: 'right_knee',
+            13: 'left_ankle',
+            14: 'right_ankle',
+            15: 'left_big_toe', # 'left_big_toe_tip'
+            16: 'left_small_toe', # 'left_small_toe_tip'
+            17: 'left_heel',
+            18: 'right_big_toe', # 'right_big_toe_tip'
+            19: 'right_small_toe', # 'right_small_toe_tip'
+            20: 'right_heel',
+            21: 'right_thumb4', # 'right_thumb_tip'
+            22: 'right_thumb3', # 'right_thumb_first_joint'
+            23: 'right_thumb2', # 'right_thumb_second_joint'
+            24: 'right_thumb_third_joint', # 'right_thumb_third_joint'
+
+            25: 'right_forefinger4', # 'right_index_tip'
+            26: 'right_forefinger3', # 'right_index_first_joint'
+            27: 'right_forefinger2', # 'right_index_second_joint'
+            28: 'right_forefinger_third_joint',  # 'right_index_third_joint'
+
+            29: 'right_middle_finger4', # 'right_middle_tip'
+            30: 'right_middle_finger3', # 'right_middle_first_joint'
+            31: 'right_middle_finger2', # 'right_middle_second_joint'
+            32: 'right_middle_finger_third_joint', # 'right_middle_third_joint'
+
+            33: 'right_ring_finger4', # 'right_ring_tip'
+            34: 'right_ring_finger3', # 'right_ring_first_joint'
+            35: 'right_ring_finger2', # 'right_ring_second_joint'
+            36: 'right_ring_finger_third_joint', # 'right_ring_third_joint'
+
+            37: 'right_pinky_finger4', # 'right_pinky_tip'
+            38: 'right_pinky_finger3', # 'right_pinky_first_joint'
+            39: 'right_pinky_finger2', # 'right_pinky_second_joint'
+            40: 'right_pinky_finger_third_joint', # 'right_pinky_third_joint'
+
+            41: 'right_wrist',
+
+            42: 'left_thumb4', # 'left_thumb_tip'
+            43: 'left_thumb3', # 'left_thumb_first_joint'
+            44: 'left_thumb2', # 'left_thumb_second_joint'
+            45: 'left_thumb_third_joint', # 'left_thumb_third_joint'
+            
+            46: 'left_forefinger4', # 'left_index_tip'
+            47: 'left_forefinger3', # 'left_index_first_joint'
+            48: 'left_forefinger2', # 'left_index_second_joint'
+            49: 'left_forefinger_third_joint', # 'left_index_third_joint'
+
+            50: 'left_middle_finger4', # 'left_middle_tip'
+            51: 'left_middle_finger3', # 'left_middle_first_joint'
+            52: 'left_middle_finger2', # 'left_middle_second_joint'
+            53: 'left_middle_finger_third_joint', # 'left_middle_third_joint'
+
+            54: 'left_ring_finger4', # 'left_ring_tip'
+            55: 'left_ring_finger3', # 'left_ring_first_joint'
+            56: 'left_ring_finger2', # 'left_ring_second_joint'
+            57: 'left_ring_finger_third_joint', # 'left_ring_third_joint'
+
+            58: 'left_pinky_finger4', # 'left_pinky_tip' 
+            59: 'left_pinky_finger3',   # 'left_pinky_first_joint'
+            60: 'left_pinky_finger2', # 'left_pinky_second_joint'
+            61: 'left_pinky_finger_third_joint', # 'left_pinky_third_joint'
+
+            62: 'left_wrist', # 'left_wrist'
+
+            63: 'left_olecranon',
+            64: 'right_olecranon',
+            65: 'left_cubital_fossa',
+            66: 'right_cubital_fossa',
+            67: 'left_acromion',
+            68: 'right_acromion',
+            69: 'neck',
+            70: 'center_of_glabella',
+            71: 'center_of_nose_root',
+            72: 'tip_of_nose_bridge',
+            73: 'midpoint_1_of_nose_bridge',
+            74: 'midpoint_2_of_nose_bridge',
+            75: 'midpoint_3_of_nose_bridge',
+            76: 'center_of_labiomental_groove',
+            77: 'tip_of_chin',
+            78: 'upper_startpoint_of_r_eyebrow',
+            79: 'lower_startpoint_of_r_eyebrow',
+            80: 'end_of_r_eyebrow',
+            81: 'upper_midpoint_1_of_r_eyebrow',
+            82: 'lower_midpoint_1_of_r_eyebrow',
+            83: 'upper_midpoint_2_of_r_eyebrow',
+            84: 'upper_midpoint_3_of_r_eyebrow',
+            85: 'lower_midpoint_2_of_r_eyebrow',
+            86: 'lower_midpoint_3_of_r_eyebrow',
+            87: 'upper_startpoint_of_l_eyebrow',
+            88: 'lower_startpoint_of_l_eyebrow',
+            89: 'end_of_l_eyebrow',
+            90: 'upper_midpoint_1_of_l_eyebrow',
+            91: 'lower_midpoint_1_of_l_eyebrow',
+            92: 'upper_midpoint_2_of_l_eyebrow',
+            93: 'upper_midpoint_3_of_l_eyebrow',
+            94: 'lower_midpoint_2_of_l_eyebrow',
+            95: 'lower_midpoint_3_of_l_eyebrow',
+            96: 'l_inner_end_of_upper_lash_line',
+            97: 'l_outer_end_of_upper_lash_line',
+            98: 'l_centerpoint_of_upper_lash_line',
+            99: 'l_midpoint_2_of_upper_lash_line',
+            100: 'l_midpoint_1_of_upper_lash_line',
+            101: 'l_midpoint_6_of_upper_lash_line',
+            102: 'l_midpoint_5_of_upper_lash_line',
+            103: 'l_midpoint_4_of_upper_lash_line',
+            104: 'l_midpoint_3_of_upper_lash_line',
+            105: 'l_outer_end_of_upper_eyelid_line',
+            106: 'l_midpoint_6_of_upper_eyelid_line',
+            107: 'l_midpoint_2_of_upper_eyelid_line',
+            108: 'l_midpoint_5_of_upper_eyelid_line',
+            109: 'l_centerpoint_of_upper_eyelid_line',
+            110: 'l_midpoint_4_of_upper_eyelid_line',
+            111: 'l_midpoint_1_of_upper_eyelid_line',
+            112: 'l_midpoint_3_of_upper_eyelid_line',
+            113: 'l_midpoint_6_of_upper_crease_line',
+            114: 'l_midpoint_2_of_upper_crease_line',
+            115: 'l_midpoint_5_of_upper_crease_line',
+            116: 'l_centerpoint_of_upper_crease_line',
+            117: 'l_midpoint_4_of_upper_crease_line',
+            118: 'l_midpoint_1_of_upper_crease_line',
+            119: 'l_midpoint_3_of_upper_crease_line',
+            120: 'r_inner_end_of_upper_lash_line',
+            121: 'r_outer_end_of_upper_lash_line',
+            122: 'r_centerpoint_of_upper_lash_line',
+            123: 'r_midpoint_1_of_upper_lash_line',
+            124: 'r_midpoint_2_of_upper_lash_line',
+            125: 'r_midpoint_3_of_upper_lash_line',
+            126: 'r_midpoint_4_of_upper_lash_line',
+            127: 'r_midpoint_5_of_upper_lash_line',
+            128: 'r_midpoint_6_of_upper_lash_line',
+            129: 'r_outer_end_of_upper_eyelid_line',
+            130: 'r_midpoint_3_of_upper_eyelid_line',
+            131: 'r_midpoint_1_of_upper_eyelid_line',
+            132: 'r_midpoint_4_of_upper_eyelid_line',
+            133: 'r_centerpoint_of_upper_eyelid_line',
+            134: 'r_midpoint_5_of_upper_eyelid_line',
+            135: 'r_midpoint_2_of_upper_eyelid_line',
+            136: 'r_midpoint_6_of_upper_eyelid_line',
+            137: 'r_midpoint_3_of_upper_crease_line',
+            138: 'r_midpoint_1_of_upper_crease_line',
+            139: 'r_midpoint_4_of_upper_crease_line',
+            140: 'r_centerpoint_of_upper_crease_line',
+            141: 'r_midpoint_5_of_upper_crease_line',
+            142: 'r_midpoint_2_of_upper_crease_line',
+            143: 'r_midpoint_6_of_upper_crease_line',
+            144: 'l_inner_end_of_lower_lash_line',
+            145: 'l_outer_end_of_lower_lash_line',
+            146: 'l_centerpoint_of_lower_lash_line',
+            147: 'l_midpoint_2_of_lower_lash_line',
+            148: 'l_midpoint_1_of_lower_lash_line',
+            149: 'l_midpoint_6_of_lower_lash_line',
+            150: 'l_midpoint_5_of_lower_lash_line',
+            151: 'l_midpoint_4_of_lower_lash_line',
+            152: 'l_midpoint_3_of_lower_lash_line',
+            153: 'l_outer_end_of_lower_eyelid_line',
+            154: 'l_midpoint_6_of_lower_eyelid_line',
+            155: 'l_midpoint_2_of_lower_eyelid_line',
+            156: 'l_midpoint_5_of_lower_eyelid_line',
+            157: 'l_centerpoint_of_lower_eyelid_line',
+            158: 'l_midpoint_4_of_lower_eyelid_line',
+            159: 'l_midpoint_1_of_lower_eyelid_line',
+            160: 'l_midpoint_3_of_lower_eyelid_line',
+            161: 'r_inner_end_of_lower_lash_line',
+            162: 'r_outer_end_of_lower_lash_line',
+            163: 'r_centerpoint_of_lower_lash_line',
+            164: 'r_midpoint_1_of_lower_lash_line',
+            165: 'r_midpoint_2_of_lower_lash_line',
+            166: 'r_midpoint_3_of_lower_lash_line',
+            167: 'r_midpoint_4_of_lower_lash_line',
+            168: 'r_midpoint_5_of_lower_lash_line',
+            169: 'r_midpoint_6_of_lower_lash_line',
+            170: 'r_outer_end_of_lower_eyelid_line',
+            171: 'r_midpoint_3_of_lower_eyelid_line',
+            172: 'r_midpoint_1_of_lower_eyelid_line',
+            173: 'r_midpoint_4_of_lower_eyelid_line',
+            174: 'r_centerpoint_of_lower_eyelid_line',
+            175: 'r_midpoint_5_of_lower_eyelid_line',
+            176: 'r_midpoint_2_of_lower_eyelid_line',
+            177: 'r_midpoint_6_of_lower_eyelid_line',
+            178: 'tip_of_nose',
+            179: 'bottom_center_of_nose',
+            180: 'r_outer_corner_of_nose',
+            181: 'l_outer_corner_of_nose',
+            182: 'inner_corner_of_r_nostril',
+            183: 'outer_corner_of_r_nostril',
+            184: 'upper_corner_of_r_nostril',
+            185: 'inner_corner_of_l_nostril',
+            186: 'outer_corner_of_l_nostril',
+            187: 'upper_corner_of_l_nostril',
+            188: 'r_outer_corner_of_mouth',
+            189: 'l_outer_corner_of_mouth',
+            190: 'center_of_cupid_bow',
+            191: 'center_of_lower_outer_lip',
+            192: 'midpoint_1_of_upper_outer_lip',
+            193: 'midpoint_2_of_upper_outer_lip',
+            194: 'midpoint_1_of_lower_outer_lip',
+            195: 'midpoint_2_of_lower_outer_lip',
+            196: 'midpoint_3_of_upper_outer_lip',
+            197: 'midpoint_4_of_upper_outer_lip',
+            198: 'midpoint_5_of_upper_outer_lip',
+            199: 'midpoint_6_of_upper_outer_lip',
+            200: 'midpoint_3_of_lower_outer_lip',
+            201: 'midpoint_4_of_lower_outer_lip',
+            202: 'midpoint_5_of_lower_outer_lip',
+            203: 'midpoint_6_of_lower_outer_lip',
+            204: 'r_inner_corner_of_mouth',
+            205: 'l_inner_corner_of_mouth',
+            206: 'center_of_upper_inner_lip',
+            207: 'center_of_lower_inner_lip',
+            208: 'midpoint_1_of_upper_inner_lip',
+            209: 'midpoint_2_of_upper_inner_lip',
+            210: 'midpoint_1_of_lower_inner_lip',
+            211: 'midpoint_2_of_lower_inner_lip',
+            212: 'midpoint_3_of_upper_inner_lip',
+            213: 'midpoint_4_of_upper_inner_lip',
+            214: 'midpoint_5_of_upper_inner_lip',
+            215: 'midpoint_6_of_upper_inner_lip',
+            216: 'midpoint_3_of_lower_inner_lip',
+            217: 'midpoint_4_of_lower_inner_lip',
+            218: 'midpoint_5_of_lower_inner_lip',
+            219: 'midpoint_6_of_lower_inner_lip',
+            220: 'teeth',
+            221: 'teeth',
+            222: 'teeth',
+            223: 'teeth',
+            224: 'teeth',
+            225: 'teeth',
+            226: 'teeth',
+            227: 'teeth',
+            228: 'teeth',
+            229: 'teeth',
+            230: 'teeth',
+            231: 'teeth',
+            232: 'teeth',
+            233: 'teeth',
+            234: 'teeth',
+            235: 'teeth',
+            236: 'teeth',
+            237: 'teeth',
+            238: 'teeth',
+            239: 'teeth',
+            240: 'teeth',
+            241: 'teeth',
+            242: 'teeth',
+            243: 'teeth',
+            244: 'teeth',
+            245: 'teeth',
+            246: 'teeth',
+            247: 'teeth',
+            248: 'teeth',
+            249: 'teeth',
+            250: 'teeth',
+            251: 'teeth',
+            252: 'teeth',
+            253: 'teeth',
+            254: 'teeth',
+            255: 'teeth',
+            256: 'l_top_end_of_inferior_crus',
+            257: 'l_top_end_of_superior_crus',
+            258: 'l_start_of_antihelix',
+            259: 'l_end_of_antihelix',
+            260: 'l_midpoint_1_of_antihelix',
+            261: 'l_midpoint_1_of_inferior_crus',
+            262: 'l_midpoint_2_of_antihelix',
+            263: 'l_midpoint_3_of_antihelix',
+            264: 'l_point_1_of_inner_helix',
+            265: 'l_point_2_of_inner_helix',
+            266: 'l_point_3_of_inner_helix',
+            267: 'l_point_4_of_inner_helix',
+            268: 'l_point_5_of_inner_helix',
+            269: 'l_point_6_of_inner_helix',
+            270: 'l_point_7_of_inner_helix',
+            271: 'l_highest_point_of_antitragus',
+            272: 'l_bottom_point_of_tragus',
+            273: 'l_protruding_point_of_tragus',
+            274: 'l_top_point_of_tragus',
+            275: 'l_start_point_of_crus_of_helix',
+            276: 'l_deepest_point_of_concha',
+            277: 'l_tip_of_ear_lobe',
+            278: 'l_midpoint_between_22_15',
+            279: 'l_bottom_connecting_point_of_ear_lobe',
+            280: 'l_top_connecting_point_of_helix',
+            281: 'l_point_8_of_inner_helix',
+            282: 'r_top_end_of_inferior_crus',
+            283: 'r_top_end_of_superior_crus',
+            284: 'r_start_of_antihelix',
+            285: 'r_end_of_antihelix',
+            286: 'r_midpoint_1_of_antihelix',
+            287: 'r_midpoint_1_of_inferior_crus',
+            288: 'r_midpoint_2_of_antihelix',
+            289: 'r_midpoint_3_of_antihelix',
+            290: 'r_point_1_of_inner_helix',
+            291: 'r_point_8_of_inner_helix',
+            292: 'r_point_3_of_inner_helix',
+            293: 'r_point_4_of_inner_helix',
+            294: 'r_point_5_of_inner_helix',
+            295: 'r_point_6_of_inner_helix',
+            296: 'r_point_7_of_inner_helix',
+            297: 'r_highest_point_of_antitragus',
+            298: 'r_bottom_point_of_tragus',
+            299: 'r_protruding_point_of_tragus',
+            300: 'r_top_point_of_tragus',
+            301: 'r_start_point_of_crus_of_helix',
+            302: 'r_deepest_point_of_concha',
+            303: 'r_tip_of_ear_lobe',
+            304: 'r_midpoint_between_22_15',
+            305: 'r_bottom_connecting_point_of_ear_lobe',
+            306: 'r_top_connecting_point_of_helix',
+            307: 'r_point_2_of_inner_helix',
+            308: 'l_center_of_iris',
+            309: 'l_border_of_iris_3',
+            310: 'l_border_of_iris_midpoint_1',
+            311: 'l_border_of_iris_12',
+            312: 'l_border_of_iris_midpoint_4',
+            313: 'l_border_of_iris_9',
+            314: 'l_border_of_iris_midpoint_3',
+            315: 'l_border_of_iris_6',
+            316: 'l_border_of_iris_midpoint_2',
+            317: 'r_center_of_iris',
+            318: 'r_border_of_iris_3',
+            319: 'r_border_of_iris_midpoint_1',
+            320: 'r_border_of_iris_12',
+            321: 'r_border_of_iris_midpoint_4',
+            322: 'r_border_of_iris_9',
+            323: 'r_border_of_iris_midpoint_3',
+            324: 'r_border_of_iris_6',
+            325: 'r_border_of_iris_midpoint_2',
+            326: 'l_center_of_pupil',
+            327: 'l_border_of_pupil_3',
+            328: 'l_border_of_pupil_midpoint_1',
+            329: 'l_border_of_pupil_12',
+            330: 'l_border_of_pupil_midpoint_4',
+            331: 'l_border_of_pupil_9',
+            332: 'l_border_of_pupil_midpoint_3',
+            333: 'l_border_of_pupil_6',
+            334: 'l_border_of_pupil_midpoint_2',
+            335: 'r_center_of_pupil',
+            336: 'r_border_of_pupil_3',
+            337: 'r_border_of_pupil_midpoint_1',
+            338: 'r_border_of_pupil_12',
+            339: 'r_border_of_pupil_midpoint_4',
+            340: 'r_border_of_pupil_9',
+            341: 'r_border_of_pupil_midpoint_3',
+            342: 'r_border_of_pupil_6',
+            343: 'r_border_of_pupil_midpoint_2',
+            },
+        keypoint_info={
+        0: dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
+        1: dict(name='left_eye', id=1, color=[51, 153, 255], type='upper', swap='right_eye'),
+        2: dict(name='right_eye', id=2, color=[51, 153, 255], type='upper', swap='left_eye'),
+        3: dict(name='left_ear', id=3, color=[51, 153, 255], type='upper', swap='right_ear'),
+        4: dict(name='right_ear', id=4, color=[51, 153, 255], type='upper', swap='left_ear'),
+        5: dict(name='left_shoulder', id=5, color=[51, 153, 255], type='upper', swap='right_shoulder'),
+        6: dict(name='right_shoulder', id=6, color=[51, 153, 255], type='upper', swap='left_shoulder'),
+        7: dict(name='left_elbow', id=7, color=[51, 153, 255], type='upper', swap='right_elbow'),
+        8: dict(name='right_elbow', id=8, color=[51, 153, 255], type='upper', swap='left_elbow'),
+        9: dict(name='left_hip', id=9, color=[51, 153, 255], type='lower', swap='right_hip'),
+        10: dict(name='right_hip', id=10, color=[51, 153, 255], type='lower', swap='left_hip'),
+        11: dict(name='left_knee', id=11, color=[51, 153, 255], type='lower', swap='right_knee'),
+        12: dict(name='right_knee', id=12, color=[51, 153, 255], type='lower', swap='left_knee'),
+        13: dict(name='left_ankle', id=13, color=[51, 153, 255], type='lower', swap='right_ankle'),
+        14: dict(name='right_ankle', id=14, color=[51, 153, 255], type='lower', swap='left_ankle'),
+
+        15: dict(name='left_big_toe', id=15, color=[51, 153, 255], type='lower', swap='right_big_toe'),
+        16: dict(name='left_small_toe', id=16, color=[51, 153, 255], type='lower', swap='right_small_toe'),
+        17: dict(name='left_heel', id=17, color=[51, 153, 255], type='lower', swap='right_heel'),
+        18: dict(name='right_big_toe', id=18, color=[51, 153, 255], type='lower', swap='left_big_toe'),
+        19: dict(name='right_small_toe', id=19, color=[51, 153, 255], type='lower', swap='left_small_toe'),
+        20: dict(name='right_heel', id=20, color=[51, 153, 255], type='lower', swap='left_heel'),
+
+        21: dict(name='right_thumb4', id=21, color=[51, 153, 255], type='upper', swap='left_thumb4'),
+        22: dict(name='right_thumb3', id=22, color=[51, 153, 255], type='upper', swap='left_thumb3'),
+        23: dict(name='right_thumb2', id=23, color=[51, 153, 255], type='upper', swap='left_thumb2'),
+        24: dict(name='right_thumb_third_joint', id=24, color=[51, 153, 255], type='upper', swap='left_thumb_third_joint'),
+
+        25: dict(name='right_forefinger4', id=25, color=[51, 153, 255], type='upper', swap='left_forefinger4'),
+        26: dict(name='right_forefinger3', id=26, color=[51, 153, 255], type='upper', swap='left_forefinger3'),
+        27: dict(name='right_forefinger2', id=27, color=[51, 153, 255], type='upper', swap='left_forefinger2'),
+        28: dict(name='right_forefinger_third_joint', id=28, color=[51, 153, 255], type='upper', swap='left_forefinger_third_joint'),
+
+        29: dict(name='right_middle_finger4', id=29, color=[51, 153, 255], type='upper', swap='left_middle_finger4'),
+        30: dict(name='right_middle_finger3', id=30, color=[51, 153, 255], type='upper', swap='left_middle_finger3'),
+        31: dict(name='right_middle_finger2', id=31, color=[51, 153, 255], type='upper', swap='left_middle_finger2'),
+        32: dict(name='right_middle_finger_third_joint', id=32, color=[51, 153, 255], type='upper', swap='left_middle_finger_third_joint'),
+
+        33: dict(name='right_ring_finger4', id=33, color=[51, 153, 255], type='upper', swap='left_ring_finger4'),
+        34: dict(name='right_ring_finger3', id=34, color=[51, 153, 255], type='upper', swap='left_ring_finger3'),
+        35: dict(name='right_ring_finger2', id=35, color=[51, 153, 255], type='upper', swap='left_ring_finger2'),
+        36: dict(name='right_ring_finger_third_joint', id=36, color=[51, 153, 255], type='upper', swap='left_ring_finger_third_joint'),
+
+        37: dict(name='right_pinky_finger4', id=37, color=[51, 153, 255], type='upper', swap='left_pinky_finger4'),
+        38: dict(name='right_pinky_finger3', id=38, color=[51, 153, 255], type='upper', swap='left_pinky_finger3'),
+        39: dict(name='right_pinky_finger2', id=39, color=[51, 153, 255], type='upper', swap='left_pinky_finger2'),
+        40: dict(name='right_pinky_finger_third_joint', id=40, color=[51, 153, 255], type='upper', swap='left_pinky_finger_third_joint'),
+
+        41: dict(name='right_wrist', id=41, color=[51, 153, 255], type='upper', swap='left_wrist'),
+
+        42: dict(name='left_thumb4', id=42, color=[51, 153, 255], type='upper', swap='right_thumb4'),
+        43: dict(name='left_thumb3', id=43, color=[51, 153, 255], type='upper', swap='right_thumb3'),
+        44: dict(name='left_thumb2', id=44, color=[51, 153, 255], type='upper', swap='right_thumb2'),
+        45: dict(name='left_thumb_third_joint', id=45, color=[51, 153, 255], type='upper', swap='right_thumb_third_joint'), ## doesnt match with wholebody
+
+        46: dict(name='left_forefinger4', id=46, color=[51, 153, 255], type='upper', swap='right_forefinger4'),
+        47: dict(name='left_forefinger3', id=47, color=[51, 153, 255], type='upper', swap='right_forefinger3'),
+        48: dict(name='left_forefinger2', id=48, color=[51, 153, 255], type='upper', swap='right_forefinger2'),
+        49: dict(name='left_forefinger_third_joint', id=49, color=[51, 153, 255], type='upper', swap='right_forefinger_third_joint'),
+
+        50: dict(name='left_middle_finger4', id=50, color=[51, 153, 255], type='upper', swap='right_middle_finger4'),
+        51: dict(name='left_middle_finger3', id=51, color=[51, 153, 255], type='upper', swap='right_middle_finger3'),
+        52: dict(name='left_middle_finger2', id=52, color=[51, 153, 255], type='upper', swap='right_middle_finger2'),
+        53: dict(name='left_middle_finger_third_joint', id=53, color=[51, 153, 255], type='upper', swap='right_middle_finger_third_joint'),
+
+        54: dict(name='left_ring_finger4', id=54, color=[51, 153, 255], type='upper', swap='right_ring_finger4'),
+        55: dict(name='left_ring_finger3', id=55, color=[51, 153, 255], type='upper', swap='right_ring_finger3'),
+        56: dict(name='left_ring_finger2', id=56, color=[51, 153, 255], type='upper', swap='right_ring_finger2'),
+        57: dict(name='left_ring_finger_third_joint', id=57, color=[51, 153, 255], type='upper', swap='right_ring_finger_third_joint'),
+
+        58: dict(name='left_pinky_finger4', id=58, color=[51, 153, 255], type='upper', swap='right_pinky_finger4'),
+        59: dict(name='left_pinky_finger3', id=59, color=[51, 153, 255], type='upper', swap='right_pinky_finger3'),
+        60: dict(name='left_pinky_finger2', id=60, color=[51, 153, 255], type='upper', swap='right_pinky_finger2'),
+        61: dict(name='left_pinky_finger_third_joint', id=61, color=[51, 153, 255], type='upper', swap='right_pinky_finger_third_joint'),
+
+        62: dict(name='left_wrist', id=62, color=[51, 153, 255], type='upper', swap='right_wrist'),
+
+        63: dict(name='left_olecranon', id=63, color=[51, 153, 255], type='', swap='right_olecranon'),
+        64: dict(name='right_olecranon', id=64, color=[51, 153, 255], type='', swap='left_olecranon'),
+        65: dict(name='left_cubital_fossa', id=65, color=[51, 153, 255], type='', swap='right_cubital_fossa'),
+        66: dict(name='right_cubital_fossa', id=66, color=[51, 153, 255], type='', swap='left_cubital_fossa'),
+        67: dict(name='left_acromion', id=67, color=[51, 153, 255], type='', swap='right_acromion'),
+        68: dict(name='right_acromion', id=68, color=[51, 153, 255], type='', swap='left_acromion'),
+        69: dict(name='neck', id=69, color=[51, 153, 255], type='', swap=''),
+
+        # Jaw line
+        70: dict(name='center_of_glabella', id=70, color=[255, 255, 255], type='', swap=''),
+        71: dict(name='tip_of_chin', id=71, color=[255, 255, 255], type='', swap=''),
+
+        # Right eyebrow
+        72: dict(name='upper_startpoint_of_r_eyebrow', id=72, color=[255, 255, 255], type='upper', swap='upper_startpoint_of_l_eyebrow'),
+        73: dict(name='end_of_r_eyebrow', id=73, color=[255, 255, 255], type='upper', swap='end_of_l_eyebrow'),
+        74: dict(name='upper_midpoint_1_of_r_eyebrow', id=74, color=[255, 255, 255], type='upper', swap='upper_midpoint_1_of_l_eyebrow'),
+        75: dict(name='upper_midpoint_2_of_r_eyebrow', id=75, color=[255, 255, 255], type='upper', swap='upper_midpoint_3_of_l_eyebrow'),
+        76: dict(name='upper_midpoint_3_of_r_eyebrow', id=76, color=[255, 255, 255], type='upper', swap='upper_midpoint_2_of_l_eyebrow'),
+
+        # Left eyebrow
+        77: dict(name='upper_startpoint_of_l_eyebrow', id=77, color=[255, 255, 255], type='upper', swap='upper_startpoint_of_r_eyebrow'),
+        78: dict(name='end_of_l_eyebrow', id=78, color=[255, 255, 255], type='upper', swap='end_of_r_eyebrow'),
+        79: dict(name='upper_midpoint_1_of_l_eyebrow', id=79, color=[255, 255, 255], type='upper', swap='upper_midpoint_1_of_r_eyebrow'),
+        80: dict(name='upper_midpoint_2_of_l_eyebrow', id=80, color=[255, 255, 255], type='upper', swap='upper_midpoint_3_of_r_eyebrow'),
+        81: dict(name='upper_midpoint_3_of_l_eyebrow', id=81, color=[255, 255, 255], type='upper', swap='upper_midpoint_2_of_r_eyebrow'),
+
+        # Nose
+        82: dict(name='center_of_nose_root', id=82, color=[255, 255, 255], type='upper', swap=''),
+        83: dict(name='tip_of_nose_bridge', id=83, color=[255, 255, 255], type='upper', swap=''),
+        84: dict(name='midpoint_1_of_nose_bridge', id=84, color=[255, 255, 255], type='upper', swap=''),
+        85: dict(name='midpoint_2_of_nose_bridge', id=85, color=[255, 255, 255], type='upper', swap=''),
+        86: dict(name='midpoint_3_of_nose_bridge', id=86, color=[255, 255, 255], type='upper', swap=''),
+        87: dict(name='center_of_labiomental_groove', id=87, color=[255, 255, 255], type='upper', swap=''),
+
+        # Right eye
+        88: dict(name='l_inner_end_of_upper_lash_line', id=88, color=[192, 64, 128], type='upper', swap='r_inner_end_of_upper_lash_line'),
+        89: dict(name='l_outer_end_of_upper_lash_line', id=89, color=[192, 64, 128], type='upper', swap='r_outer_end_of_upper_lash_line'),
+        90: dict(name='l_centerpoint_of_upper_lash_line', id=90, color=[192, 64, 128], type='upper', swap='r_centerpoint_of_upper_lash_line'),
+        91: dict(name='l_inner_end_of_lower_lash_line', id=91, color=[64, 192, 128], type='upper', swap='r_inner_end_of_lower_lash_line'),
+        92: dict(name='l_outer_end_of_lower_lash_line', id=92, color=[64, 192, 128], type='upper', swap='r_outer_end_of_lower_lash_line'),
+        93: dict(name='l_centerpoint_of_lower_lash_line', id=93, color=[64, 192, 128], type='upper', swap='r_centerpoint_of_lower_lash_line'),
+
+        # Left eye
+        94: dict(name='r_inner_end_of_upper_lash_line', id=94, color=[64, 32, 192], type='upper', swap='l_inner_end_of_upper_lash_line'),
+        95: dict(name='r_outer_end_of_upper_lash_line', id=95, color=[64, 32, 192], type='upper', swap='l_outer_end_of_upper_lash_line'),
+        96: dict(name='r_centerpoint_of_upper_lash_line', id=96, color=[64, 32, 192], type='upper', swap='l_centerpoint_of_upper_lash_line'),
+        97: dict(name='r_inner_end_of_lower_lash_line', id=97, color=[64, 192, 32], type='upper', swap='l_inner_end_of_lower_lash_line'),
+        98: dict(name='r_outer_end_of_lower_lash_line', id=98, color=[64, 192, 32], type='upper', swap='l_outer_end_of_lower_lash_line'),
+        99: dict(name='r_centerpoint_of_lower_lash_line', id=99, color=[64, 192, 32], type='upper', swap='l_centerpoint_of_lower_lash_line'),
+
+        # Mouth
+        100: dict(name='r_outer_corner_of_mouth', id=100, color=[192, 0, 0], type='upper', swap='l_outer_corner_of_mouth'),
+        101: dict(name='l_outer_corner_of_mouth', id=101, color=[192, 0, 0], type='upper', swap='r_outer_corner_of_mouth'),
+        102: dict(name='center_of_cupid_bow', id=102, color=[192, 0, 0], type='upper', swap=''),
+        103: dict(name='center_of_lower_outer_lip', id=103, color=[192, 0, 0], type='upper', swap=''),
+        104: dict(name='midpoint_1_of_upper_outer_lip', id=104, color=[192, 0, 0], type='upper', swap='midpoint_2_of_upper_outer_lip'),
+        105: dict(name='midpoint_2_of_upper_outer_lip', id=105, color=[192, 0, 0], type='upper', swap='midpoint_1_of_upper_outer_lip'),
+        106: dict(name='midpoint_1_of_lower_outer_lip', id=106, color=[192, 0, 0], type='upper', swap='midpoint_2_of_lower_outer_lip'),
+        107: dict(name='midpoint_2_of_lower_outer_lip', id=107, color=[192, 0, 0], type='upper', swap='midpoint_1_of_lower_outer_lip'),
+        108: dict(name='r_inner_corner_of_mouth', id=108, color=[0, 192, 192], type='upper', swap='l_inner_corner_of_mouth'),
+        109: dict(name='l_inner_corner_of_mouth', id=109, color=[0, 192, 192], type='upper', swap='r_inner_corner_of_mouth'),
+        110: dict(name='center_of_upper_inner_lip', id=110, color=[0, 192, 192], type='upper', swap=''),
+        111: dict(name='center_of_lower_inner_lip', id=111, color=[0, 192, 192], type='upper', swap=''),
+        112: dict(name='midpoint_1_of_upper_inner_lip', id=112, color=[0, 192, 192], type='upper', swap='midpoint_2_of_upper_inner_lip'),
+        113: dict(name='midpoint_2_of_upper_inner_lip', id=113, color=[0, 192, 192], type='upper', swap='midpoint_1_of_upper_inner_lip'),
+        114: dict(name='midpoint_1_of_lower_inner_lip', id=114, color=[0, 192, 192], type='upper', swap='midpoint_2_of_lower_inner_lip'),
+        115: dict(name='midpoint_2_of_lower_inner_lip', id=115, color=[0, 192, 192], type='upper', swap='midpoint_1_of_lower_inner_lip'),
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
+        5:
+        dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
+        6:
+        dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
+        7:
+        dict(
+            link=('left_shoulder', 'right_shoulder'),
+            id=7,
+            color=[51, 153, 255]),
+        8:
+        dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
+        9:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
+        13:
+        dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
+        14:
+        dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
+        15:
+        dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
+        16:
+        dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
+        17:
+        dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
+        18:
+        dict(
+            link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255]),
+        19:
+        dict(link=('left_ankle', 'left_big_toe'), id=19, color=[0, 255, 0]),
+        20:
+        dict(link=('left_ankle', 'left_small_toe'), id=20, color=[0, 255, 0]),
+        21:
+        dict(link=('left_ankle', 'left_heel'), id=21, color=[0, 255, 0]),
+        22:
+        dict(
+            link=('right_ankle', 'right_big_toe'), id=22, color=[255, 128, 0]),
+        23:
+        dict(
+            link=('right_ankle', 'right_small_toe'),
+            id=23,
+            color=[255, 128, 0]),
+        24:
+        dict(link=('right_ankle', 'right_heel'), id=24, color=[255, 128, 0]),
+        25:
+        dict(
+            link=('left_wrist', 'left_thumb_third_joint'), id=25, color=[255, 128,
+                                                                  0]),
+        26:
+        dict(link=('left_thumb_third_joint', 'left_thumb2'), id=26, color=[255, 128, 0]),
+        27:
+        dict(link=('left_thumb2', 'left_thumb3'), id=27, color=[255, 128, 0]),
+        28:
+        dict(link=('left_thumb3', 'left_thumb4'), id=28, color=[255, 128, 0]),
+        29:
+        dict(
+            link=('left_wrist', 'left_forefinger_third_joint'),
+            id=29,
+            color=[255, 153, 255]),
+        30:
+        dict(
+            link=('left_forefinger_third_joint', 'left_forefinger2'),
+            id=30,
+            color=[255, 153, 255]),
+        31:
+        dict(
+            link=('left_forefinger2', 'left_forefinger3'),
+            id=31,
+            color=[255, 153, 255]),
+        32:
+        dict(
+            link=('left_forefinger3', 'left_forefinger4'),
+            id=32,
+            color=[255, 153, 255]),
+        33:
+        dict(
+            link=('left_wrist', 'left_middle_finger_third_joint'),
+            id=33,
+            color=[102, 178, 255]),
+        34:
+        dict(
+            link=('left_middle_finger_third_joint', 'left_middle_finger2'),
+            id=34,
+            color=[102, 178, 255]),
+        35:
+        dict(
+            link=('left_middle_finger2', 'left_middle_finger3'),
+            id=35,
+            color=[102, 178, 255]),
+        36:
+        dict(
+            link=('left_middle_finger3', 'left_middle_finger4'),
+            id=36,
+            color=[102, 178, 255]),
+        37:
+        dict(
+            link=('left_wrist', 'left_ring_finger_third_joint'),
+            id=37,
+            color=[255, 51, 51]),
+        38:
+        dict(
+            link=('left_ring_finger_third_joint', 'left_ring_finger2'),
+            id=38,
+            color=[255, 51, 51]),
+        39:
+        dict(
+            link=('left_ring_finger2', 'left_ring_finger3'),
+            id=39,
+            color=[255, 51, 51]),
+        40:
+        dict(
+            link=('left_ring_finger3', 'left_ring_finger4'),
+            id=40,
+            color=[255, 51, 51]),
+        41:
+        dict(
+            link=('left_wrist', 'left_pinky_finger_third_joint'),
+            id=41,
+            color=[0, 255, 0]),
+        42:
+        dict(
+            link=('left_pinky_finger_third_joint', 'left_pinky_finger2'),
+            id=42,
+            color=[0, 255, 0]),
+        43:
+        dict(
+            link=('left_pinky_finger2', 'left_pinky_finger3'),
+            id=43,
+            color=[0, 255, 0]),
+        44:
+        dict(
+            link=('left_pinky_finger3', 'left_pinky_finger4'),
+            id=44,
+            color=[0, 255, 0]),
+        45:
+        dict(
+            link=('right_wrist', 'right_thumb_third_joint'),
+            id=45,
+            color=[255, 128, 0]),
+        46:
+        dict(
+            link=('right_thumb_third_joint', 'right_thumb2'), id=46, color=[255, 128, 0]),
+        47:
+        dict(
+            link=('right_thumb2', 'right_thumb3'), id=47, color=[255, 128, 0]),
+        48:
+        dict(
+            link=('right_thumb3', 'right_thumb4'), id=48, color=[255, 128, 0]),
+        49:
+        dict(
+            link=('right_wrist', 'right_forefinger_third_joint'),
+            id=49,
+            color=[255, 153, 255]),
+        50:
+        dict(
+            link=('right_forefinger_third_joint', 'right_forefinger2'),
+            id=50,
+            color=[255, 153, 255]),
+        51:
+        dict(
+            link=('right_forefinger2', 'right_forefinger3'),
+            id=51,
+            color=[255, 153, 255]),
+        52:
+        dict(
+            link=('right_forefinger3', 'right_forefinger4'),
+            id=52,
+            color=[255, 153, 255]),
+        53:
+        dict(
+            link=('right_wrist', 'right_middle_finger_third_joint'),
+            id=53,
+            color=[102, 178, 255]),
+        54:
+        dict(
+            link=('right_middle_finger_third_joint', 'right_middle_finger2'),
+            id=54,
+            color=[102, 178, 255]),
+        55:
+        dict(
+            link=('right_middle_finger2', 'right_middle_finger3'),
+            id=55,
+            color=[102, 178, 255]),
+        56:
+        dict(
+            link=('right_middle_finger3', 'right_middle_finger4'),
+            id=56,
+            color=[102, 178, 255]),
+        57:
+        dict(
+            link=('right_wrist', 'right_ring_finger_third_joint'),
+            id=57,
+            color=[255, 51, 51]),
+        58:
+        dict(
+            link=('right_ring_finger_third_joint', 'right_ring_finger2'),
+            id=58,
+            color=[255, 51, 51]),
+        59:
+        dict(
+            link=('right_ring_finger2', 'right_ring_finger3'),
+            id=59,
+            color=[255, 51, 51]),
+        60:
+        dict(
+            link=('right_ring_finger3', 'right_ring_finger4'),
+            id=60,
+            color=[255, 51, 51]),
+        61:
+        dict(
+            link=('right_wrist', 'right_pinky_finger_third_joint'),
+            id=61,
+            color=[0, 255, 0]),
+        62:
+        dict(
+            link=('right_pinky_finger_third_joint', 'right_pinky_finger2'),
+            id=62,
+            color=[0, 255, 0]),
+        63:
+        dict(
+            link=('right_pinky_finger2', 'right_pinky_finger3'),
+            id=63,
+            color=[0, 255, 0]),
+        64:
+        dict(
+            link=('right_pinky_finger3', 'right_pinky_finger4'),
+            id=64,
+            color=[0, 255, 0])
+    },
+    joint_weights=[1.] * 116,
+    body_keypoint_names=[
+        'nose',
+        'left_eye',
+        'right_eye',
+        'left_ear',
+        'right_ear',
+        'left_shoulder',
+        'right_shoulder',
+        'left_elbow',
+        'right_elbow',
+        'left_wrist',
+        'right_wrist',
+        'left_hip',
+        'right_hip',
+        'left_knee',
+        'right_knee',
+        'left_ankle',
+        'right_ankle',
+    ],
+    foot_keypoint_names=[
+        'left_big_toe',
+        'left_small_toe',
+        'left_heel',
+        'right_big_toe',
+        'right_small_toe',
+        'right_heel'
+    ],
+    left_hand_keypoint_names=[
+        'left_thumb4',
+        'left_thumb3',
+        'left_thumb2',
+        'left_thumb_third_joint',
+        'left_forefinger4',
+        'left_forefinger3',
+        'left_forefinger2',
+        'left_forefinger_third_joint',
+        'left_middle_finger4',
+        'left_middle_finger3',
+        'left_middle_finger2',
+        'left_middle_finger_third_joint',
+        'left_ring_finger4',
+        'left_ring_finger3',
+        'left_ring_finger2',
+        'left_ring_finger_third_joint',
+        'left_pinky_finger4',
+        'left_pinky_finger3',
+        'left_pinky_finger2',
+        'left_pinky_finger_third_joint'
+    ],
+    right_hand_keypoint_names=[
+        'right_thumb4',
+        'right_thumb3',
+        'right_thumb2',
+        'right_thumb_third_joint',
+        'right_forefinger4',
+        'right_forefinger3',
+        'right_forefinger2',
+        'right_forefinger_third_joint',
+        'right_middle_finger4',
+        'right_middle_finger3',
+        'right_middle_finger2',
+        'right_middle_finger_third_joint',
+        'right_ring_finger4',
+        'right_ring_finger3',
+        'right_ring_finger2',
+        'right_ring_finger_third_joint',
+        'right_pinky_finger4',
+        'right_pinky_finger3',
+        'right_pinky_finger2',
+        'right_pinky_finger_third_joint'
+    ],
+    ## 7 of them
+    extra_keypoint_names=[
+        'neck',
+        'left_olecranon',
+        'right_olecranon',
+        'left_cubital_fossa',
+        'right_cubital_fossa',
+        'left_acromion',
+        'right_acromion',
+    ],
+    face_keypoint_names = [
+        'center_of_glabella',
+        'tip_of_chin',
+        'upper_startpoint_of_r_eyebrow',
+        'end_of_r_eyebrow',
+        'upper_midpoint_1_of_r_eyebrow',
+        'upper_midpoint_2_of_r_eyebrow',
+        'upper_midpoint_3_of_r_eyebrow',
+        'upper_startpoint_of_l_eyebrow',
+        'end_of_l_eyebrow',
+        'upper_midpoint_1_of_l_eyebrow',
+        'upper_midpoint_2_of_l_eyebrow',
+        'upper_midpoint_3_of_l_eyebrow',
+        'center_of_nose_root',
+        'tip_of_nose_bridge',
+        'midpoint_1_of_nose_bridge',
+        'midpoint_2_of_nose_bridge',
+        'midpoint_3_of_nose_bridge',
+        'center_of_labiomental_groove',
+        'l_inner_end_of_upper_lash_line',
+        'l_outer_end_of_upper_lash_line',
+        'l_centerpoint_of_upper_lash_line',
+        'l_inner_end_of_lower_lash_line',
+        'l_outer_end_of_lower_lash_line',
+        'l_centerpoint_of_lower_lash_line',
+        'r_inner_end_of_upper_lash_line',
+        'r_outer_end_of_upper_lash_line',
+        'r_centerpoint_of_upper_lash_line',
+        'r_inner_end_of_lower_lash_line',
+        'r_outer_end_of_lower_lash_line',
+        'r_centerpoint_of_lower_lash_line',
+        'r_outer_corner_of_mouth',
+        'l_outer_corner_of_mouth',
+        'center_of_cupid_bow',
+        'center_of_lower_outer_lip',
+        'midpoint_1_of_upper_outer_lip',
+        'midpoint_2_of_upper_outer_lip',
+        'midpoint_1_of_lower_outer_lip',
+        'midpoint_2_of_lower_outer_lip',
+        'r_inner_corner_of_mouth',
+        'l_inner_corner_of_mouth',
+        'center_of_upper_inner_lip',
+        'center_of_lower_inner_lip',
+        'midpoint_1_of_upper_inner_lip',
+        'midpoint_2_of_upper_inner_lip',
+        'midpoint_1_of_lower_inner_lip',
+        'midpoint_2_of_lower_inner_lip'
+    ]
+)
+
+##------------------------------------------------------------------------------------------------------------------
+inverse_original_keypoint_info = {name: id for (id, name) in dataset_info['original_keypoint_info'].items()}
+
+## create index to original mapping
+dataset_info['idx_to_original_idx_mapping'] = {}
+for keypoint_index, keypoint_info in dataset_info['keypoint_info'].items():
+    keypoint_name = keypoint_info['name']
+    dataset_info['idx_to_original_idx_mapping'][keypoint_index] = inverse_original_keypoint_info[keypoint_name]
+
+##------------------------------------------------------------------------------------------------------------------
+## reconfigure in the order of coco_whole_body
+coco_wholebody_keypoint_info = {keypoint_info['name']: keypoint_info for (keypoint_index, keypoint_info) in coco_wholebody_info['keypoint_info'].items()}
+coco_wholebody_to_goliath_mapping = {} ## coco_wholebody_index to goliath_index
+coco_wholebody_to_goliath_keypoint_info = {}
+
+## find out common keypoints between goliath and coco_whole_body
+for (keypoint_index, keypoint_info) in dataset_info['keypoint_info'].items():
+    keypoint_name = keypoint_info['name']
+    keypoint_index_ = keypoint_info['id']
+    assert(keypoint_index == keypoint_index_)
+
+    if keypoint_name in coco_wholebody_keypoint_info.keys():
+        coco_wholebody_to_goliath_keypoint_info[keypoint_name] = coco_wholebody_keypoint_info[keypoint_name]
+        coco_wholebody_to_goliath_mapping[coco_wholebody_keypoint_info[keypoint_name]['id']] = keypoint_info['id']
+
+dataset_info['coco_wholebody_to_goliath_mapping'] = coco_wholebody_to_goliath_mapping ## store the cocowholebody indices
+dataset_info['coco_wholebody_to_goliath_keypoint_info'] = coco_wholebody_to_goliath_keypoint_info
+
+##------------------------------------------------------------------------------------------------------------------
+coco_wholebody_sigmas = {}
+
+## compute the coco_wholebody_sigmas
+for keypoint_index, keypoint_info in coco_wholebody_info['keypoint_info'].items():
+    coco_wholebody_sigmas[keypoint_info['name']] = coco_wholebody_info['sigmas'][keypoint_info['id']]
+
+default_sigma = 0.010 ## for mostly face keypoints
+dataset_info['sigmas'] = [default_sigma]*len(dataset_info['keypoint_info'])
+
+## we copy sigmas from coco_wholebody. Rest are assigned as below:
+custom_sigmas = {
+    'left_thumb_third_joint': 0.022,
+    'left_forefinger_third_joint': 0.026,
+    'left_middle_finger_third_joint': 0.018,
+    'left_ring_finger_third_joint': 0.017,
+    'left_pinky_finger_third_joint': 0.02,
+    'right_thumb_third_joint': 0.022,
+    'right_forefinger_third_joint': 0.026,
+    'right_middle_finger_third_joint': 0.018,
+    'right_ring_finger_third_joint': 0.017,
+    'right_pinky_finger_third_joint': 0.02,
+    'neck': 0.079, ## same as shoulder
+    'left_olecranon': 0.072, ## same as elbow
+    'right_olecranon': 0.072, ## same as elbow
+    'left_cubital_fossa': 0.072, ## same as elbow
+    'right_cubital_fossa': 0.072, ## same as elbow
+    'left_acromion': 0.079, ## same as shoulder
+    'right_acromion': 0.079, ## same as shoulder
+}
+
+## copy custom sigmas
+for keypoint_name, sigma in custom_sigmas.items():
+    keypoint_id = -1
+
+    ## search for keypoint id from keypoint name
+    for keypoint_id_ in dataset_info['keypoint_info'].keys():
+        if dataset_info['keypoint_info'][keypoint_id_]['name'] == keypoint_name:
+            keypoint_id = keypoint_id_
+            break
+
+    if keypoint_id != -1:
+        keypoint_info = dataset_info['keypoint_info'][keypoint_id]
+        assert(keypoint_info['name'] == keypoint_name)
+        assert(keypoint_info['id'] == keypoint_id)
+        dataset_info['sigmas'][keypoint_info['id']] = sigma
+
+## copy coco_wholebody sigmas
+for keypoint_index, keypoint_info in dataset_info['keypoint_info'].items():
+    if keypoint_info['name'] in coco_wholebody_sigmas.keys():
+        dataset_info['sigmas'][keypoint_info['id']] = coco_wholebody_sigmas[keypoint_info['name']]

configs/_base_/datasets/h36m.py

@@ -0,0 +1,158 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='h36m',
+    paper_info=dict(
+        author='Ionescu, Catalin and Papava, Dragos and '
+        'Olaru, Vlad and Sminchisescu, Cristian',
+        title='Human3.6M: Large Scale Datasets and Predictive '
+        'Methods for 3D Human Sensing in Natural Environments',
+        container='IEEE Transactions on Pattern Analysis and '
+        'Machine Intelligence',
+        year='2014',
+        homepage='http://vision.imar.ro/human3.6m/description.php',
+    ),
+    keypoint_info={
+        0:
+        dict(name='root', id=0, color=[51, 153, 255], type='lower', swap=''),
+        1:
+        dict(
+            name='right_hip',
+            id=1,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        2:
+        dict(
+            name='right_knee',
+            id=2,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        3:
+        dict(
+            name='right_foot',
+            id=3,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_foot'),
+        4:
+        dict(
+            name='left_hip',
+            id=4,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        5:
+        dict(
+            name='left_knee',
+            id=5,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        6:
+        dict(
+            name='left_foot',
+            id=6,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_foot'),
+        7:
+        dict(name='spine', id=7, color=[51, 153, 255], type='upper', swap=''),
+        8:
+        dict(name='thorax', id=8, color=[51, 153, 255], type='upper', swap=''),
+        9:
+        dict(
+            name='neck_base',
+            id=9,
+            color=[51, 153, 255],
+            type='upper',
+            swap=''),
+        10:
+        dict(name='head', id=10, color=[51, 153, 255], type='upper', swap=''),
+        11:
+        dict(
+            name='left_shoulder',
+            id=11,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        12:
+        dict(
+            name='left_elbow',
+            id=12,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        13:
+        dict(
+            name='left_wrist',
+            id=13,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        14:
+        dict(
+            name='right_shoulder',
+            id=14,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        15:
+        dict(
+            name='right_elbow',
+            id=15,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        16:
+        dict(
+            name='right_wrist',
+            id=16,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist')
+    },
+    skeleton_info={
+        0:
+        dict(link=('root', 'left_hip'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_hip', 'left_knee'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('left_knee', 'left_foot'), id=2, color=[0, 255, 0]),
+        3:
+        dict(link=('root', 'right_hip'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('right_hip', 'right_knee'), id=4, color=[255, 128, 0]),
+        5:
+        dict(link=('right_knee', 'right_foot'), id=5, color=[255, 128, 0]),
+        6:
+        dict(link=('root', 'spine'), id=6, color=[51, 153, 255]),
+        7:
+        dict(link=('spine', 'thorax'), id=7, color=[51, 153, 255]),
+        8:
+        dict(link=('thorax', 'neck_base'), id=8, color=[51, 153, 255]),
+        9:
+        dict(link=('neck_base', 'head'), id=9, color=[51, 153, 255]),
+        10:
+        dict(link=('thorax', 'left_shoulder'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('left_shoulder', 'left_elbow'), id=11, color=[0, 255, 0]),
+        12:
+        dict(link=('left_elbow', 'left_wrist'), id=12, color=[0, 255, 0]),
+        13:
+        dict(link=('thorax', 'right_shoulder'), id=13, color=[255, 128, 0]),
+        14:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=14, color=[255, 128,
+                                                                  0]),
+        15:
+        dict(link=('right_elbow', 'right_wrist'), id=15, color=[255, 128, 0])
+    },
+    joint_weights=[1.] * 17,
+    sigmas=[],
+    stats_info=dict(bbox_center=(528., 427.), bbox_scale=400.))

configs/_base_/datasets/halpe.py

@@ -0,0 +1,1163 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='halpe',
+    paper_info=dict(
+        author='Li, Yong-Lu and Xu, Liang and Liu, Xinpeng and Huang, Xijie'
+        ' and Xu, Yue and Wang, Shiyi and Fang, Hao-Shu'
+        ' and Ma, Ze and Chen, Mingyang and Lu, Cewu',
+        title='PaStaNet: Toward Human Activity Knowledge Engine',
+        container='CVPR',
+        year='2020',
+        homepage='https://github.com/Fang-Haoshu/Halpe-FullBody/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
+        1:
+        dict(
+            name='left_eye',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_eye'),
+        2:
+        dict(
+            name='right_eye',
+            id=2,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_eye'),
+        3:
+        dict(
+            name='left_ear',
+            id=3,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_ear'),
+        4:
+        dict(
+            name='right_ear',
+            id=4,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_ear'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='right_shoulder',
+            id=6,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        7:
+        dict(
+            name='left_elbow',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        8:
+        dict(
+            name='right_elbow',
+            id=8,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        9:
+        dict(
+            name='left_wrist',
+            id=9,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='right_hip',
+            id=12,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        13:
+        dict(
+            name='left_knee',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        14:
+        dict(
+            name='right_knee',
+            id=14,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        15:
+        dict(
+            name='left_ankle',
+            id=15,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        16:
+        dict(
+            name='right_ankle',
+            id=16,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle'),
+        17:
+        dict(name='head', id=17, color=[255, 128, 0], type='upper', swap=''),
+        18:
+        dict(name='neck', id=18, color=[255, 128, 0], type='upper', swap=''),
+        19:
+        dict(name='hip', id=19, color=[255, 128, 0], type='lower', swap=''),
+        20:
+        dict(
+            name='left_big_toe',
+            id=20,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_big_toe'),
+        21:
+        dict(
+            name='right_big_toe',
+            id=21,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_big_toe'),
+        22:
+        dict(
+            name='left_small_toe',
+            id=22,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_small_toe'),
+        23:
+        dict(
+            name='right_small_toe',
+            id=23,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_small_toe'),
+        24:
+        dict(
+            name='left_heel',
+            id=24,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_heel'),
+        25:
+        dict(
+            name='right_heel',
+            id=25,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_heel'),
+        26:
+        dict(
+            name='face-0',
+            id=26,
+            color=[255, 255, 255],
+            type='',
+            swap='face-16'),
+        27:
+        dict(
+            name='face-1',
+            id=27,
+            color=[255, 255, 255],
+            type='',
+            swap='face-15'),
+        28:
+        dict(
+            name='face-2',
+            id=28,
+            color=[255, 255, 255],
+            type='',
+            swap='face-14'),
+        29:
+        dict(
+            name='face-3',
+            id=29,
+            color=[255, 255, 255],
+            type='',
+            swap='face-13'),
+        30:
+        dict(
+            name='face-4',
+            id=30,
+            color=[255, 255, 255],
+            type='',
+            swap='face-12'),
+        31:
+        dict(
+            name='face-5',
+            id=31,
+            color=[255, 255, 255],
+            type='',
+            swap='face-11'),
+        32:
+        dict(
+            name='face-6',
+            id=32,
+            color=[255, 255, 255],
+            type='',
+            swap='face-10'),
+        33:
+        dict(
+            name='face-7',
+            id=33,
+            color=[255, 255, 255],
+            type='',
+            swap='face-9'),
+        34:
+        dict(name='face-8', id=34, color=[255, 255, 255], type='', swap=''),
+        35:
+        dict(
+            name='face-9',
+            id=35,
+            color=[255, 255, 255],
+            type='',
+            swap='face-7'),
+        36:
+        dict(
+            name='face-10',
+            id=36,
+            color=[255, 255, 255],
+            type='',
+            swap='face-6'),
+        37:
+        dict(
+            name='face-11',
+            id=37,
+            color=[255, 255, 255],
+            type='',
+            swap='face-5'),
+        38:
+        dict(
+            name='face-12',
+            id=38,
+            color=[255, 255, 255],
+            type='',
+            swap='face-4'),
+        39:
+        dict(
+            name='face-13',
+            id=39,
+            color=[255, 255, 255],
+            type='',
+            swap='face-3'),
+        40:
+        dict(
+            name='face-14',
+            id=40,
+            color=[255, 255, 255],
+            type='',
+            swap='face-2'),
+        41:
+        dict(
+            name='face-15',
+            id=41,
+            color=[255, 255, 255],
+            type='',
+            swap='face-1'),
+        42:
+        dict(
+            name='face-16',
+            id=42,
+            color=[255, 255, 255],
+            type='',
+            swap='face-0'),
+        43:
+        dict(
+            name='face-17',
+            id=43,
+            color=[255, 255, 255],
+            type='',
+            swap='face-26'),
+        44:
+        dict(
+            name='face-18',
+            id=44,
+            color=[255, 255, 255],
+            type='',
+            swap='face-25'),
+        45:
+        dict(
+            name='face-19',
+            id=45,
+            color=[255, 255, 255],
+            type='',
+            swap='face-24'),
+        46:
+        dict(
+            name='face-20',
+            id=46,
+            color=[255, 255, 255],
+            type='',
+            swap='face-23'),
+        47:
+        dict(
+            name='face-21',
+            id=47,
+            color=[255, 255, 255],
+            type='',
+            swap='face-22'),
+        48:
+        dict(
+            name='face-22',
+            id=48,
+            color=[255, 255, 255],
+            type='',
+            swap='face-21'),
+        49:
+        dict(
+            name='face-23',
+            id=49,
+            color=[255, 255, 255],
+            type='',
+            swap='face-20'),
+        50:
+        dict(
+            name='face-24',
+            id=50,
+            color=[255, 255, 255],
+            type='',
+            swap='face-19'),
+        51:
+        dict(
+            name='face-25',
+            id=51,
+            color=[255, 255, 255],
+            type='',
+            swap='face-18'),
+        52:
+        dict(
+            name='face-26',
+            id=52,
+            color=[255, 255, 255],
+            type='',
+            swap='face-17'),
+        53:
+        dict(name='face-27', id=53, color=[255, 255, 255], type='', swap=''),
+        54:
+        dict(name='face-28', id=54, color=[255, 255, 255], type='', swap=''),
+        55:
+        dict(name='face-29', id=55, color=[255, 255, 255], type='', swap=''),
+        56:
+        dict(name='face-30', id=56, color=[255, 255, 255], type='', swap=''),
+        57:
+        dict(
+            name='face-31',
+            id=57,
+            color=[255, 255, 255],
+            type='',
+            swap='face-35'),
+        58:
+        dict(
+            name='face-32',
+            id=58,
+            color=[255, 255, 255],
+            type='',
+            swap='face-34'),
+        59:
+        dict(name='face-33', id=59, color=[255, 255, 255], type='', swap=''),
+        60:
+        dict(
+            name='face-34',
+            id=60,
+            color=[255, 255, 255],
+            type='',
+            swap='face-32'),
+        61:
+        dict(
+            name='face-35',
+            id=61,
+            color=[255, 255, 255],
+            type='',
+            swap='face-31'),
+        62:
+        dict(
+            name='face-36',
+            id=62,
+            color=[255, 255, 255],
+            type='',
+            swap='face-45'),
+        63:
+        dict(
+            name='face-37',
+            id=63,
+            color=[255, 255, 255],
+            type='',
+            swap='face-44'),
+        64:
+        dict(
+            name='face-38',
+            id=64,
+            color=[255, 255, 255],
+            type='',
+            swap='face-43'),
+        65:
+        dict(
+            name='face-39',
+            id=65,
+            color=[255, 255, 255],
+            type='',
+            swap='face-42'),
+        66:
+        dict(
+            name='face-40',
+            id=66,
+            color=[255, 255, 255],
+            type='',
+            swap='face-47'),
+        67:
+        dict(
+            name='face-41',
+            id=67,
+            color=[255, 255, 255],
+            type='',
+            swap='face-46'),
+        68:
+        dict(
+            name='face-42',
+            id=68,
+            color=[255, 255, 255],
+            type='',
+            swap='face-39'),
+        69:
+        dict(
+            name='face-43',
+            id=69,
+            color=[255, 255, 255],
+            type='',
+            swap='face-38'),
+        70:
+        dict(
+            name='face-44',
+            id=70,
+            color=[255, 255, 255],
+            type='',
+            swap='face-37'),
+        71:
+        dict(
+            name='face-45',
+            id=71,
+            color=[255, 255, 255],
+            type='',
+            swap='face-36'),
+        72:
+        dict(
+            name='face-46',
+            id=72,
+            color=[255, 255, 255],
+            type='',
+            swap='face-41'),
+        73:
+        dict(
+            name='face-47',
+            id=73,
+            color=[255, 255, 255],
+            type='',
+            swap='face-40'),
+        74:
+        dict(
+            name='face-48',
+            id=74,
+            color=[255, 255, 255],
+            type='',
+            swap='face-54'),
+        75:
+        dict(
+            name='face-49',
+            id=75,
+            color=[255, 255, 255],
+            type='',
+            swap='face-53'),
+        76:
+        dict(
+            name='face-50',
+            id=76,
+            color=[255, 255, 255],
+            type='',
+            swap='face-52'),
+        77:
+        dict(name='face-51', id=77, color=[255, 255, 255], type='', swap=''),
+        78:
+        dict(
+            name='face-52',
+            id=78,
+            color=[255, 255, 255],
+            type='',
+            swap='face-50'),
+        79:
+        dict(
+            name='face-53',
+            id=79,
+            color=[255, 255, 255],
+            type='',
+            swap='face-49'),
+        80:
+        dict(
+            name='face-54',
+            id=80,
+            color=[255, 255, 255],
+            type='',
+            swap='face-48'),
+        81:
+        dict(
+            name='face-55',
+            id=81,
+            color=[255, 255, 255],
+            type='',
+            swap='face-59'),
+        82:
+        dict(
+            name='face-56',
+            id=82,
+            color=[255, 255, 255],
+            type='',
+            swap='face-58'),
+        83:
+        dict(name='face-57', id=83, color=[255, 255, 255], type='', swap=''),
+        84:
+        dict(
+            name='face-58',
+            id=84,
+            color=[255, 255, 255],
+            type='',
+            swap='face-56'),
+        85:
+        dict(
+            name='face-59',
+            id=85,
+            color=[255, 255, 255],
+            type='',
+            swap='face-55'),
+        86:
+        dict(
+            name='face-60',
+            id=86,
+            color=[255, 255, 255],
+            type='',
+            swap='face-64'),
+        87:
+        dict(
+            name='face-61',
+            id=87,
+            color=[255, 255, 255],
+            type='',
+            swap='face-63'),
+        88:
+        dict(name='face-62', id=88, color=[255, 255, 255], type='', swap=''),
+        89:
+        dict(
+            name='face-63',
+            id=89,
+            color=[255, 255, 255],
+            type='',
+            swap='face-61'),
+        90:
+        dict(
+            name='face-64',
+            id=90,
+            color=[255, 255, 255],
+            type='',
+            swap='face-60'),
+        91:
+        dict(
+            name='face-65',
+            id=91,
+            color=[255, 255, 255],
+            type='',
+            swap='face-67'),
+        92:
+        dict(name='face-66', id=92, color=[255, 255, 255], type='', swap=''),
+        93:
+        dict(
+            name='face-67',
+            id=93,
+            color=[255, 255, 255],
+            type='',
+            swap='face-65'),
+        94:
+        dict(
+            name='left_hand_root',
+            id=94,
+            color=[255, 255, 255],
+            type='',
+            swap='right_hand_root'),
+        95:
+        dict(
+            name='left_thumb1',
+            id=95,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb1'),
+        96:
+        dict(
+            name='left_thumb2',
+            id=96,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb2'),
+        97:
+        dict(
+            name='left_thumb3',
+            id=97,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb3'),
+        98:
+        dict(
+            name='left_thumb4',
+            id=98,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb4'),
+        99:
+        dict(
+            name='left_forefinger1',
+            id=99,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger1'),
+        100:
+        dict(
+            name='left_forefinger2',
+            id=100,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger2'),
+        101:
+        dict(
+            name='left_forefinger3',
+            id=101,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger3'),
+        102:
+        dict(
+            name='left_forefinger4',
+            id=102,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger4'),
+        103:
+        dict(
+            name='left_middle_finger1',
+            id=103,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger1'),
+        104:
+        dict(
+            name='left_middle_finger2',
+            id=104,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger2'),
+        105:
+        dict(
+            name='left_middle_finger3',
+            id=105,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger3'),
+        106:
+        dict(
+            name='left_middle_finger4',
+            id=106,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger4'),
+        107:
+        dict(
+            name='left_ring_finger1',
+            id=107,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger1'),
+        108:
+        dict(
+            name='left_ring_finger2',
+            id=108,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger2'),
+        109:
+        dict(
+            name='left_ring_finger3',
+            id=109,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger3'),
+        110:
+        dict(
+            name='left_ring_finger4',
+            id=110,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger4'),
+        111:
+        dict(
+            name='left_pinky_finger1',
+            id=111,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger1'),
+        112:
+        dict(
+            name='left_pinky_finger2',
+            id=112,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger2'),
+        113:
+        dict(
+            name='left_pinky_finger3',
+            id=113,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger3'),
+        114:
+        dict(
+            name='left_pinky_finger4',
+            id=114,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger4'),
+        115:
+        dict(
+            name='right_hand_root',
+            id=115,
+            color=[255, 255, 255],
+            type='',
+            swap='left_hand_root'),
+        116:
+        dict(
+            name='right_thumb1',
+            id=116,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb1'),
+        117:
+        dict(
+            name='right_thumb2',
+            id=117,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb2'),
+        118:
+        dict(
+            name='right_thumb3',
+            id=118,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb3'),
+        119:
+        dict(
+            name='right_thumb4',
+            id=119,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb4'),
+        120:
+        dict(
+            name='right_forefinger1',
+            id=120,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger1'),
+        121:
+        dict(
+            name='right_forefinger2',
+            id=121,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger2'),
+        122:
+        dict(
+            name='right_forefinger3',
+            id=122,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger3'),
+        123:
+        dict(
+            name='right_forefinger4',
+            id=123,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger4'),
+        124:
+        dict(
+            name='right_middle_finger1',
+            id=124,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger1'),
+        125:
+        dict(
+            name='right_middle_finger2',
+            id=125,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger2'),
+        126:
+        dict(
+            name='right_middle_finger3',
+            id=126,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger3'),
+        127:
+        dict(
+            name='right_middle_finger4',
+            id=127,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger4'),
+        128:
+        dict(
+            name='right_ring_finger1',
+            id=128,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger1'),
+        129:
+        dict(
+            name='right_ring_finger2',
+            id=129,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger2'),
+        130:
+        dict(
+            name='right_ring_finger3',
+            id=130,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger3'),
+        131:
+        dict(
+            name='right_ring_finger4',
+            id=131,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger4'),
+        132:
+        dict(
+            name='right_pinky_finger1',
+            id=132,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger1'),
+        133:
+        dict(
+            name='right_pinky_finger2',
+            id=133,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger2'),
+        134:
+        dict(
+            name='right_pinky_finger3',
+            id=134,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger3'),
+        135:
+        dict(
+            name='right_pinky_finger4',
+            id=135,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger4')
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('left_hip', 'hip'), id=2, color=[0, 255, 0]),
+        3:
+        dict(link=('right_ankle', 'right_knee'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('right_knee', 'right_hip'), id=4, color=[255, 128, 0]),
+        5:
+        dict(link=('right_hip', 'hip'), id=5, color=[255, 128, 0]),
+        6:
+        dict(link=('head', 'neck'), id=6, color=[51, 153, 255]),
+        7:
+        dict(link=('neck', 'hip'), id=7, color=[51, 153, 255]),
+        8:
+        dict(link=('neck', 'left_shoulder'), id=8, color=[0, 255, 0]),
+        9:
+        dict(link=('left_shoulder', 'left_elbow'), id=9, color=[0, 255, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('neck', 'right_shoulder'), id=11, color=[255, 128, 0]),
+        12:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=12, color=[255, 128,
+                                                                  0]),
+        13:
+        dict(link=('right_elbow', 'right_wrist'), id=13, color=[255, 128, 0]),
+        14:
+        dict(link=('left_eye', 'right_eye'), id=14, color=[51, 153, 255]),
+        15:
+        dict(link=('nose', 'left_eye'), id=15, color=[51, 153, 255]),
+        16:
+        dict(link=('nose', 'right_eye'), id=16, color=[51, 153, 255]),
+        17:
+        dict(link=('left_eye', 'left_ear'), id=17, color=[51, 153, 255]),
+        18:
+        dict(link=('right_eye', 'right_ear'), id=18, color=[51, 153, 255]),
+        19:
+        dict(link=('left_ear', 'left_shoulder'), id=19, color=[51, 153, 255]),
+        20:
+        dict(
+            link=('right_ear', 'right_shoulder'), id=20, color=[51, 153, 255]),
+        21:
+        dict(link=('left_ankle', 'left_big_toe'), id=21, color=[0, 255, 0]),
+        22:
+        dict(link=('left_ankle', 'left_small_toe'), id=22, color=[0, 255, 0]),
+        23:
+        dict(link=('left_ankle', 'left_heel'), id=23, color=[0, 255, 0]),
+        24:
+        dict(
+            link=('right_ankle', 'right_big_toe'), id=24, color=[255, 128, 0]),
+        25:
+        dict(
+            link=('right_ankle', 'right_small_toe'),
+            id=25,
+            color=[255, 128, 0]),
+        26:
+        dict(link=('right_ankle', 'right_heel'), id=26, color=[255, 128, 0]),
+        27:
+        dict(link=('left_wrist', 'left_thumb1'), id=27, color=[255, 128, 0]),
+        28:
+        dict(link=('left_thumb1', 'left_thumb2'), id=28, color=[255, 128, 0]),
+        29:
+        dict(link=('left_thumb2', 'left_thumb3'), id=29, color=[255, 128, 0]),
+        30:
+        dict(link=('left_thumb3', 'left_thumb4'), id=30, color=[255, 128, 0]),
+        31:
+        dict(
+            link=('left_wrist', 'left_forefinger1'),
+            id=31,
+            color=[255, 153, 255]),
+        32:
+        dict(
+            link=('left_forefinger1', 'left_forefinger2'),
+            id=32,
+            color=[255, 153, 255]),
+        33:
+        dict(
+            link=('left_forefinger2', 'left_forefinger3'),
+            id=33,
+            color=[255, 153, 255]),
+        34:
+        dict(
+            link=('left_forefinger3', 'left_forefinger4'),
+            id=34,
+            color=[255, 153, 255]),
+        35:
+        dict(
+            link=('left_wrist', 'left_middle_finger1'),
+            id=35,
+            color=[102, 178, 255]),
+        36:
+        dict(
+            link=('left_middle_finger1', 'left_middle_finger2'),
+            id=36,
+            color=[102, 178, 255]),
+        37:
+        dict(
+            link=('left_middle_finger2', 'left_middle_finger3'),
+            id=37,
+            color=[102, 178, 255]),
+        38:
+        dict(
+            link=('left_middle_finger3', 'left_middle_finger4'),
+            id=38,
+            color=[102, 178, 255]),
+        39:
+        dict(
+            link=('left_wrist', 'left_ring_finger1'),
+            id=39,
+            color=[255, 51, 51]),
+        40:
+        dict(
+            link=('left_ring_finger1', 'left_ring_finger2'),
+            id=40,
+            color=[255, 51, 51]),
+        41:
+        dict(
+            link=('left_ring_finger2', 'left_ring_finger3'),
+            id=41,
+            color=[255, 51, 51]),
+        42:
+        dict(
+            link=('left_ring_finger3', 'left_ring_finger4'),
+            id=42,
+            color=[255, 51, 51]),
+        43:
+        dict(
+            link=('left_wrist', 'left_pinky_finger1'),
+            id=43,
+            color=[0, 255, 0]),
+        44:
+        dict(
+            link=('left_pinky_finger1', 'left_pinky_finger2'),
+            id=44,
+            color=[0, 255, 0]),
+        45:
+        dict(
+            link=('left_pinky_finger2', 'left_pinky_finger3'),
+            id=45,
+            color=[0, 255, 0]),
+        46:
+        dict(
+            link=('left_pinky_finger3', 'left_pinky_finger4'),
+            id=46,
+            color=[0, 255, 0]),
+        47:
+        dict(link=('right_wrist', 'right_thumb1'), id=47, color=[255, 128, 0]),
+        48:
+        dict(
+            link=('right_thumb1', 'right_thumb2'), id=48, color=[255, 128, 0]),
+        49:
+        dict(
+            link=('right_thumb2', 'right_thumb3'), id=49, color=[255, 128, 0]),
+        50:
+        dict(
+            link=('right_thumb3', 'right_thumb4'), id=50, color=[255, 128, 0]),
+        51:
+        dict(
+            link=('right_wrist', 'right_forefinger1'),
+            id=51,
+            color=[255, 153, 255]),
+        52:
+        dict(
+            link=('right_forefinger1', 'right_forefinger2'),
+            id=52,
+            color=[255, 153, 255]),
+        53:
+        dict(
+            link=('right_forefinger2', 'right_forefinger3'),
+            id=53,
+            color=[255, 153, 255]),
+        54:
+        dict(
+            link=('right_forefinger3', 'right_forefinger4'),
+            id=54,
+            color=[255, 153, 255]),
+        55:
+        dict(
+            link=('right_wrist', 'right_middle_finger1'),
+            id=55,
+            color=[102, 178, 255]),
+        56:
+        dict(
+            link=('right_middle_finger1', 'right_middle_finger2'),
+            id=56,
+            color=[102, 178, 255]),
+        57:
+        dict(
+            link=('right_middle_finger2', 'right_middle_finger3'),
+            id=57,
+            color=[102, 178, 255]),
+        58:
+        dict(
+            link=('right_middle_finger3', 'right_middle_finger4'),
+            id=58,
+            color=[102, 178, 255]),
+        59:
+        dict(
+            link=('right_wrist', 'right_ring_finger1'),
+            id=59,
+            color=[255, 51, 51]),
+        60:
+        dict(
+            link=('right_ring_finger1', 'right_ring_finger2'),
+            id=60,
+            color=[255, 51, 51]),
+        61:
+        dict(
+            link=('right_ring_finger2', 'right_ring_finger3'),
+            id=61,
+            color=[255, 51, 51]),
+        62:
+        dict(
+            link=('right_ring_finger3', 'right_ring_finger4'),
+            id=62,
+            color=[255, 51, 51]),
+        63:
+        dict(
+            link=('right_wrist', 'right_pinky_finger1'),
+            id=63,
+            color=[0, 255, 0]),
+        64:
+        dict(
+            link=('right_pinky_finger1', 'right_pinky_finger2'),
+            id=64,
+            color=[0, 255, 0]),
+        65:
+        dict(
+            link=('right_pinky_finger2', 'right_pinky_finger3'),
+            id=65,
+            color=[0, 255, 0]),
+        66:
+        dict(
+            link=('right_pinky_finger3', 'right_pinky_finger4'),
+            id=66,
+            color=[0, 255, 0])
+    },
+    joint_weights=[1.] * 136,
+
+    # 'https://github.com/Fang-Haoshu/Halpe-FullBody/blob/master/'
+    # 'HalpeCOCOAPI/PythonAPI/halpecocotools/cocoeval.py#L245'
+    sigmas=[
+        0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
+        0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089, 0.08, 0.08, 0.08,
+        0.089, 0.089, 0.089, 0.089, 0.089, 0.089, 0.015, 0.015, 0.015, 0.015,
+        0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
+        0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
+        0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
+        0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
+        0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
+        0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
+        0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
+        0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
+        0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
+        0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015, 0.015,
+        0.015, 0.015, 0.015, 0.015, 0.015, 0.015
+    ])

configs/_base_/datasets/halpe26.py

@@ -0,0 +1,280 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='halpe26',
+    paper_info=dict(
+        author='Li, Yong-Lu and Xu, Liang and Liu, Xinpeng and Huang, Xijie'
+        ' and Xu, Yue and Wang, Shiyi and Fang, Hao-Shu'
+        ' and Ma, Ze and Chen, Mingyang and Lu, Cewu',
+        title='PaStaNet: Toward Human Activity Knowledge Engine',
+        container='CVPR',
+        year='2020',
+        homepage='https://github.com/Fang-Haoshu/Halpe-FullBody/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
+        1:
+        dict(
+            name='left_eye',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_eye'),
+        2:
+        dict(
+            name='right_eye',
+            id=2,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_eye'),
+        3:
+        dict(
+            name='left_ear',
+            id=3,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_ear'),
+        4:
+        dict(
+            name='right_ear',
+            id=4,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_ear'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='right_shoulder',
+            id=6,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        7:
+        dict(
+            name='left_elbow',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        8:
+        dict(
+            name='right_elbow',
+            id=8,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        9:
+        dict(
+            name='left_wrist',
+            id=9,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='right_hip',
+            id=12,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        13:
+        dict(
+            name='left_knee',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        14:
+        dict(
+            name='right_knee',
+            id=14,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        15:
+        dict(
+            name='left_ankle',
+            id=15,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        16:
+        dict(
+            name='right_ankle',
+            id=16,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle'),
+        17:
+        dict(name='head', id=17, color=[255, 128, 0], type='upper', swap=''),
+        18:
+        dict(name='neck', id=18, color=[255, 128, 0], type='upper', swap=''),
+        19:
+        dict(name='hip', id=19, color=[255, 128, 0], type='lower', swap=''),
+        20:
+        dict(
+            name='left_big_toe',
+            id=20,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_big_toe'),
+        21:
+        dict(
+            name='right_big_toe',
+            id=21,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_big_toe'),
+        22:
+        dict(
+            name='left_small_toe',
+            id=22,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_small_toe'),
+        23:
+        dict(
+            name='right_small_toe',
+            id=23,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_small_toe'),
+        24:
+        dict(
+            name='left_heel',
+            id=24,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_heel'),
+        25:
+        dict(
+            name='right_heel',
+            id=25,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_heel')
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('left_hip', 'hip'), id=2, color=[0, 255, 0]),
+        3:
+        dict(link=('right_ankle', 'right_knee'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('right_knee', 'right_hip'), id=4, color=[255, 128, 0]),
+        5:
+        dict(link=('right_hip', 'hip'), id=5, color=[255, 128, 0]),
+        6:
+        dict(link=('head', 'neck'), id=6, color=[51, 153, 255]),
+        7:
+        dict(link=('neck', 'hip'), id=7, color=[51, 153, 255]),
+        8:
+        dict(link=('neck', 'left_shoulder'), id=8, color=[0, 255, 0]),
+        9:
+        dict(link=('left_shoulder', 'left_elbow'), id=9, color=[0, 255, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('neck', 'right_shoulder'), id=11, color=[255, 128, 0]),
+        12:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=12, color=[255, 128,
+                                                                  0]),
+        13:
+        dict(link=('right_elbow', 'right_wrist'), id=13, color=[255, 128, 0]),
+        14:
+        dict(link=('left_eye', 'right_eye'), id=14, color=[51, 153, 255]),
+        15:
+        dict(link=('nose', 'left_eye'), id=15, color=[51, 153, 255]),
+        16:
+        dict(link=('nose', 'right_eye'), id=16, color=[51, 153, 255]),
+        17:
+        dict(link=('left_eye', 'left_ear'), id=17, color=[51, 153, 255]),
+        18:
+        dict(link=('right_eye', 'right_ear'), id=18, color=[51, 153, 255]),
+        19:
+        dict(link=('left_ear', 'left_shoulder'), id=19, color=[51, 153, 255]),
+        20:
+        dict(
+            link=('right_ear', 'right_shoulder'), id=20, color=[51, 153, 255]),
+        21:
+        dict(link=('left_ankle', 'left_big_toe'), id=21, color=[0, 255, 0]),
+        22:
+        dict(link=('left_ankle', 'left_small_toe'), id=22, color=[0, 255, 0]),
+        23:
+        dict(link=('left_ankle', 'left_heel'), id=23, color=[0, 255, 0]),
+        24:
+        dict(
+            link=('right_ankle', 'right_big_toe'), id=24, color=[255, 128, 0]),
+        25:
+        dict(
+            link=('right_ankle', 'right_small_toe'),
+            id=25,
+            color=[255, 128, 0]),
+        26:
+        dict(link=('right_ankle', 'right_heel'), id=26, color=[255, 128, 0]),
+    },
+    # the joint_weights is modified by MMPose Team
+    joint_weights=[
+        1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
+        1.5
+    ] + [1., 1., 1.2] + [1.5] * 6,
+
+    # 'https://github.com/Fang-Haoshu/Halpe-FullBody/blob/master/'
+    # 'HalpeCOCOAPI/PythonAPI/halpecocotools/cocoeval.py#L245'
+    sigmas=[
+        0.026,
+        0.025,
+        0.025,
+        0.035,
+        0.035,
+        0.079,
+        0.079,
+        0.072,
+        0.072,
+        0.062,
+        0.062,
+        0.107,
+        0.107,
+        0.087,
+        0.087,
+        0.089,
+        0.089,
+        0.026,
+        0.026,
+        0.066,
+        0.079,
+        0.079,
+        0.079,
+        0.079,
+        0.079,
+        0.079,
+    ])

configs/_base_/datasets/horse10.py

@@ -0,0 +1,207 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='horse10',
+    paper_info=dict(
+        author='Mathis, Alexander and Biasi, Thomas and '
+        'Schneider, Steffen and '
+        'Yuksekgonul, Mert and Rogers, Byron and '
+        'Bethge, Matthias and '
+        'Mathis, Mackenzie W',
+        title='Pretraining boosts out-of-domain robustness '
+        'for pose estimation',
+        container='Proceedings of the IEEE/CVF Winter Conference on '
+        'Applications of Computer Vision',
+        year='2021',
+        homepage='http://www.mackenziemathislab.org/horse10',
+    ),
+    keypoint_info={
+        0:
+        dict(name='Nose', id=0, color=[255, 153, 255], type='upper', swap=''),
+        1:
+        dict(name='Eye', id=1, color=[255, 153, 255], type='upper', swap=''),
+        2:
+        dict(
+            name='Nearknee',
+            id=2,
+            color=[255, 102, 255],
+            type='upper',
+            swap=''),
+        3:
+        dict(
+            name='Nearfrontfetlock',
+            id=3,
+            color=[255, 102, 255],
+            type='upper',
+            swap=''),
+        4:
+        dict(
+            name='Nearfrontfoot',
+            id=4,
+            color=[255, 102, 255],
+            type='upper',
+            swap=''),
+        5:
+        dict(
+            name='Offknee', id=5, color=[255, 102, 255], type='upper',
+            swap=''),
+        6:
+        dict(
+            name='Offfrontfetlock',
+            id=6,
+            color=[255, 102, 255],
+            type='upper',
+            swap=''),
+        7:
+        dict(
+            name='Offfrontfoot',
+            id=7,
+            color=[255, 102, 255],
+            type='upper',
+            swap=''),
+        8:
+        dict(
+            name='Shoulder',
+            id=8,
+            color=[255, 153, 255],
+            type='upper',
+            swap=''),
+        9:
+        dict(
+            name='Midshoulder',
+            id=9,
+            color=[255, 153, 255],
+            type='upper',
+            swap=''),
+        10:
+        dict(
+            name='Elbow', id=10, color=[255, 153, 255], type='upper', swap=''),
+        11:
+        dict(
+            name='Girth', id=11, color=[255, 153, 255], type='upper', swap=''),
+        12:
+        dict(
+            name='Wither', id=12, color=[255, 153, 255], type='upper',
+            swap=''),
+        13:
+        dict(
+            name='Nearhindhock',
+            id=13,
+            color=[255, 51, 255],
+            type='lower',
+            swap=''),
+        14:
+        dict(
+            name='Nearhindfetlock',
+            id=14,
+            color=[255, 51, 255],
+            type='lower',
+            swap=''),
+        15:
+        dict(
+            name='Nearhindfoot',
+            id=15,
+            color=[255, 51, 255],
+            type='lower',
+            swap=''),
+        16:
+        dict(name='Hip', id=16, color=[255, 153, 255], type='lower', swap=''),
+        17:
+        dict(
+            name='Stifle', id=17, color=[255, 153, 255], type='lower',
+            swap=''),
+        18:
+        dict(
+            name='Offhindhock',
+            id=18,
+            color=[255, 51, 255],
+            type='lower',
+            swap=''),
+        19:
+        dict(
+            name='Offhindfetlock',
+            id=19,
+            color=[255, 51, 255],
+            type='lower',
+            swap=''),
+        20:
+        dict(
+            name='Offhindfoot',
+            id=20,
+            color=[255, 51, 255],
+            type='lower',
+            swap=''),
+        21:
+        dict(
+            name='Ischium',
+            id=21,
+            color=[255, 153, 255],
+            type='lower',
+            swap='')
+    },
+    skeleton_info={
+        0:
+        dict(link=('Nose', 'Eye'), id=0, color=[255, 153, 255]),
+        1:
+        dict(link=('Eye', 'Wither'), id=1, color=[255, 153, 255]),
+        2:
+        dict(link=('Wither', 'Hip'), id=2, color=[255, 153, 255]),
+        3:
+        dict(link=('Hip', 'Ischium'), id=3, color=[255, 153, 255]),
+        4:
+        dict(link=('Ischium', 'Stifle'), id=4, color=[255, 153, 255]),
+        5:
+        dict(link=('Stifle', 'Girth'), id=5, color=[255, 153, 255]),
+        6:
+        dict(link=('Girth', 'Elbow'), id=6, color=[255, 153, 255]),
+        7:
+        dict(link=('Elbow', 'Shoulder'), id=7, color=[255, 153, 255]),
+        8:
+        dict(link=('Shoulder', 'Midshoulder'), id=8, color=[255, 153, 255]),
+        9:
+        dict(link=('Midshoulder', 'Wither'), id=9, color=[255, 153, 255]),
+        10:
+        dict(
+            link=('Nearknee', 'Nearfrontfetlock'),
+            id=10,
+            color=[255, 102, 255]),
+        11:
+        dict(
+            link=('Nearfrontfetlock', 'Nearfrontfoot'),
+            id=11,
+            color=[255, 102, 255]),
+        12:
+        dict(
+            link=('Offknee', 'Offfrontfetlock'), id=12, color=[255, 102, 255]),
+        13:
+        dict(
+            link=('Offfrontfetlock', 'Offfrontfoot'),
+            id=13,
+            color=[255, 102, 255]),
+        14:
+        dict(
+            link=('Nearhindhock', 'Nearhindfetlock'),
+            id=14,
+            color=[255, 51, 255]),
+        15:
+        dict(
+            link=('Nearhindfetlock', 'Nearhindfoot'),
+            id=15,
+            color=[255, 51, 255]),
+        16:
+        dict(
+            link=('Offhindhock', 'Offhindfetlock'),
+            id=16,
+            color=[255, 51, 255]),
+        17:
+        dict(
+            link=('Offhindfetlock', 'Offhindfoot'),
+            id=17,
+            color=[255, 51, 255])
+    },
+    joint_weights=[1.] * 22,
+    sigmas=[])

configs/_base_/datasets/humanart.py

@@ -0,0 +1,187 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='Human-Art',
+    paper_info=dict(
+        author='Ju, Xuan and Zeng, Ailing and '
+        'Wang, Jianan and Xu, Qiang and Zhang, Lei',
+        title='Human-Art: A Versatile Human-Centric Dataset '
+        'Bridging Natural and Artificial Scenes',
+        container='Proceedings of the IEEE/CVF Conference on '
+        'Computer Vision and Pattern Recognition',
+        year='2023',
+        homepage='https://idea-research.github.io/HumanArt/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
+        1:
+        dict(
+            name='left_eye',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_eye'),
+        2:
+        dict(
+            name='right_eye',
+            id=2,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_eye'),
+        3:
+        dict(
+            name='left_ear',
+            id=3,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_ear'),
+        4:
+        dict(
+            name='right_ear',
+            id=4,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_ear'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='right_shoulder',
+            id=6,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        7:
+        dict(
+            name='left_elbow',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        8:
+        dict(
+            name='right_elbow',
+            id=8,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        9:
+        dict(
+            name='left_wrist',
+            id=9,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='right_hip',
+            id=12,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        13:
+        dict(
+            name='left_knee',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        14:
+        dict(
+            name='right_knee',
+            id=14,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        15:
+        dict(
+            name='left_ankle',
+            id=15,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        16:
+        dict(
+            name='right_ankle',
+            id=16,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle')
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
+        5:
+        dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
+        6:
+        dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
+        7:
+        dict(
+            link=('left_shoulder', 'right_shoulder'),
+            id=7,
+            color=[51, 153, 255]),
+        8:
+        dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
+        9:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
+        13:
+        dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
+        14:
+        dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
+        15:
+        dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
+        16:
+        dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
+        17:
+        dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
+        18:
+        dict(
+            link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255])
+    },
+    joint_weights=[
+        1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
+        1.5
+    ],
+    sigmas=[
+        0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
+        0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089
+    ])

configs/_base_/datasets/humanart_aic.py

@@ -0,0 +1,211 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='humanart',
+    paper_info=[
+        dict(
+            author='Ju, Xuan and Zeng, Ailing and '
+            'Wang, Jianan and Xu, Qiang and Zhang, '
+            'Lei',
+            title='Human-Art: A Versatile Human-Centric Dataset '
+            'Bridging Natural and Artificial Scenes',
+            container='CVPR',
+            year='2023',
+            homepage='https://idea-research.github.io/HumanArt/',
+        ),
+        dict(
+            author='Wu, Jiahong and Zheng, He and Zhao, Bo and '
+            'Li, Yixin and Yan, Baoming and Liang, Rui and '
+            'Wang, Wenjia and Zhou, Shipei and Lin, Guosen and '
+            'Fu, Yanwei and others',
+            title='Ai challenger: A large-scale dataset for going '
+            'deeper in image understanding',
+            container='arXiv',
+            year='2017',
+            homepage='https://github.com/AIChallenger/AI_Challenger_2017',
+        ),
+    ],
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
+        1:
+        dict(
+            name='left_eye',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_eye'),
+        2:
+        dict(
+            name='right_eye',
+            id=2,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_eye'),
+        3:
+        dict(
+            name='left_ear',
+            id=3,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_ear'),
+        4:
+        dict(
+            name='right_ear',
+            id=4,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_ear'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='right_shoulder',
+            id=6,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        7:
+        dict(
+            name='left_elbow',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        8:
+        dict(
+            name='right_elbow',
+            id=8,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        9:
+        dict(
+            name='left_wrist',
+            id=9,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='right_hip',
+            id=12,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        13:
+        dict(
+            name='left_knee',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        14:
+        dict(
+            name='right_knee',
+            id=14,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        15:
+        dict(
+            name='left_ankle',
+            id=15,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        16:
+        dict(
+            name='right_ankle',
+            id=16,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle'),
+        17:
+        dict(
+            name='head_top',
+            id=17,
+            color=[51, 153, 255],
+            type='upper',
+            swap=''),
+        18:
+        dict(name='neck', id=18, color=[51, 153, 255], type='upper', swap='')
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
+        5:
+        dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
+        6:
+        dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
+        7:
+        dict(
+            link=('left_shoulder', 'right_shoulder'),
+            id=7,
+            color=[51, 153, 255]),
+        8:
+        dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
+        9:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
+        13:
+        dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
+        14:
+        dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
+        15:
+        dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
+        16:
+        dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
+        17:
+        dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
+        18:
+        dict(
+            link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255]),
+        19:
+        dict(link=('head_top', 'neck'), id=11, color=[51, 153, 255]),
+    },
+    joint_weights=[
+        1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
+        1.5, 1.5
+    ],
+    sigmas=[
+        0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
+        0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089, 0.026, 0.026
+    ])

configs/_base_/datasets/interhand2d.py

@@ -0,0 +1,148 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='interhand2d',
+    paper_info=dict(
+        author='Moon, Gyeongsik and Yu, Shoou-I and Wen, He and '
+        'Shiratori, Takaaki and Lee, Kyoung Mu',
+        title='InterHand2.6M: A dataset and baseline for 3D '
+        'interacting hand pose estimation from a single RGB image',
+        container='arXiv',
+        year='2020',
+        homepage='https://mks0601.github.io/InterHand2.6M/',
+    ),
+    keypoint_info={
+        0:
+        dict(name='thumb4', id=0, color=[255, 128, 0], type='', swap=''),
+        1:
+        dict(name='thumb3', id=1, color=[255, 128, 0], type='', swap=''),
+        2:
+        dict(name='thumb2', id=2, color=[255, 128, 0], type='', swap=''),
+        3:
+        dict(name='thumb1', id=3, color=[255, 128, 0], type='', swap=''),
+        4:
+        dict(
+            name='forefinger4', id=4, color=[255, 153, 255], type='', swap=''),
+        5:
+        dict(
+            name='forefinger3', id=5, color=[255, 153, 255], type='', swap=''),
+        6:
+        dict(
+            name='forefinger2', id=6, color=[255, 153, 255], type='', swap=''),
+        7:
+        dict(
+            name='forefinger1', id=7, color=[255, 153, 255], type='', swap=''),
+        8:
+        dict(
+            name='middle_finger4',
+            id=8,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        9:
+        dict(
+            name='middle_finger3',
+            id=9,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        10:
+        dict(
+            name='middle_finger2',
+            id=10,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        11:
+        dict(
+            name='middle_finger1',
+            id=11,
+            color=[102, 178, 255],
+            type='',
+            swap=''),
+        12:
+        dict(
+            name='ring_finger4', id=12, color=[255, 51, 51], type='', swap=''),
+        13:
+        dict(
+            name='ring_finger3', id=13, color=[255, 51, 51], type='', swap=''),
+        14:
+        dict(
+            name='ring_finger2', id=14, color=[255, 51, 51], type='', swap=''),
+        15:
+        dict(
+            name='ring_finger1', id=15, color=[255, 51, 51], type='', swap=''),
+        16:
+        dict(name='pinky_finger4', id=16, color=[0, 255, 0], type='', swap=''),
+        17:
+        dict(name='pinky_finger3', id=17, color=[0, 255, 0], type='', swap=''),
+        18:
+        dict(name='pinky_finger2', id=18, color=[0, 255, 0], type='', swap=''),
+        19:
+        dict(name='pinky_finger1', id=19, color=[0, 255, 0], type='', swap=''),
+        20:
+        dict(name='wrist', id=20, color=[255, 255, 255], type='', swap='')
+    },
+    skeleton_info={
+        0:
+        dict(link=('wrist', 'thumb1'), id=0, color=[255, 128, 0]),
+        1:
+        dict(link=('thumb1', 'thumb2'), id=1, color=[255, 128, 0]),
+        2:
+        dict(link=('thumb2', 'thumb3'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('thumb3', 'thumb4'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('wrist', 'forefinger1'), id=4, color=[255, 153, 255]),
+        5:
+        dict(link=('forefinger1', 'forefinger2'), id=5, color=[255, 153, 255]),
+        6:
+        dict(link=('forefinger2', 'forefinger3'), id=6, color=[255, 153, 255]),
+        7:
+        dict(link=('forefinger3', 'forefinger4'), id=7, color=[255, 153, 255]),
+        8:
+        dict(link=('wrist', 'middle_finger1'), id=8, color=[102, 178, 255]),
+        9:
+        dict(
+            link=('middle_finger1', 'middle_finger2'),
+            id=9,
+            color=[102, 178, 255]),
+        10:
+        dict(
+            link=('middle_finger2', 'middle_finger3'),
+            id=10,
+            color=[102, 178, 255]),
+        11:
+        dict(
+            link=('middle_finger3', 'middle_finger4'),
+            id=11,
+            color=[102, 178, 255]),
+        12:
+        dict(link=('wrist', 'ring_finger1'), id=12, color=[255, 51, 51]),
+        13:
+        dict(
+            link=('ring_finger1', 'ring_finger2'), id=13, color=[255, 51, 51]),
+        14:
+        dict(
+            link=('ring_finger2', 'ring_finger3'), id=14, color=[255, 51, 51]),
+        15:
+        dict(
+            link=('ring_finger3', 'ring_finger4'), id=15, color=[255, 51, 51]),
+        16:
+        dict(link=('wrist', 'pinky_finger1'), id=16, color=[0, 255, 0]),
+        17:
+        dict(
+            link=('pinky_finger1', 'pinky_finger2'), id=17, color=[0, 255, 0]),
+        18:
+        dict(
+            link=('pinky_finger2', 'pinky_finger3'), id=18, color=[0, 255, 0]),
+        19:
+        dict(
+            link=('pinky_finger3', 'pinky_finger4'), id=19, color=[0, 255, 0])
+    },
+    joint_weights=[1.] * 21,
+    sigmas=[])

configs/_base_/datasets/interhand3d.py

@@ -0,0 +1,493 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='interhand3d',
+    paper_info=dict(
+        author='Moon, Gyeongsik and Yu, Shoou-I and Wen, He and '
+        'Shiratori, Takaaki and Lee, Kyoung Mu',
+        title='InterHand2.6M: A dataset and baseline for 3D '
+        'interacting hand pose estimation from a single RGB image',
+        container='arXiv',
+        year='2020',
+        homepage='https://mks0601.github.io/InterHand2.6M/',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='right_thumb4',
+            id=0,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb4'),
+        1:
+        dict(
+            name='right_thumb3',
+            id=1,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb3'),
+        2:
+        dict(
+            name='right_thumb2',
+            id=2,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb2'),
+        3:
+        dict(
+            name='right_thumb1',
+            id=3,
+            color=[255, 128, 0],
+            type='',
+            swap='left_thumb1'),
+        4:
+        dict(
+            name='right_forefinger4',
+            id=4,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger4'),
+        5:
+        dict(
+            name='right_forefinger3',
+            id=5,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger3'),
+        6:
+        dict(
+            name='right_forefinger2',
+            id=6,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger2'),
+        7:
+        dict(
+            name='right_forefinger1',
+            id=7,
+            color=[255, 153, 255],
+            type='',
+            swap='left_forefinger1'),
+        8:
+        dict(
+            name='right_middle_finger4',
+            id=8,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger4'),
+        9:
+        dict(
+            name='right_middle_finger3',
+            id=9,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger3'),
+        10:
+        dict(
+            name='right_middle_finger2',
+            id=10,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger2'),
+        11:
+        dict(
+            name='right_middle_finger1',
+            id=11,
+            color=[102, 178, 255],
+            type='',
+            swap='left_middle_finger1'),
+        12:
+        dict(
+            name='right_ring_finger4',
+            id=12,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger4'),
+        13:
+        dict(
+            name='right_ring_finger3',
+            id=13,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger3'),
+        14:
+        dict(
+            name='right_ring_finger2',
+            id=14,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger2'),
+        15:
+        dict(
+            name='right_ring_finger1',
+            id=15,
+            color=[255, 51, 51],
+            type='',
+            swap='left_ring_finger1'),
+        16:
+        dict(
+            name='right_pinky_finger4',
+            id=16,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger4'),
+        17:
+        dict(
+            name='right_pinky_finger3',
+            id=17,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger3'),
+        18:
+        dict(
+            name='right_pinky_finger2',
+            id=18,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger2'),
+        19:
+        dict(
+            name='right_pinky_finger1',
+            id=19,
+            color=[0, 255, 0],
+            type='',
+            swap='left_pinky_finger1'),
+        20:
+        dict(
+            name='right_wrist',
+            id=20,
+            color=[255, 255, 255],
+            type='',
+            swap='left_wrist'),
+        21:
+        dict(
+            name='left_thumb4',
+            id=21,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb4'),
+        22:
+        dict(
+            name='left_thumb3',
+            id=22,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb3'),
+        23:
+        dict(
+            name='left_thumb2',
+            id=23,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb2'),
+        24:
+        dict(
+            name='left_thumb1',
+            id=24,
+            color=[255, 128, 0],
+            type='',
+            swap='right_thumb1'),
+        25:
+        dict(
+            name='left_forefinger4',
+            id=25,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger4'),
+        26:
+        dict(
+            name='left_forefinger3',
+            id=26,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger3'),
+        27:
+        dict(
+            name='left_forefinger2',
+            id=27,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger2'),
+        28:
+        dict(
+            name='left_forefinger1',
+            id=28,
+            color=[255, 153, 255],
+            type='',
+            swap='right_forefinger1'),
+        29:
+        dict(
+            name='left_middle_finger4',
+            id=29,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger4'),
+        30:
+        dict(
+            name='left_middle_finger3',
+            id=30,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger3'),
+        31:
+        dict(
+            name='left_middle_finger2',
+            id=31,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger2'),
+        32:
+        dict(
+            name='left_middle_finger1',
+            id=32,
+            color=[102, 178, 255],
+            type='',
+            swap='right_middle_finger1'),
+        33:
+        dict(
+            name='left_ring_finger4',
+            id=33,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger4'),
+        34:
+        dict(
+            name='left_ring_finger3',
+            id=34,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger3'),
+        35:
+        dict(
+            name='left_ring_finger2',
+            id=35,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger2'),
+        36:
+        dict(
+            name='left_ring_finger1',
+            id=36,
+            color=[255, 51, 51],
+            type='',
+            swap='right_ring_finger1'),
+        37:
+        dict(
+            name='left_pinky_finger4',
+            id=37,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger4'),
+        38:
+        dict(
+            name='left_pinky_finger3',
+            id=38,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger3'),
+        39:
+        dict(
+            name='left_pinky_finger2',
+            id=39,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger2'),
+        40:
+        dict(
+            name='left_pinky_finger1',
+            id=40,
+            color=[0, 255, 0],
+            type='',
+            swap='right_pinky_finger1'),
+        41:
+        dict(
+            name='left_wrist',
+            id=41,
+            color=[255, 255, 255],
+            type='',
+            swap='right_wrist'),
+    },
+    skeleton_info={
+        0:
+        dict(link=('right_wrist', 'right_thumb1'), id=0, color=[255, 128, 0]),
+        1:
+        dict(link=('right_thumb1', 'right_thumb2'), id=1, color=[255, 128, 0]),
+        2:
+        dict(link=('right_thumb2', 'right_thumb3'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('right_thumb3', 'right_thumb4'), id=3, color=[255, 128, 0]),
+        4:
+        dict(
+            link=('right_wrist', 'right_forefinger1'),
+            id=4,
+            color=[255, 153, 255]),
+        5:
+        dict(
+            link=('right_forefinger1', 'right_forefinger2'),
+            id=5,
+            color=[255, 153, 255]),
+        6:
+        dict(
+            link=('right_forefinger2', 'right_forefinger3'),
+            id=6,
+            color=[255, 153, 255]),
+        7:
+        dict(
+            link=('right_forefinger3', 'right_forefinger4'),
+            id=7,
+            color=[255, 153, 255]),
+        8:
+        dict(
+            link=('right_wrist', 'right_middle_finger1'),
+            id=8,
+            color=[102, 178, 255]),
+        9:
+        dict(
+            link=('right_middle_finger1', 'right_middle_finger2'),
+            id=9,
+            color=[102, 178, 255]),
+        10:
+        dict(
+            link=('right_middle_finger2', 'right_middle_finger3'),
+            id=10,
+            color=[102, 178, 255]),
+        11:
+        dict(
+            link=('right_middle_finger3', 'right_middle_finger4'),
+            id=11,
+            color=[102, 178, 255]),
+        12:
+        dict(
+            link=('right_wrist', 'right_ring_finger1'),
+            id=12,
+            color=[255, 51, 51]),
+        13:
+        dict(
+            link=('right_ring_finger1', 'right_ring_finger2'),
+            id=13,
+            color=[255, 51, 51]),
+        14:
+        dict(
+            link=('right_ring_finger2', 'right_ring_finger3'),
+            id=14,
+            color=[255, 51, 51]),
+        15:
+        dict(
+            link=('right_ring_finger3', 'right_ring_finger4'),
+            id=15,
+            color=[255, 51, 51]),
+        16:
+        dict(
+            link=('right_wrist', 'right_pinky_finger1'),
+            id=16,
+            color=[0, 255, 0]),
+        17:
+        dict(
+            link=('right_pinky_finger1', 'right_pinky_finger2'),
+            id=17,
+            color=[0, 255, 0]),
+        18:
+        dict(
+            link=('right_pinky_finger2', 'right_pinky_finger3'),
+            id=18,
+            color=[0, 255, 0]),
+        19:
+        dict(
+            link=('right_pinky_finger3', 'right_pinky_finger4'),
+            id=19,
+            color=[0, 255, 0]),
+        20:
+        dict(link=('left_wrist', 'left_thumb1'), id=20, color=[255, 128, 0]),
+        21:
+        dict(link=('left_thumb1', 'left_thumb2'), id=21, color=[255, 128, 0]),
+        22:
+        dict(link=('left_thumb2', 'left_thumb3'), id=22, color=[255, 128, 0]),
+        23:
+        dict(link=('left_thumb3', 'left_thumb4'), id=23, color=[255, 128, 0]),
+        24:
+        dict(
+            link=('left_wrist', 'left_forefinger1'),
+            id=24,
+            color=[255, 153, 255]),
+        25:
+        dict(
+            link=('left_forefinger1', 'left_forefinger2'),
+            id=25,
+            color=[255, 153, 255]),
+        26:
+        dict(
+            link=('left_forefinger2', 'left_forefinger3'),
+            id=26,
+            color=[255, 153, 255]),
+        27:
+        dict(
+            link=('left_forefinger3', 'left_forefinger4'),
+            id=27,
+            color=[255, 153, 255]),
+        28:
+        dict(
+            link=('left_wrist', 'left_middle_finger1'),
+            id=28,
+            color=[102, 178, 255]),
+        29:
+        dict(
+            link=('left_middle_finger1', 'left_middle_finger2'),
+            id=29,
+            color=[102, 178, 255]),
+        30:
+        dict(
+            link=('left_middle_finger2', 'left_middle_finger3'),
+            id=30,
+            color=[102, 178, 255]),
+        31:
+        dict(
+            link=('left_middle_finger3', 'left_middle_finger4'),
+            id=31,
+            color=[102, 178, 255]),
+        32:
+        dict(
+            link=('left_wrist', 'left_ring_finger1'),
+            id=32,
+            color=[255, 51, 51]),
+        33:
+        dict(
+            link=('left_ring_finger1', 'left_ring_finger2'),
+            id=33,
+            color=[255, 51, 51]),
+        34:
+        dict(
+            link=('left_ring_finger2', 'left_ring_finger3'),
+            id=34,
+            color=[255, 51, 51]),
+        35:
+        dict(
+            link=('left_ring_finger3', 'left_ring_finger4'),
+            id=35,
+            color=[255, 51, 51]),
+        36:
+        dict(
+            link=('left_wrist', 'left_pinky_finger1'),
+            id=36,
+            color=[0, 255, 0]),
+        37:
+        dict(
+            link=('left_pinky_finger1', 'left_pinky_finger2'),
+            id=37,
+            color=[0, 255, 0]),
+        38:
+        dict(
+            link=('left_pinky_finger2', 'left_pinky_finger3'),
+            id=38,
+            color=[0, 255, 0]),
+        39:
+        dict(
+            link=('left_pinky_finger3', 'left_pinky_finger4'),
+            id=39,
+            color=[0, 255, 0]),
+    },
+    joint_weights=[1.] * 42,
+    sigmas=[])

configs/_base_/datasets/jhmdb.py

@@ -0,0 +1,135 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='jhmdb',
+    paper_info=dict(
+        author='H. Jhuang and J. Gall and S. Zuffi and '
+        'C. Schmid and M. J. Black',
+        title='Towards understanding action recognition',
+        container='International Conf. on Computer Vision (ICCV)',
+        year='2013',
+        homepage='http://jhmdb.is.tue.mpg.de/dataset',
+    ),
+    keypoint_info={
+        0:
+        dict(name='neck', id=0, color=[255, 128, 0], type='upper', swap=''),
+        1:
+        dict(name='belly', id=1, color=[255, 128, 0], type='upper', swap=''),
+        2:
+        dict(name='head', id=2, color=[255, 128, 0], type='upper', swap=''),
+        3:
+        dict(
+            name='right_shoulder',
+            id=3,
+            color=[0, 255, 0],
+            type='upper',
+            swap='left_shoulder'),
+        4:
+        dict(
+            name='left_shoulder',
+            id=4,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        5:
+        dict(
+            name='right_hip',
+            id=5,
+            color=[0, 255, 0],
+            type='lower',
+            swap='left_hip'),
+        6:
+        dict(
+            name='left_hip',
+            id=6,
+            color=[51, 153, 255],
+            type='lower',
+            swap='right_hip'),
+        7:
+        dict(
+            name='right_elbow',
+            id=7,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_elbow'),
+        8:
+        dict(
+            name='left_elbow',
+            id=8,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_elbow'),
+        9:
+        dict(
+            name='right_knee',
+            id=9,
+            color=[51, 153, 255],
+            type='lower',
+            swap='left_knee'),
+        10:
+        dict(
+            name='left_knee',
+            id=10,
+            color=[255, 128, 0],
+            type='lower',
+            swap='right_knee'),
+        11:
+        dict(
+            name='right_wrist',
+            id=11,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        12:
+        dict(
+            name='left_wrist',
+            id=12,
+            color=[255, 128, 0],
+            type='upper',
+            swap='right_wrist'),
+        13:
+        dict(
+            name='right_ankle',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='left_ankle'),
+        14:
+        dict(
+            name='left_ankle',
+            id=14,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle')
+    },
+    skeleton_info={
+        0: dict(link=('right_ankle', 'right_knee'), id=0, color=[255, 128, 0]),
+        1: dict(link=('right_knee', 'right_hip'), id=1, color=[255, 128, 0]),
+        2: dict(link=('right_hip', 'belly'), id=2, color=[255, 128, 0]),
+        3: dict(link=('belly', 'left_hip'), id=3, color=[0, 255, 0]),
+        4: dict(link=('left_hip', 'left_knee'), id=4, color=[0, 255, 0]),
+        5: dict(link=('left_knee', 'left_ankle'), id=5, color=[0, 255, 0]),
+        6: dict(link=('belly', 'neck'), id=6, color=[51, 153, 255]),
+        7: dict(link=('neck', 'head'), id=7, color=[51, 153, 255]),
+        8: dict(link=('neck', 'right_shoulder'), id=8, color=[255, 128, 0]),
+        9: dict(
+            link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('right_elbow', 'right_wrist'), id=10, color=[255, 128, 0]),
+        11: dict(link=('neck', 'left_shoulder'), id=11, color=[0, 255, 0]),
+        12:
+        dict(link=('left_shoulder', 'left_elbow'), id=12, color=[0, 255, 0]),
+        13: dict(link=('left_elbow', 'left_wrist'), id=13, color=[0, 255, 0])
+    },
+    joint_weights=[
+        1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.2, 1.2, 1.5, 1.5, 1.5, 1.5
+    ],
+    # Adapted from COCO dataset.
+    sigmas=[
+        0.025, 0.107, 0.025, 0.079, 0.079, 0.107, 0.107, 0.072, 0.072, 0.087,
+        0.087, 0.062, 0.062, 0.089, 0.089
+    ])

configs/_base_/datasets/lapa.py

@@ -0,0 +1,694 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='lapa',
+    paper_info=dict(
+        author='Liu, Yinglu and Shi, Hailin and Shen, Hao and Si, '
+        'Yue and Wang, Xiaobo and Mei, Tao',
+        title='A New Dataset and Boundary-Attention Semantic '
+        'Segmentation for Face Parsing.',
+        container='Proceedings of the AAAI Conference on '
+        'Artificial Intelligence 2020',
+        year='2020',
+        homepage='https://github.com/JDAI-CV/lapa-dataset',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='kpt-0', id=0, color=[255, 0, 0], type='upper',
+            swap='kpt-32'),
+        1:
+        dict(
+            name='kpt-1', id=1, color=[255, 0, 0], type='upper',
+            swap='kpt-31'),
+        2:
+        dict(
+            name='kpt-2', id=2, color=[255, 0, 0], type='upper',
+            swap='kpt-30'),
+        3:
+        dict(
+            name='kpt-3', id=3, color=[255, 0, 0], type='lower',
+            swap='kpt-29'),
+        4:
+        dict(
+            name='kpt-4', id=4, color=[255, 0, 0], type='lower',
+            swap='kpt-28'),
+        5:
+        dict(
+            name='kpt-5', id=5, color=[255, 0, 0], type='lower',
+            swap='kpt-27'),
+        6:
+        dict(
+            name='kpt-6', id=6, color=[255, 0, 0], type='lower',
+            swap='kpt-26'),
+        7:
+        dict(
+            name='kpt-7', id=7, color=[255, 0, 0], type='lower',
+            swap='kpt-25'),
+        8:
+        dict(
+            name='kpt-8', id=8, color=[255, 0, 0], type='lower',
+            swap='kpt-24'),
+        9:
+        dict(
+            name='kpt-9', id=9, color=[255, 0, 0], type='lower',
+            swap='kpt-23'),
+        10:
+        dict(
+            name='kpt-10',
+            id=10,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-22'),
+        11:
+        dict(
+            name='kpt-11',
+            id=11,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-21'),
+        12:
+        dict(
+            name='kpt-12',
+            id=12,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-20'),
+        13:
+        dict(
+            name='kpt-13',
+            id=13,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-19'),
+        14:
+        dict(
+            name='kpt-14',
+            id=14,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-18'),
+        15:
+        dict(
+            name='kpt-15',
+            id=15,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-17'),
+        16:
+        dict(name='kpt-16', id=16, color=[255, 0, 0], type='lower', swap=''),
+        17:
+        dict(
+            name='kpt-17',
+            id=17,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-15'),
+        18:
+        dict(
+            name='kpt-18',
+            id=18,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-14'),
+        19:
+        dict(
+            name='kpt-19',
+            id=19,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-13'),
+        20:
+        dict(
+            name='kpt-20',
+            id=20,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-12'),
+        21:
+        dict(
+            name='kpt-21',
+            id=21,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-11'),
+        22:
+        dict(
+            name='kpt-22',
+            id=22,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-10'),
+        23:
+        dict(
+            name='kpt-23',
+            id=23,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-9'),
+        24:
+        dict(
+            name='kpt-24',
+            id=24,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-8'),
+        25:
+        dict(
+            name='kpt-25',
+            id=25,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-7'),
+        26:
+        dict(
+            name='kpt-26',
+            id=26,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-6'),
+        27:
+        dict(
+            name='kpt-27',
+            id=27,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-5'),
+        28:
+        dict(
+            name='kpt-28',
+            id=28,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-4'),
+        29:
+        dict(
+            name='kpt-29',
+            id=29,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-3'),
+        30:
+        dict(
+            name='kpt-30',
+            id=30,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-2'),
+        31:
+        dict(
+            name='kpt-31',
+            id=31,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-1'),
+        32:
+        dict(
+            name='kpt-32',
+            id=32,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-0'),
+        33:
+        dict(
+            name='kpt-33',
+            id=33,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-46'),
+        34:
+        dict(
+            name='kpt-34',
+            id=34,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-45'),
+        35:
+        dict(
+            name='kpt-35',
+            id=35,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-44'),
+        36:
+        dict(
+            name='kpt-36',
+            id=36,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-43'),
+        37:
+        dict(
+            name='kpt-37',
+            id=37,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-42'),
+        38:
+        dict(
+            name='kpt-38',
+            id=38,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-50'),
+        39:
+        dict(
+            name='kpt-39',
+            id=39,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-49'),
+        40:
+        dict(
+            name='kpt-40',
+            id=40,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-48'),
+        41:
+        dict(
+            name='kpt-41',
+            id=41,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-47'),
+        42:
+        dict(
+            name='kpt-42',
+            id=42,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-37'),
+        43:
+        dict(
+            name='kpt-43',
+            id=43,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-36'),
+        44:
+        dict(
+            name='kpt-44',
+            id=44,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-35'),
+        45:
+        dict(
+            name='kpt-45',
+            id=45,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-34'),
+        46:
+        dict(
+            name='kpt-46',
+            id=46,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-33'),
+        47:
+        dict(
+            name='kpt-47',
+            id=47,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-41'),
+        48:
+        dict(
+            name='kpt-48',
+            id=48,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-40'),
+        49:
+        dict(
+            name='kpt-49',
+            id=49,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-39'),
+        50:
+        dict(
+            name='kpt-50',
+            id=50,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-38'),
+        51:
+        dict(name='kpt-51', id=51, color=[255, 0, 0], type='upper', swap=''),
+        52:
+        dict(name='kpt-52', id=52, color=[255, 0, 0], type='upper', swap=''),
+        53:
+        dict(name='kpt-53', id=53, color=[255, 0, 0], type='lower', swap=''),
+        54:
+        dict(name='kpt-54', id=54, color=[255, 0, 0], type='lower', swap=''),
+        55:
+        dict(
+            name='kpt-55',
+            id=55,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-65'),
+        56:
+        dict(
+            name='kpt-56',
+            id=56,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-64'),
+        57:
+        dict(
+            name='kpt-57',
+            id=57,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-63'),
+        58:
+        dict(
+            name='kpt-58',
+            id=58,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-62'),
+        59:
+        dict(
+            name='kpt-59',
+            id=59,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-61'),
+        60:
+        dict(name='kpt-60', id=60, color=[255, 0, 0], type='lower', swap=''),
+        61:
+        dict(
+            name='kpt-61',
+            id=61,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-59'),
+        62:
+        dict(
+            name='kpt-62',
+            id=62,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-58'),
+        63:
+        dict(
+            name='kpt-63',
+            id=63,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-57'),
+        64:
+        dict(
+            name='kpt-64',
+            id=64,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-56'),
+        65:
+        dict(
+            name='kpt-65',
+            id=65,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-55'),
+        66:
+        dict(
+            name='kpt-66',
+            id=66,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-79'),
+        67:
+        dict(
+            name='kpt-67',
+            id=67,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-78'),
+        68:
+        dict(
+            name='kpt-68',
+            id=68,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-77'),
+        69:
+        dict(
+            name='kpt-69',
+            id=69,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-76'),
+        70:
+        dict(
+            name='kpt-70',
+            id=70,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-75'),
+        71:
+        dict(
+            name='kpt-71',
+            id=71,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-82'),
+        72:
+        dict(
+            name='kpt-72',
+            id=72,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-81'),
+        73:
+        dict(
+            name='kpt-73',
+            id=73,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-80'),
+        74:
+        dict(
+            name='kpt-74',
+            id=74,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-83'),
+        75:
+        dict(
+            name='kpt-75',
+            id=75,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-70'),
+        76:
+        dict(
+            name='kpt-76',
+            id=76,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-69'),
+        77:
+        dict(
+            name='kpt-77',
+            id=77,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-68'),
+        78:
+        dict(
+            name='kpt-78',
+            id=78,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-67'),
+        79:
+        dict(
+            name='kpt-79',
+            id=79,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-66'),
+        80:
+        dict(
+            name='kpt-80',
+            id=80,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-73'),
+        81:
+        dict(
+            name='kpt-81',
+            id=81,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-72'),
+        82:
+        dict(
+            name='kpt-82',
+            id=82,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-71'),
+        83:
+        dict(
+            name='kpt-83',
+            id=83,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-74'),
+        84:
+        dict(
+            name='kpt-84',
+            id=84,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-90'),
+        85:
+        dict(
+            name='kpt-85',
+            id=85,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-89'),
+        86:
+        dict(
+            name='kpt-86',
+            id=86,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-88'),
+        87:
+        dict(name='kpt-87', id=87, color=[255, 0, 0], type='lower', swap=''),
+        88:
+        dict(
+            name='kpt-88',
+            id=88,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-86'),
+        89:
+        dict(
+            name='kpt-89',
+            id=89,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-85'),
+        90:
+        dict(
+            name='kpt-90',
+            id=90,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-84'),
+        91:
+        dict(
+            name='kpt-91',
+            id=91,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-95'),
+        92:
+        dict(
+            name='kpt-92',
+            id=92,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-94'),
+        93:
+        dict(name='kpt-93', id=93, color=[255, 0, 0], type='lower', swap=''),
+        94:
+        dict(
+            name='kpt-94',
+            id=94,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-92'),
+        95:
+        dict(
+            name='kpt-95',
+            id=95,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-91'),
+        96:
+        dict(
+            name='kpt-96',
+            id=96,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-100'),
+        97:
+        dict(
+            name='kpt-97',
+            id=97,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-99'),
+        98:
+        dict(name='kpt-98', id=98, color=[255, 0, 0], type='lower', swap=''),
+        99:
+        dict(
+            name='kpt-99',
+            id=99,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-97'),
+        100:
+        dict(
+            name='kpt-100',
+            id=100,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-96'),
+        101:
+        dict(
+            name='kpt-101',
+            id=101,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-103'),
+        102:
+        dict(name='kpt-102', id=102, color=[255, 0, 0], type='lower', swap=''),
+        103:
+        dict(
+            name='kpt-103',
+            id=103,
+            color=[255, 0, 0],
+            type='lower',
+            swap='kpt-101'),
+        104:
+        dict(
+            name='kpt-104',
+            id=104,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-105'),
+        105:
+        dict(
+            name='kpt-105',
+            id=105,
+            color=[255, 0, 0],
+            type='upper',
+            swap='kpt-104')
+    },
+    skeleton_info={},
+    joint_weights=[
+        0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8,
+        0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8, 0.8,
+        0.8, 0.8, 0.8, 0.8, 0.8, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
+        1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0,
+        1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 2.0, 2.0, 2.0, 2.0,
+        2.0, 2.0, 2.0, 2.0, 1.0, 2.0, 2.0, 2.0, 2.0, 2.0, 2.0, 2.0, 2.0, 1.0,
+        1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5,
+        1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.0, 1.0
+    ],
+    sigmas=[])

configs/_base_/datasets/locust.py

@@ -0,0 +1,269 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='locust',
+    paper_info=dict(
+        author='Graving, Jacob M and Chae, Daniel and Naik, Hemal and '
+        'Li, Liang and Koger, Benjamin and Costelloe, Blair R and '
+        'Couzin, Iain D',
+        title='DeepPoseKit, a software toolkit for fast and robust '
+        'animal pose estimation using deep learning',
+        container='Elife',
+        year='2019',
+        homepage='https://github.com/jgraving/DeepPoseKit-Data',
+    ),
+    keypoint_info={
+        0:
+        dict(name='head', id=0, color=[255, 255, 255], type='', swap=''),
+        1:
+        dict(name='neck', id=1, color=[255, 255, 255], type='', swap=''),
+        2:
+        dict(name='thorax', id=2, color=[255, 255, 255], type='', swap=''),
+        3:
+        dict(name='abdomen1', id=3, color=[255, 255, 255], type='', swap=''),
+        4:
+        dict(name='abdomen2', id=4, color=[255, 255, 255], type='', swap=''),
+        5:
+        dict(
+            name='anttipL',
+            id=5,
+            color=[255, 255, 255],
+            type='',
+            swap='anttipR'),
+        6:
+        dict(
+            name='antbaseL',
+            id=6,
+            color=[255, 255, 255],
+            type='',
+            swap='antbaseR'),
+        7:
+        dict(name='eyeL', id=7, color=[255, 255, 255], type='', swap='eyeR'),
+        8:
+        dict(
+            name='forelegL1',
+            id=8,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegR1'),
+        9:
+        dict(
+            name='forelegL2',
+            id=9,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegR2'),
+        10:
+        dict(
+            name='forelegL3',
+            id=10,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegR3'),
+        11:
+        dict(
+            name='forelegL4',
+            id=11,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegR4'),
+        12:
+        dict(
+            name='midlegL1',
+            id=12,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegR1'),
+        13:
+        dict(
+            name='midlegL2',
+            id=13,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegR2'),
+        14:
+        dict(
+            name='midlegL3',
+            id=14,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegR3'),
+        15:
+        dict(
+            name='midlegL4',
+            id=15,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegR4'),
+        16:
+        dict(
+            name='hindlegL1',
+            id=16,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegR1'),
+        17:
+        dict(
+            name='hindlegL2',
+            id=17,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegR2'),
+        18:
+        dict(
+            name='hindlegL3',
+            id=18,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegR3'),
+        19:
+        dict(
+            name='hindlegL4',
+            id=19,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegR4'),
+        20:
+        dict(
+            name='anttipR',
+            id=20,
+            color=[255, 255, 255],
+            type='',
+            swap='anttipL'),
+        21:
+        dict(
+            name='antbaseR',
+            id=21,
+            color=[255, 255, 255],
+            type='',
+            swap='antbaseL'),
+        22:
+        dict(name='eyeR', id=22, color=[255, 255, 255], type='', swap='eyeL'),
+        23:
+        dict(
+            name='forelegR1',
+            id=23,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegL1'),
+        24:
+        dict(
+            name='forelegR2',
+            id=24,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegL2'),
+        25:
+        dict(
+            name='forelegR3',
+            id=25,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegL3'),
+        26:
+        dict(
+            name='forelegR4',
+            id=26,
+            color=[255, 255, 255],
+            type='',
+            swap='forelegL4'),
+        27:
+        dict(
+            name='midlegR1',
+            id=27,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegL1'),
+        28:
+        dict(
+            name='midlegR2',
+            id=28,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegL2'),
+        29:
+        dict(
+            name='midlegR3',
+            id=29,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegL3'),
+        30:
+        dict(
+            name='midlegR4',
+            id=30,
+            color=[255, 255, 255],
+            type='',
+            swap='midlegL4'),
+        31:
+        dict(
+            name='hindlegR1',
+            id=31,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegL1'),
+        32:
+        dict(
+            name='hindlegR2',
+            id=32,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegL2'),
+        33:
+        dict(
+            name='hindlegR3',
+            id=33,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegL3'),
+        34:
+        dict(
+            name='hindlegR4',
+            id=34,
+            color=[255, 255, 255],
+            type='',
+            swap='hindlegL4')
+    },
+    skeleton_info={
+        0: dict(link=('neck', 'head'), id=0, color=[255, 255, 255]),
+        1: dict(link=('thorax', 'neck'), id=1, color=[255, 255, 255]),
+        2: dict(link=('abdomen1', 'thorax'), id=2, color=[255, 255, 255]),
+        3: dict(link=('abdomen2', 'abdomen1'), id=3, color=[255, 255, 255]),
+        4: dict(link=('antbaseL', 'anttipL'), id=4, color=[255, 255, 255]),
+        5: dict(link=('eyeL', 'antbaseL'), id=5, color=[255, 255, 255]),
+        6: dict(link=('forelegL2', 'forelegL1'), id=6, color=[255, 255, 255]),
+        7: dict(link=('forelegL3', 'forelegL2'), id=7, color=[255, 255, 255]),
+        8: dict(link=('forelegL4', 'forelegL3'), id=8, color=[255, 255, 255]),
+        9: dict(link=('midlegL2', 'midlegL1'), id=9, color=[255, 255, 255]),
+        10: dict(link=('midlegL3', 'midlegL2'), id=10, color=[255, 255, 255]),
+        11: dict(link=('midlegL4', 'midlegL3'), id=11, color=[255, 255, 255]),
+        12:
+        dict(link=('hindlegL2', 'hindlegL1'), id=12, color=[255, 255, 255]),
+        13:
+        dict(link=('hindlegL3', 'hindlegL2'), id=13, color=[255, 255, 255]),
+        14:
+        dict(link=('hindlegL4', 'hindlegL3'), id=14, color=[255, 255, 255]),
+        15: dict(link=('antbaseR', 'anttipR'), id=15, color=[255, 255, 255]),
+        16: dict(link=('eyeR', 'antbaseR'), id=16, color=[255, 255, 255]),
+        17:
+        dict(link=('forelegR2', 'forelegR1'), id=17, color=[255, 255, 255]),
+        18:
+        dict(link=('forelegR3', 'forelegR2'), id=18, color=[255, 255, 255]),
+        19:
+        dict(link=('forelegR4', 'forelegR3'), id=19, color=[255, 255, 255]),
+        20: dict(link=('midlegR2', 'midlegR1'), id=20, color=[255, 255, 255]),
+        21: dict(link=('midlegR3', 'midlegR2'), id=21, color=[255, 255, 255]),
+        22: dict(link=('midlegR4', 'midlegR3'), id=22, color=[255, 255, 255]),
+        23:
+        dict(link=('hindlegR2', 'hindlegR1'), id=23, color=[255, 255, 255]),
+        24:
+        dict(link=('hindlegR3', 'hindlegR2'), id=24, color=[255, 255, 255]),
+        25:
+        dict(link=('hindlegR4', 'hindlegR3'), id=25, color=[255, 255, 255])
+    },
+    joint_weights=[1.] * 35,
+    sigmas=[])

configs/_base_/datasets/macaque.py

@@ -0,0 +1,189 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='macaque',
+    paper_info=dict(
+        author='Labuguen, Rollyn and Matsumoto, Jumpei and '
+        'Negrete, Salvador and Nishimaru, Hiroshi and '
+        'Nishijo, Hisao and Takada, Masahiko and '
+        'Go, Yasuhiro and Inoue, Ken-ichi and Shibata, Tomohiro',
+        title='MacaquePose: A novel "in the wild" macaque monkey pose dataset '
+        'for markerless motion capture',
+        container='bioRxiv',
+        year='2020',
+        homepage='http://www.pri.kyoto-u.ac.jp/datasets/'
+        'macaquepose/index.html',
+    ),
+    keypoint_info={
+        0:
+        dict(name='nose', id=0, color=[51, 153, 255], type='upper', swap=''),
+        1:
+        dict(
+            name='left_eye',
+            id=1,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_eye'),
+        2:
+        dict(
+            name='right_eye',
+            id=2,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_eye'),
+        3:
+        dict(
+            name='left_ear',
+            id=3,
+            color=[51, 153, 255],
+            type='upper',
+            swap='right_ear'),
+        4:
+        dict(
+            name='right_ear',
+            id=4,
+            color=[51, 153, 255],
+            type='upper',
+            swap='left_ear'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='right_shoulder',
+            id=6,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        7:
+        dict(
+            name='left_elbow',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        8:
+        dict(
+            name='right_elbow',
+            id=8,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        9:
+        dict(
+            name='left_wrist',
+            id=9,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='right_hip',
+            id=12,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        13:
+        dict(
+            name='left_knee',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        14:
+        dict(
+            name='right_knee',
+            id=14,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        15:
+        dict(
+            name='left_ankle',
+            id=15,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        16:
+        dict(
+            name='right_ankle',
+            id=16,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle')
+    },
+    skeleton_info={
+        0:
+        dict(link=('left_ankle', 'left_knee'), id=0, color=[0, 255, 0]),
+        1:
+        dict(link=('left_knee', 'left_hip'), id=1, color=[0, 255, 0]),
+        2:
+        dict(link=('right_ankle', 'right_knee'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('right_knee', 'right_hip'), id=3, color=[255, 128, 0]),
+        4:
+        dict(link=('left_hip', 'right_hip'), id=4, color=[51, 153, 255]),
+        5:
+        dict(link=('left_shoulder', 'left_hip'), id=5, color=[51, 153, 255]),
+        6:
+        dict(link=('right_shoulder', 'right_hip'), id=6, color=[51, 153, 255]),
+        7:
+        dict(
+            link=('left_shoulder', 'right_shoulder'),
+            id=7,
+            color=[51, 153, 255]),
+        8:
+        dict(link=('left_shoulder', 'left_elbow'), id=8, color=[0, 255, 0]),
+        9:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=9, color=[255, 128, 0]),
+        10:
+        dict(link=('left_elbow', 'left_wrist'), id=10, color=[0, 255, 0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('left_eye', 'right_eye'), id=12, color=[51, 153, 255]),
+        13:
+        dict(link=('nose', 'left_eye'), id=13, color=[51, 153, 255]),
+        14:
+        dict(link=('nose', 'right_eye'), id=14, color=[51, 153, 255]),
+        15:
+        dict(link=('left_eye', 'left_ear'), id=15, color=[51, 153, 255]),
+        16:
+        dict(link=('right_eye', 'right_ear'), id=16, color=[51, 153, 255]),
+        17:
+        dict(link=('left_ear', 'left_shoulder'), id=17, color=[51, 153, 255]),
+        18:
+        dict(
+            link=('right_ear', 'right_shoulder'), id=18, color=[51, 153, 255])
+    },
+    joint_weights=[
+        1., 1., 1., 1., 1., 1., 1., 1.2, 1.2, 1.5, 1.5, 1., 1., 1.2, 1.2, 1.5,
+        1.5
+    ],
+    sigmas=[
+        0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062,
+        0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089
+    ])

configs/_base_/datasets/mhp.py

@@ -0,0 +1,162 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='mhp',
+    paper_info=dict(
+        author='Zhao, Jian and Li, Jianshu and Cheng, Yu and '
+        'Sim, Terence and Yan, Shuicheng and Feng, Jiashi',
+        title='Understanding humans in crowded scenes: '
+        'Deep nested adversarial learning and a '
+        'new benchmark for multi-human parsing',
+        container='Proceedings of the 26th ACM '
+        'international conference on Multimedia',
+        year='2018',
+        homepage='https://lv-mhp.github.io/dataset',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='right_ankle',
+            id=0,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle'),
+        1:
+        dict(
+            name='right_knee',
+            id=1,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        2:
+        dict(
+            name='right_hip',
+            id=2,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        3:
+        dict(
+            name='left_hip',
+            id=3,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        4:
+        dict(
+            name='left_knee',
+            id=4,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        5:
+        dict(
+            name='left_ankle',
+            id=5,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        6:
+        dict(name='pelvis', id=6, color=[51, 153, 255], type='lower', swap=''),
+        7:
+        dict(name='thorax', id=7, color=[51, 153, 255], type='upper', swap=''),
+        8:
+        dict(
+            name='upper_neck',
+            id=8,
+            color=[51, 153, 255],
+            type='upper',
+            swap=''),
+        9:
+        dict(
+            name='head_top', id=9, color=[51, 153, 255], type='upper',
+            swap=''),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='right_elbow',
+            id=11,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        12:
+        dict(
+            name='right_shoulder',
+            id=12,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        13:
+        dict(
+            name='left_shoulder',
+            id=13,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        14:
+        dict(
+            name='left_elbow',
+            id=14,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        15:
+        dict(
+            name='left_wrist',
+            id=15,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist')
+    },
+    skeleton_info={
+        0:
+        dict(link=('right_ankle', 'right_knee'), id=0, color=[255, 128, 0]),
+        1:
+        dict(link=('right_knee', 'right_hip'), id=1, color=[255, 128, 0]),
+        2:
+        dict(link=('right_hip', 'pelvis'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('pelvis', 'left_hip'), id=3, color=[0, 255, 0]),
+        4:
+        dict(link=('left_hip', 'left_knee'), id=4, color=[0, 255, 0]),
+        5:
+        dict(link=('left_knee', 'left_ankle'), id=5, color=[0, 255, 0]),
+        6:
+        dict(link=('pelvis', 'thorax'), id=6, color=[51, 153, 255]),
+        7:
+        dict(link=('thorax', 'upper_neck'), id=7, color=[51, 153, 255]),
+        8:
+        dict(link=('upper_neck', 'head_top'), id=8, color=[51, 153, 255]),
+        9:
+        dict(link=('upper_neck', 'right_shoulder'), id=9, color=[255, 128, 0]),
+        10:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=10, color=[255, 128,
+                                                                  0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('upper_neck', 'left_shoulder'), id=12, color=[0, 255, 0]),
+        13:
+        dict(link=('left_shoulder', 'left_elbow'), id=13, color=[0, 255, 0]),
+        14:
+        dict(link=('left_elbow', 'left_wrist'), id=14, color=[0, 255, 0])
+    },
+    joint_weights=[
+        1.5, 1.2, 1., 1., 1.2, 1.5, 1., 1., 1., 1., 1.5, 1.2, 1., 1., 1.2, 1.5
+    ],
+    # Adapted from COCO dataset.
+    sigmas=[
+        0.089, 0.083, 0.107, 0.107, 0.083, 0.089, 0.026, 0.026, 0.026, 0.026,
+        0.062, 0.072, 0.179, 0.179, 0.072, 0.062
+    ])

configs/_base_/datasets/mpi_inf_3dhp.py

@@ -0,0 +1,138 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='mpi_inf_3dhp',
+    paper_info=dict(
+        author='ehta, Dushyant and Rhodin, Helge and Casas, Dan and '
+        'Fua, Pascal and Sotnychenko, Oleksandr and Xu, Weipeng and '
+        'Theobalt, Christian',
+        title='Monocular 3D Human Pose Estimation In The Wild Using Improved '
+        'CNN Supervision',
+        container='2017 international conference on 3D vision (3DV)',
+        year='2017',
+        homepage='http://gvv.mpi-inf.mpg.de/3dhp-dataset',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='head_top', id=0, color=[51, 153, 255], type='upper',
+            swap=''),
+        1:
+        dict(name='neck', id=1, color=[51, 153, 255], type='upper', swap=''),
+        2:
+        dict(
+            name='right_shoulder',
+            id=2,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        3:
+        dict(
+            name='right_elbow',
+            id=3,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        4:
+        dict(
+            name='right_wrist',
+            id=4,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        5:
+        dict(
+            name='left_shoulder',
+            id=5,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        6:
+        dict(
+            name='left_elbow',
+            id=6,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        7:
+        dict(
+            name='left_wrist',
+            id=7,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist'),
+        8:
+        dict(
+            name='right_hip',
+            id=8,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        9:
+        dict(
+            name='right_knee',
+            id=9,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        10:
+        dict(
+            name='right_ankle',
+            id=10,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle'),
+        11:
+        dict(
+            name='left_hip',
+            id=11,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        12:
+        dict(
+            name='left_knee',
+            id=12,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        13:
+        dict(
+            name='left_ankle',
+            id=13,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        14:
+        dict(name='root', id=14, color=[51, 153, 255], type='lower', swap=''),
+        15:
+        dict(name='spine', id=15, color=[51, 153, 255], type='upper', swap=''),
+        16:
+        dict(name='head', id=16, color=[51, 153, 255], type='upper', swap='')
+    },
+    skeleton_info={
+        0: dict(link=('neck', 'right_shoulder'), id=0, color=[255, 128, 0]),
+        1: dict(
+            link=('right_shoulder', 'right_elbow'), id=1, color=[255, 128, 0]),
+        2:
+        dict(link=('right_elbow', 'right_wrist'), id=2, color=[255, 128, 0]),
+        3: dict(link=('neck', 'left_shoulder'), id=3, color=[0, 255, 0]),
+        4: dict(link=('left_shoulder', 'left_elbow'), id=4, color=[0, 255, 0]),
+        5: dict(link=('left_elbow', 'left_wrist'), id=5, color=[0, 255, 0]),
+        6: dict(link=('root', 'right_hip'), id=6, color=[255, 128, 0]),
+        7: dict(link=('right_hip', 'right_knee'), id=7, color=[255, 128, 0]),
+        8: dict(link=('right_knee', 'right_ankle'), id=8, color=[255, 128, 0]),
+        9: dict(link=('root', 'left_hip'), id=9, color=[0, 255, 0]),
+        10: dict(link=('left_hip', 'left_knee'), id=10, color=[0, 255, 0]),
+        11: dict(link=('left_knee', 'left_ankle'), id=11, color=[0, 255, 0]),
+        12: dict(link=('head_top', 'head'), id=12, color=[51, 153, 255]),
+        13: dict(link=('head', 'neck'), id=13, color=[51, 153, 255]),
+        14: dict(link=('neck', 'spine'), id=14, color=[51, 153, 255]),
+        15: dict(link=('spine', 'root'), id=15, color=[51, 153, 255])
+    },
+    joint_weights=[1.] * 17,
+    sigmas=[])

configs/_base_/datasets/mpii.py

@@ -0,0 +1,161 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+dataset_info = dict(
+    dataset_name='mpii',
+    paper_info=dict(
+        author='Mykhaylo Andriluka and Leonid Pishchulin and '
+        'Peter Gehler and Schiele, Bernt',
+        title='2D Human Pose Estimation: New Benchmark and '
+        'State of the Art Analysis',
+        container='IEEE Conference on Computer Vision and '
+        'Pattern Recognition (CVPR)',
+        year='2014',
+        homepage='http://human-pose.mpi-inf.mpg.de/',
+    ),
+    keypoint_info={
+        0:
+        dict(
+            name='right_ankle',
+            id=0,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_ankle'),
+        1:
+        dict(
+            name='right_knee',
+            id=1,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_knee'),
+        2:
+        dict(
+            name='right_hip',
+            id=2,
+            color=[255, 128, 0],
+            type='lower',
+            swap='left_hip'),
+        3:
+        dict(
+            name='left_hip',
+            id=3,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_hip'),
+        4:
+        dict(
+            name='left_knee',
+            id=4,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_knee'),
+        5:
+        dict(
+            name='left_ankle',
+            id=5,
+            color=[0, 255, 0],
+            type='lower',
+            swap='right_ankle'),
+        6:
+        dict(name='pelvis', id=6, color=[51, 153, 255], type='lower', swap=''),
+        7:
+        dict(name='thorax', id=7, color=[51, 153, 255], type='upper', swap=''),
+        8:
+        dict(
+            name='upper_neck',
+            id=8,
+            color=[51, 153, 255],
+            type='upper',
+            swap=''),
+        9:
+        dict(
+            name='head_top', id=9, color=[51, 153, 255], type='upper',
+            swap=''),
+        10:
+        dict(
+            name='right_wrist',
+            id=10,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_wrist'),
+        11:
+        dict(
+            name='right_elbow',
+            id=11,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_elbow'),
+        12:
+        dict(
+            name='right_shoulder',
+            id=12,
+            color=[255, 128, 0],
+            type='upper',
+            swap='left_shoulder'),
+        13:
+        dict(
+            name='left_shoulder',
+            id=13,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_shoulder'),
+        14:
+        dict(
+            name='left_elbow',
+            id=14,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_elbow'),
+        15:
+        dict(
+            name='left_wrist',
+            id=15,
+            color=[0, 255, 0],
+            type='upper',
+            swap='right_wrist')
+    },
+    skeleton_info={
+        0:
+        dict(link=('right_ankle', 'right_knee'), id=0, color=[255, 128, 0]),
+        1:
+        dict(link=('right_knee', 'right_hip'), id=1, color=[255, 128, 0]),
+        2:
+        dict(link=('right_hip', 'pelvis'), id=2, color=[255, 128, 0]),
+        3:
+        dict(link=('pelvis', 'left_hip'), id=3, color=[0, 255, 0]),
+        4:
+        dict(link=('left_hip', 'left_knee'), id=4, color=[0, 255, 0]),
+        5:
+        dict(link=('left_knee', 'left_ankle'), id=5, color=[0, 255, 0]),
+        6:
+        dict(link=('pelvis', 'thorax'), id=6, color=[51, 153, 255]),
+        7:
+        dict(link=('thorax', 'upper_neck'), id=7, color=[51, 153, 255]),
+        8:
+        dict(link=('upper_neck', 'head_top'), id=8, color=[51, 153, 255]),
+        9:
+        dict(link=('upper_neck', 'right_shoulder'), id=9, color=[255, 128, 0]),
+        10:
+        dict(
+            link=('right_shoulder', 'right_elbow'), id=10, color=[255, 128,
+                                                                  0]),
+        11:
+        dict(link=('right_elbow', 'right_wrist'), id=11, color=[255, 128, 0]),
+        12:
+        dict(link=('upper_neck', 'left_shoulder'), id=12, color=[0, 255, 0]),
+        13:
+        dict(link=('left_shoulder', 'left_elbow'), id=13, color=[0, 255, 0]),
+        14:
+        dict(link=('left_elbow', 'left_wrist'), id=14, color=[0, 255, 0])
+    },
+    joint_weights=[
+        1.5, 1.2, 1., 1., 1.2, 1.5, 1., 1., 1., 1., 1.5, 1.2, 1., 1., 1.2, 1.5
+    ],
+    # Adapted from COCO dataset.
+    sigmas=[
+        0.089, 0.083, 0.107, 0.107, 0.083, 0.089, 0.026, 0.026, 0.026, 0.026,
+        0.062, 0.072, 0.179, 0.179, 0.072, 0.062
+    ])