smplx pkl格式可视化

import glob
import os
import pickleimport torch
import numpy as npfrom smplpytorch.pytorch.smpl_layer import SMPL_Layer
from display_utils import display_model, display_model_continuousfrom matplotlib import pyplot as plt
from matplotlib.animation import FuncAnimation
from mpl_toolkits.mplot3d.art3d import Poly3DCollectiondef draw_skeleton(joints3D, kintree_table, ax=None, with_numbers=True):if ax is None:fig = plt.figure(frameon=False)ax = fig.add_subplot(111, projection='3d')else:ax = axcolors = []left_right_mid = ['r', 'g', 'b']kintree_colors = [2, 0, 1, 2, 0, 1, 2, 0, 1, 2, 0, 1, 2, 0, 1, 2, 0, 1, 0, 1, 0, 1, 0, 1]for c in kintree_colors:colors &