Update run_mmpose.py

mks0601 · Sep 29, 2024 · 62db97b · 62db97b
1 parent 0141bd6
commit 62db97b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/fitting/tools/code_to_copy/run_mmpose.py b/fitting/tools/code_to_copy/run_mmpose.py
@@ -36,16 +36,15 @@ def parse_args():
     with open(output_path) as f:
         out = json.load(f)
 
-    idx_save, kpt_save = None, None
+    kpt_save = None
     for i in range(len(out['instance_info'])):
         xy = np.array(out['instance_info'][i]['keypoints'], dtype=np.float32).reshape(-1,2)
         score = np.array(out['instance_info'][i]['keypoint_scores'], dtype=np.float32).reshape(-1,1)
         kpt = np.concatenate((xy, score),1) # x, y, score
         if (kpt_save is None) or (kpt_save[:,2].mean() < kpt[:,2].mean()):
-            idx_save = i
             kpt_save = kpt
     with open(osp.join(root_path, 'keypoints_whole_body', str(frame_idx) + '.json'), 'w') as f:
-        json.dump(kpt.tolist(), f)
+        json.dump(kpt_save.tolist(), f)
 
 # add original image and frame index to the video
 output_path_list = glob(osp.join(output_root, '*.png'))