import os

os.environ["CUDA_VISIBLE_DEVICES"] = '6'

import numpy as np
from smpl_tf import SMPL
from utils import plot_mesh

if __name__ == '__main__':
    smpl_tf : SMPL = SMPL('model_files')

    betas = np.zeros((1, 11), dtype=np.float32)
    poses = np.zeros((1, 72), dtype=np.float32)
    R = smpl_tf.run_rodrigues(poses)
    posed, _, _ = smpl_tf.run(R, betas)

    plot_mesh(posed[0].numpy(), smpl_tf.f).show()

    pass