Added flag for minimal group size.

scripts/test_skelda_dataset.py

@@ -333,13 +333,22 @@ def main():
         "koarob": 0.91,
     }
     minscore = minscores.get(dataset_use, 0.95)
+    min_group_sizes = {
+        # If the number of cameras is high, and the views are not occluded, use a higher value
+        "panoptic": 1,
+        "shelf": 2,
+        "tsinghua": 2,
+    }
+    min_group_size = min_group_sizes.get(dataset_use, 1)
+    if dataset_use == "panoptic" and len(datasets["panoptic"]["cams"]) == 10:
+        min_group_size = 5
 
     print("\nRunning predictions ...")
     all_poses = []
     all_ids = []
     all_paths = []
     times = []
-    triangulator = spt.Triangulator(min_score=minscore)
+    triangulator = spt.Triangulator(min_score=minscore, min_group_size=min_group_size)
     old_scene = ""
     for label in tqdm.tqdm(labels):
         images_2d = []