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

Updated skelda version.

Browse Source
This commit is contained in:
Daniel
2024-10-14 16:10:19 +02:00
parent 87a205c36c
commit 256fd26f3f
4 changed files with 10 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/test_skelda_dataset.py
View File

@ -197,10 +197,6 @@ def load_labels(dataset: dict):
if take_interval > 1:
labels = [l for i, l in enumerate(labels) if i % take_interval == 0]
# Filter joints
fj_func = lambda x: utils_pose.filter_joints_3d(x, eval_joints)
labels = list(map(fj_func, labels))
return labels
@ -332,7 +328,7 @@ def main():
print("3D time:", time_3d)
old_index = label["index"]
all_poses.append(np.array(poses3D))
all_poses.append(np.array(poses3D).tolist())
all_ids.append(label["id"])
all_paths.append(label["imgpaths"])
times.append((time_2d, time_3d))