From 981027e502cabddc957c9df83a8e78d0649bb60a Mon Sep 17 00:00:00 2001 From: Rabbit314271 <59315380+Rabbit314271@users.noreply.github.com> Date: Wed, 6 Jul 2022 19:57:45 +0800 Subject: [PATCH] Update eval_AGORA.py --- simple_romp/evaluation/eval_AGORA.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/simple_romp/evaluation/eval_AGORA.py b/simple_romp/evaluation/eval_AGORA.py index ed4ccfb..e08c6d6 100755 --- a/simple_romp/evaluation/eval_AGORA.py +++ b/simple_romp/evaluation/eval_AGORA.py @@ -83,8 +83,8 @@ def save_agora_predictions_v6(image_path, outputs, save_dir): else: cam_params = np.array([[995.55555556, 0., 640.],[0.,995.55555556,360.],[0.,0.,1.]]) - predicts_j3ds = outputs['joints'].contiguous().cpu().numpy() - predicts_pj2ds = outputs['pj2d_org'].cpu().numpy() + predicts_j3ds = torch.from_numpy(outputs['joints']).contiguous().cpu().numpy() + predicts_pj2ds = torch.from_numpy(outputs['pj2d_org']).cpu().numpy() predicts_j3ds = predicts_j3ds[:,:24] - predicts_j3ds[:,[0]] predicts_pj2ds = predicts_pj2ds[:,:24] outputs['cam_trans'] = estimate_translation(predicts_j3ds, predicts_pj2ds, outputs['cam_trans'],\ @@ -123,4 +123,4 @@ def get_results_on_AGORA(set_name='test'): if __name__ == '__main__': - get_results_on_AGORA(set_name) \ No newline at end of file + get_results_on_AGORA(set_name)