From 9e99faea4678b047712a4ff0544dc71a86d5b4fe Mon Sep 17 00:00:00 2001 From: Victor Joos Date: Fri, 27 Jun 2025 14:36:34 +0200 Subject: [PATCH] add needed columns to empty detections --- tracklab/engine/offline.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tracklab/engine/offline.py b/tracklab/engine/offline.py index 6214adf2..ad00fea5 100644 --- a/tracklab/engine/offline.py +++ b/tracklab/engine/offline.py @@ -1,4 +1,5 @@ import logging +import pandas as pd from tracklab.engine import TrackingEngine from tracklab.utils.cv2 import cv2_load_image @@ -31,5 +32,6 @@ def video_loop(self, tracker_state, video, video_id): detections, image_pred = self.default_step(batch, model_name, detections, image_pred) self.callback("on_module_end", task=model_name, detections=detections) if detections.empty: + detections = pd.DataFrame(columns=["image_id", "video_id"]) return detections, image_pred return detections, image_pred