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

Mixed updates and improvements.

Browse Source
This commit is contained in:
Daniel
2025-01-21 16:23:19 +01:00
parent c5f190ab35
commit 29c072400f
11 changed files with 49 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/test_skelda_dataset_py.py
View File

@ -330,7 +330,9 @@ def main():
# Load 2D pose model
if utils_pipeline.use_whole_body(whole_body):
kpt_model = utils_2d_pose.load_wb_model(min_bbox_score, min_bbox_area, batch_poses)
kpt_model = utils_2d_pose.load_wb_model(
min_bbox_score, min_bbox_area, batch_poses
)
else:
kpt_model = utils_2d_pose.load_model(min_bbox_score, min_bbox_area, batch_poses)
@ -446,10 +448,7 @@ def main():
all_ids.append(label["id"])
times[i][2] = time_3d
# Print per-step triangulation timings
print("")
triangulator.print_stats()
# Print per-step timings
warmup_iters = 10
if len(times) > warmup_iters:
times = times[warmup_iters:]
@ -464,6 +463,7 @@ def main():
}
print("\nMetrics:")
print(json.dumps(tstats, indent=2))
triangulator.print_stats()
_ = evals.mpjpe.run_eval(
labels,