Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
6b3fae3
1
Parent(s):
defab3d
add
Browse files
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.
|
298 |
-
rec_pose_np = rec_pose.
|
299 |
-
rec_trans_np = rec_trans.
|
300 |
-
rec_exp_np = rec_exps.
|
301 |
-
tar_exp_np = tar_exps.
|
302 |
-
tar_trans_np = tar_trans.
|
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 |
|