Tharun156 commited on
Commit
b5725c3
Β·
verified Β·
1 Parent(s): 5d3d5c9

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +5 -6
demo.py CHANGED
@@ -117,16 +117,15 @@ class BaseTrainer(object):
117
  logger.info(self.model)
118
  logger.info(f"init {cfg.model.g_name} success")
119
 
120
- self.smplx = smplx.create(
121
- self.args.data_path_1+"smplx_models/",
122
- model_type='smplx',
123
- gender='NEUTRAL_2020',
124
  use_face_contour=False,
125
  num_betas=300,
126
- num_expression_coeffs=100,
127
  ext='npz',
128
  use_pca=False,
129
- ).eval()
130
 
131
  self.args = args
132
  self.ori_joint_list = joints_list[self.args.ori_joints]
 
117
  logger.info(self.model)
118
  logger.info(f"init {cfg.model.g_name} success")
119
 
120
+ self.smplx = smplx.SMPLX(
121
+ model_path=self.args.data_path_1 + "smplx_models/",
122
+ gender='NEUTRAL_2020',
 
123
  use_face_contour=False,
124
  num_betas=300,
125
+ num_expression_coeffs=100,
126
  ext='npz',
127
  use_pca=False,
128
+ ).eval()
129
 
130
  self.args = args
131
  self.ori_joint_list = joints_list[self.args.ori_joints]