crosstyan/RapidPoseTriangulation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Some small updates.

Browse Source
This commit is contained in:
Daniel
2024-12-16 14:34:44 +01:00
parent 07f75c53ee
commit 17ff41010a
3 changed files with 25 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/utils_2d_pose.py
+3 -3
View File
@@ -46,7 +46,7 @@ def load_wb_model():
def get_2d_pose(model, imgs, num_joints=17):
"""See: https://mmpose.readthedocs.io/en/latest/user_guides/inference.html#basic-usage"""
result_generator = model(imgs, show=False)
result_generator = model(imgs, show=False, bbox_thr=0.3, nms_thr=0.3)
new_poses = []
for _ in range(len(imgs)):
result = next(result_generator)
@@ -56,8 +56,8 @@ def get_2d_pose(model, imgs, num_joints=17):
kpts = result["predictions"][0][i]["keypoints"]
scores = result["predictions"][0][i]["keypoint_scores"]
kpts = np.array(kpts)
scores = np.array(scores).reshape(-1, 1)
kpts = np.asarray(kpts)
scores = np.asarray(scores).reshape(-1, 1)
scores = np.clip(scores, 0, 1)
pose = np.concatenate((kpts, scores), axis=-1)
poses.append(pose)