From c32bc6538d32f31de11b221cdc8367ce1fe67e27 Mon Sep 17 00:00:00 2001 From: Camilo Zapata Date: Wed, 2 Oct 2024 10:29:49 -0500 Subject: [PATCH] fix: labels should start at 1 instead of 0 (#254) labels should start at 1 instead of 0 --- vision_agent/tools/tools.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py index 95fdd56c..f2224de5 100644 --- a/vision_agent/tools/tools.py +++ b/vision_agent/tools/tools.py @@ -1603,7 +1603,7 @@ def extract_frames_and_timestamps( """ def reformat( - frames_and_timestamps: List[Tuple[np.ndarray, float]] + frames_and_timestamps: List[Tuple[np.ndarray, float]], ) -> List[Dict[str, Union[np.ndarray, float]]]: return [ {"frame": frame, "timestamp": timestamp} @@ -2017,7 +2017,7 @@ def overlay_counting_results( fontsize, ) - for i, elt in enumerate(instances): + for i, elt in enumerate(instances, 1): label = f"{i}" box = elt["bbox"]