Some small updates.

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

View File

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