Yzy00518 commited on
Commit
4f44d19
·
verified ·
1 Parent(s): 60c3e07

Update src/app/render.py

Browse files
Files changed (1) hide show
  1. src/app/render.py +3 -1
src/app/render.py CHANGED
@@ -9,7 +9,7 @@ import pyrender
9
 
10
  root_dir = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), '../..'))
11
  def get_smplx_model(bs, deivce):
12
- smpl_model = smplx.create(os.path.join(root_dir, 'deps/smplx/smplx_models'),
13
  model_type='smplx',
14
  gender='male',
15
  ext='npz',
@@ -122,4 +122,6 @@ if __name__ == "__main__":
122
  motion = pickle.load(f)
123
  output_path = args.motion_path.replace(".pkl", ".mp4")
124
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 
125
  render_motion_to_video(motion, device, output_path, args.title)
 
9
 
10
  root_dir = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), '../..'))
11
  def get_smplx_model(bs, deivce):
12
+ smpl_model = smplx.create(os.path.join(root_dir, 'deps/smplx/models'),
13
  model_type='smplx',
14
  gender='male',
15
  ext='npz',
 
122
  motion = pickle.load(f)
123
  output_path = args.motion_path.replace(".pkl", ".mp4")
124
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
125
+
126
+ print(args.title)
127
  render_motion_to_video(motion, device, output_path, args.title)