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

Some small updates.

Browse Source
This commit is contained in:
Daniel
2024-08-06 14:51:23 +02:00
parent 36c236b8d8
commit 396018195d
3 changed files with 520 additions and 514 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1024
media/RESULTS.md
View File

File diff suppressed because it is too large Load Diff

8
scripts/test_skelda_dataset.py
View File

@ -320,9 +320,15 @@ def main():
all_paths = [] all_paths = []
times = [] times = []
last_poses_3d = np.array([]) last_poses_3d = np.array([])
old_scene = ""
for label in tqdm.tqdm(labels): for label in tqdm.tqdm(labels):
images_2d = [] images_2d = []
if old_scene != label.get("scene", ""):
# Reset last poses if scene changes
old_scene = label.get("scene", "")
last_poses_3d = np.array([])
try: try:
start = time.time() start = time.time()
for i in range(len(label["imgpaths"])): for i in range(len(label["imgpaths"])):
@ -361,7 +367,7 @@ def main():
minscores = { minscores = {
# Choose this depending on the fraction of invalid/missing persons # Choose this depending on the fraction of invalid/missing persons
# A higher value reduces the number of proposals # A higher value reduces the number of proposals
"panoptic": 0.95, "panoptic": 0.94,
"human36m": 0.94, "human36m": 0.94,
"mvor": 0.86, "mvor": 0.86,
"campus": 0.96, "campus": 0.96,

2
scripts/triangulate_poses.py
View File

@ -222,7 +222,7 @@ def calc_grouping(all_pairs, min_score: float):
continue continue
# Check if the pair matches to an existing group # Check if the pair matches to an existing group
max_center_dist = 0.9 max_center_dist = 0.6
max_joint_avg_dist = 0.3 max_joint_avg_dist = 0.3
best_dist = math.inf best_dist = math.inf
best_group = -1 best_group = -1