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

Eval with all panoptic cameras.

Browse Source
This commit is contained in:
Daniel
2025-01-28 15:30:23 +01:00
parent 321aa0a714
commit b41380b033
2 changed files with 290 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/test_skelda_dataset.py
View File

@ -66,11 +66,13 @@ datasets = {
"cams": ["00_03", "00_06", "00_12", "00_13", "00_23"],
# "cams": ["00_03", "00_06", "00_12"],
# "cams": ["00_03", "00_06", "00_12", "00_13", "00_23", "00_15", "00_10", "00_21", "00_09", "00_01"],
# "cams": [],
"take_interval": 3,
"min_match_score": 0.95,
"use_scenes": ["160906_pizza1", "160422_haggling1", "160906_ian5"],
"min_group_size": 1,
# "min_group_size": 4,
# "min_group_size": 11,
"min_bbox_area": 0.05 * 0.05,
},
"mvor": {
@ -220,7 +222,7 @@ def load_labels(dataset: dict):
]
# Filter cameras
if not "cameras_depth" in labels[0]:
if not "cameras_depth" in labels[0] and len(dataset["panoptic"]["cams"]) > 0:
for label in labels:
for i, cam in reversed(list(enumerate(label["cameras"]))):
if cam["name"] not in dataset["panoptic"]["cams"]: