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

1130
media/RESULTS.md
View File

File diff suppressed because it is too large Load Diff

12
scripts/test_skelda_dataset.py
View File

@ -141,7 +141,8 @@ def load_labels(dataset: dict):
elif "ikeaasm" in dataset: elif "ikeaasm" in dataset:
labels = load_json(dataset["ikeaasm"]["path"]) 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: elif "shelf" in dataset:
labels = load_json(dataset["shelf"]["path"]) labels = load_json(dataset["shelf"]["path"])
@ -154,7 +155,8 @@ def load_labels(dataset: dict):
elif "tsinghua" in dataset: elif "tsinghua" in dataset:
labels = load_json(dataset["tsinghua"]["path"]) labels = load_json(dataset["tsinghua"]["path"])
labels = [lb for lb in labels if "test" in lb["splits"]] 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: for label in labels:
label["bodyids"] = list(range(len(label["bodies3D"]))) 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)): for i in range(len(poses3D)):
valid_joints = np.where(poses3D[i, :, 3] > 0.1)[0] 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) body_center = np.mean(poses3D[i, valid_joints, :3], axis=0)
for j in range(len(joint_names_3d)): for j in range(len(joint_names_3d)):
@ -360,7 +365,7 @@ def main():
"mvor": 0.86, "mvor": 0.86,
"campus": 0.96, "campus": 0.96,
"shelf": 0.96, "shelf": 0.96,
"ikeaasm": 0.86, "ikeaasm": 0.89,
"tsinghua": 0.96, "tsinghua": 0.96,
"human36m_wb": 0.94, "human36m_wb": 0.94,
} }
@ -416,6 +421,7 @@ def main():
joint_names_net=joint_names_3d, joint_names_net=joint_names_3d,
joint_names_use=eval_joints, joint_names_use=eval_joints,
save_error_imgs=output_dir, save_error_imgs=output_dir,
pred_imgpaths=all_paths,
) )
_ = evals.pcp.run_eval( _ = evals.pcp.run_eval(
labels, 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: if average_length < 0.1:
drop.append(i) drop.append(i)
continue continue
if drop_few_limbs and average_length > 0.5: if total_limbs > 4 and average_length > 0.5:
drop.append(i) drop.append(i)
continue continue