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

Dropping more old python code.

Browse Source
This commit is contained in:
Daniel
2025-01-28 11:50:50 +01:00
parent a674811fef
commit 1cc71cce57
5 changed files with 102 additions and 699 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
scripts/test_skelda_dataset.py
View File

@ -203,25 +203,11 @@ output_dir = ""
# ==================================================================================================
def load_json(path: str):
with open(path, "r", encoding="utf-8") as file:
data = json.load(file)
return data
def save_json(data: dict, path: str):
with open(path, "w+", encoding="utf-8") as file:
json.dump(data, file, indent=0)
# ==================================================================================================
def load_labels(dataset: dict):
"""Load labels by dataset description"""
if "panoptic" in dataset:
labels = load_json(dataset["panoptic"]["path"])
labels = utils_pipeline.load_json(dataset["panoptic"]["path"])
labels = [lb for i, lb in enumerate(labels) if i % 1500 < 90]
# Filter by maximum number of persons
@ -242,7 +228,7 @@ def load_labels(dataset: dict):
label["imgpaths"].pop(i)
elif "human36m" in dataset:
labels = load_json(dataset["human36m"]["path"])
labels = utils_pipeline.load_json(dataset["human36m"]["path"])
labels = [lb for lb in labels if lb["subject"] == "S9"]
labels = [lb for i, lb in enumerate(labels) if i % 4000 < 150]
@ -251,7 +237,7 @@ def load_labels(dataset: dict):
label.pop("frame")
elif "mvor" in dataset:
labels = load_json(dataset["mvor"]["path"])
labels = utils_pipeline.load_json(dataset["mvor"]["path"])
# Rename keys
for label in labels:
@ -259,20 +245,20 @@ def load_labels(dataset: dict):
label["imgpaths_color"] = label["imgpaths"]
elif "ikeaasm" in dataset:
labels = load_json(dataset["ikeaasm"]["path"])
labels = utils_pipeline.load_json(dataset["ikeaasm"]["path"])
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"])
labels = utils_pipeline.load_json(dataset["shelf"]["path"])
labels = [lb for lb in labels if "test" in lb["splits"]]
elif "campus" in dataset:
labels = load_json(dataset["campus"]["path"])
labels = utils_pipeline.load_json(dataset["campus"]["path"])
labels = [lb for lb in labels if "test" in lb["splits"]]
elif "tsinghua" in dataset:
labels = load_json(dataset["tsinghua"]["path"])
labels = utils_pipeline.load_json(dataset["tsinghua"]["path"])
labels = [lb for lb in labels if "test" in lb["splits"]]
labels = [lb for lb in labels if lb["seq"] == "seq_1"]
labels = [lb for i, lb in enumerate(labels) if i % 300 < 90]
@ -281,15 +267,15 @@ def load_labels(dataset: dict):
label["bodyids"] = list(range(len(label["bodies3D"])))
elif "chi3d" in dataset:
labels = load_json(dataset["chi3d"]["path"])
labels = utils_pipeline.load_json(dataset["chi3d"]["path"])
labels = [lb for lb in labels if lb["setup"] == "s03"]
labels = [lb for i, lb in enumerate(labels) if i % 2000 < 150]
elif "human36m_wb" in dataset:
labels = load_json(dataset["human36m_wb"]["path"])
labels = utils_pipeline.load_json(dataset["human36m_wb"]["path"])
elif any(("egohumans" in key for key in dataset)):
labels = load_json(dataset[dataset_use]["path"])
labels = utils_pipeline.load_json(dataset[dataset_use]["path"])
labels = [lb for lb in labels if "test" in lb["splits"]]
labels = [lb for lb in labels if dataset[dataset_use]["subset"] in lb["seq"]]
if dataset[dataset_use]["subset"] in ["volleyball", "tennis"]:
@ -359,7 +345,7 @@ def main():
"whole_body": whole_body,
"take_interval": datasets[dataset_use]["take_interval"],
}
save_json(config, config_path)
utils_pipeline.save_json(config, config_path)
# Call the CPP binary
os.system("/RapidPoseTriangulation/scripts/test_skelda_dataset.bin")
@ -367,7 +353,7 @@ def main():
# Load the results
print("Loading exports ...")
res_path = tmp_export_dir + "results.json"
results = load_json(res_path)
results = utils_pipeline.load_json(res_path)
all_poses_3d = results["all_poses_3d"]
all_poses_2d = results["all_poses_2d"]
all_ids = results["all_ids"]