From 9aa8bf5729ec10f2d3e3a390d80f7e8427543766 Mon Sep 17 00:00:00 2001 From: xiexinch Date: Thu, 25 Apr 2024 00:32:59 +0800 Subject: [PATCH] --fix=fix infer video --- projects/rtmpose3d/body3d_img2pose_demo.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/projects/rtmpose3d/body3d_img2pose_demo.py b/projects/rtmpose3d/body3d_img2pose_demo.py index 200043d7d4..e32db3323a 100644 --- a/projects/rtmpose3d/body3d_img2pose_demo.py +++ b/projects/rtmpose3d/body3d_img2pose_demo.py @@ -216,8 +216,6 @@ def process_one_image(args, detector, frame: np.ndarray, frame_idx: int, keypoints = np.squeeze(keypoints, axis=1) keypoints = -keypoints[..., [0, 2, 1]] - # keypoints[..., 0] = -keypoints[..., 0] - # keypoints[..., 2] = -keypoints[..., 2] # rebase height (z-axis) if not args.disable_rebase_keypoint: @@ -254,8 +252,7 @@ def process_one_image(args, detector, frame: np.ndarray, frame_idx: int, axis_azimuth=70, axis_elev=15, num_instances=args.num_instances, - wait_time=args.show_interval, - root_index=[11, 12]) + wait_time=args.show_interval) return pose_est_results, pose_est_results_list, pred_3d_instances, next_id