Function get_smpl_model

mbench/pose_quality.py:23–36

Load the SMPL model. :param gender: 'neutral', 'male', or 'female' :param model_path: Path to the SMPL model files :return: SMPL model

(gender='neutral', model_path='data/body_models/smpl', batch_size=1, device='cuda')

    }

def get_smpl_model(gender='neutral', model_path='data/body_models/smpl', batch_size=1, device='cuda'):
    """
    Load the SMPL model.
    :param gender: 'neutral', 'male', or 'female'
    :param model_path: Path to the SMPL model files
    :return: SMPL model
    """
    kwargs = {'model_path': model_path, 
              'gender': 'neutral', 
              'num_betas': 16,
              'batch_size': batch_size,
              'use_pca': False}
    smpl_model = SMPL(**kwargs).to(device)
    return smpl_model

def smpl_to_joints_and_verts(smpl_model, smpl_params):
    """