add ploting

2023-02-03 11:40:55 +00:00
parent ceadca02e3
commit aa13346b52
3 changed files with 30 additions and 2 deletions
@@ -4,6 +4,7 @@ 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')
@@ -11,7 +12,8 @@ if __name__ == '__main__':
    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)
+    posed, _, _ = smpl_tf.run(R, betas)

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

    pass