Add none check

TrackingLaboratory · Feb 8, 2024 · a0af02d · a0af02d
1 parent c4f2a9e
commit a0af02d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tracklab/wrappers/eval/trackeval_evaluator.py b/tracklab/wrappers/eval/trackeval_evaluator.py
@@ -47,7 +47,7 @@ def run(self, tracker_state):
         log.info(
             f"Tracking predictions saved in {self.trackeval_dataset_name} format in {pred_save_path}")
 
-        if len(tracker_state.detections_gt) == 0:
+        if tracker_state.detections_gt is None or len(tracker_state.detections_gt) == 0:
             log.info(
                 f"Stopping evaluation because the current split ({self.eval_set}) has no ground truth detections.")
             return