robinwitch commited on
Commit
6b3fae3
·
1 Parent(s): defab3d
Files changed (1) hide show
  1. app.py +13 -13
app.py CHANGED
@@ -294,12 +294,12 @@ class BaseTrainer(object):
294
  tar_pose = rc.matrix_to_axis_angle(tar_pose).reshape(bs*n, j*3)
295
 
296
 
297
- tar_pose_np = tar_pose.detach().cpu().numpy()
298
- rec_pose_np = rec_pose.detach().cpu().numpy()
299
- rec_trans_np = rec_trans.detach().cpu().numpy().reshape(bs*n, 3)
300
- rec_exp_np = rec_exps.detach().cpu().numpy().reshape(bs*n, 100)
301
- tar_exp_np = tar_exps.detach().cpu().numpy().reshape(bs*n, 100)
302
- tar_trans_np = tar_trans.detach().cpu().numpy().reshape(bs*n, 3)
303
  gt_npz = np.load("./demo/examples/2_scott_0_1_1.npz", allow_pickle=True)
304
 
305
  results_npz_file_save_path = results_save_path+f"result_{self.time_name_expend[:-1]}"+'.npz'
@@ -528,13 +528,13 @@ def _warp_g_test(loaded_data,diffusion,args,joints,joint_mask_upper,joint_mask_h
528
  tar_pose = rc.matrix_to_rotation_6d(tar_pose).reshape(bs, n, j*6)
529
 
530
  return {
531
- 'rec_pose': rec_pose,
532
- 'rec_trans': rec_trans,
533
- 'tar_pose': tar_pose,
534
- 'tar_exps': tar_exps,
535
- 'tar_beta': tar_beta,
536
- 'tar_trans': tar_trans,
537
- 'rec_exps': rec_exps,
538
  }
539
 
540
 
 
294
  tar_pose = rc.matrix_to_axis_angle(tar_pose).reshape(bs*n, j*3)
295
 
296
 
297
+ tar_pose_np = tar_pose.numpy()
298
+ rec_pose_np = rec_pose.numpy()
299
+ rec_trans_np = rec_trans.numpy().reshape(bs*n, 3)
300
+ rec_exp_np = rec_exps.numpy().reshape(bs*n, 100)
301
+ tar_exp_np = tar_exps.numpy().reshape(bs*n, 100)
302
+ tar_trans_np = tar_trans.numpy().reshape(bs*n, 3)
303
  gt_npz = np.load("./demo/examples/2_scott_0_1_1.npz", allow_pickle=True)
304
 
305
  results_npz_file_save_path = results_save_path+f"result_{self.time_name_expend[:-1]}"+'.npz'
 
528
  tar_pose = rc.matrix_to_rotation_6d(tar_pose).reshape(bs, n, j*6)
529
 
530
  return {
531
+ 'rec_pose': rec_pose.detach().cpu(),
532
+ 'rec_trans': rec_trans.detach().cpu(),
533
+ 'tar_pose': tar_pose.detach().cpu(),
534
+ 'tar_exps': tar_exps.detach().cpu(),
535
+ 'tar_beta': tar_beta.detach().cpu(),
536
+ 'tar_trans': tar_trans.detach().cpu(),
537
+ 'rec_exps': rec_exps.detach().cpu(),
538
  }
539
 
540