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

Updated ikeaasm and tsinghua dataset splits. Small other updates.

Browse Source
This commit is contained in:
Daniel
2024-07-05 16:15:07 +02:00
parent 7330b010b5
commit aea1d8bfb2
3 changed files with 575 additions and 569 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/test_skelda_dataset.py
View File

@ -141,7 +141,8 @@ def load_labels(dataset: dict):
elif "ikeaasm" in dataset:
labels = load_json(dataset["ikeaasm"]["path"])
labels = [lb for i, lb in enumerate(labels) if i % 300 < 72]
cams0 = str(labels[0]["cameras"])
labels = [lb for lb in labels if str(lb["cameras"]) == cams0]
elif "shelf" in dataset:
labels = load_json(dataset["shelf"]["path"])
@ -154,7 +155,8 @@ def load_labels(dataset: dict):
elif "tsinghua" in dataset:
labels = load_json(dataset["tsinghua"]["path"])
labels = [lb for lb in labels if "test" in lb["splits"]]
labels = [lb for i, lb in enumerate(labels) if i % 800 < 90]
labels = [lb for lb in labels if lb["seq"] == "seq_1"]
labels = [lb for i, lb in enumerate(labels) if i % 300 < 90]
for label in labels:
label["bodyids"] = list(range(len(label["bodies3D"])))
@ -237,6 +239,9 @@ def add_missing_joints(poses3D, joint_names_3d):
for i in range(len(poses3D)):
valid_joints = np.where(poses3D[i, :, 3] > 0.1)[0]
if len(valid_joints) == 0:
continue
body_center = np.mean(poses3D[i, valid_joints, :3], axis=0)
for j in range(len(joint_names_3d)):
@ -360,7 +365,7 @@ def main():
"mvor": 0.86,
"campus": 0.96,
"shelf": 0.96,
"ikeaasm": 0.86,
"ikeaasm": 0.89,
"tsinghua": 0.96,
"human36m_wb": 0.94,
}
@ -416,6 +421,7 @@ def main():
joint_names_net=joint_names_3d,
joint_names_use=eval_joints,
save_error_imgs=output_dir,
pred_imgpaths=all_paths,
)
_ = evals.pcp.run_eval(
labels,

2
scripts/test_triangulate.py
View File

@ -306,7 +306,7 @@ def filter_poses(poses3D, poses2D, roomparams, joint_names, drop_few_limbs=True)
if average_length < 0.1:
drop.append(i)
continue
if drop_few_limbs and average_length > 0.5:
if total_limbs > 4 and average_length > 0.5:
drop.append(i)
continue