From 63c8427c5bbf683c0efcd9f3ea860c2803011987 Mon Sep 17 00:00:00 2001 From: Victor Joos Date: Fri, 27 Jun 2025 10:41:12 +0200 Subject: [PATCH 1/2] Always open a video file from zero and seek to correct position --- tracklab/utils/cv2.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/tracklab/utils/cv2.py b/tracklab/utils/cv2.py index 0c198d90..144f7485 100644 --- a/tracklab/utils/cv2.py +++ b/tracklab/utils/cv2.py @@ -34,14 +34,10 @@ class VideoReader: def __init__(self): self.filename = None - self.cap = None # cv2.VideoCapture(filename) def set_filename(self, filename): + # TODO: remove this function, and set self.filename directly self.filename = filename - if self.cap is not None: - self.cap.release() - self.cap = cv2.VideoCapture(self.filename) - assert self.cap.isOpened(), "Error opening video stream or file" def __getitem__(self, idx): assert self.filename is not None, "You should first set the filename" From 393ffb8dcb1badbf5029b74c0aae7f29916079c6 Mon Sep 17 00:00:00 2001 From: Victor Joos Date: Fri, 27 Jun 2025 10:54:37 +0200 Subject: [PATCH 2/2] Bigger factor between image_ids of diff videos --- tracklab/wrappers/dataset/external_video.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tracklab/wrappers/dataset/external_video.py b/tracklab/wrappers/dataset/external_video.py index c82a6749..d3083bdc 100644 --- a/tracklab/wrappers/dataset/external_video.py +++ b/tracklab/wrappers/dataset/external_video.py @@ -73,7 +73,7 @@ def __init__(self, dataset_path: str, video_path: str, *args, **kwargs): image_metadata.extend( [ { - "id": j+1000*i, + "id": j+100_000*i, # TODO: read the number of frames to get the right factor "name": f"{video_name}_{j}", "frame": j, "nframes": nframes,