def __init__()

in empose/eval/metrics.py [0:0]

• 14 lines of code
• 7 McCabe index (conditional complexity)

    def __init__(self, smpl_model):
        """
        :param smpl_model: The SMPL model.
        """
        self.smpl_model = smpl_model
        self.eucl_dists = []  # list of Euclidean distance to ground-truth for each sample and each joint
        self.eucl_dists_pa = []  # Same as self.eucl_dists but Procrustes-aligned
        self.angle_diffs = []  # list of angular difference for each sample

        # List of joints to consider for either metric.
        self.eucl_eval_joints = ['root', 'l_hip', 'r_hip', 'spine1', 'l_knee', 'r_knee', 'spine2', 'l_ankle', 'r_ankle',
                                 'spine3', 'neck', 'l_collar', 'r_collar', 'head', 'l_shoulder', 'r_shoulder',
                                 'l_elbow', 'r_elbow', 'l_wrist', 'r_wrist']
        self.angle_eval_joints = ['l_hip', 'r_hip', 'spine1', 'l_knee', 'r_knee', 'spine2', 'spine3',
                                  'neck', 'l_collar', 'r_collar', 'head', 'l_shoulder', 'r_shoulder',
                                  'l_elbow', 'r_elbow']

        self.eucl_idxs = [C.SMPL_JOINTS.index(j) for j in self.eucl_eval_joints]

        # The pose vector does not contain the root, so adjust index by -1.
        self.angle_idxs = [C.SMPL_JOINTS.index(j)-1 for j in self.angle_eval_joints]
        self.angle_glob = True