From cda3eb6eea972dcde7dcd8b7c79e7b75870f3ee5 Mon Sep 17 00:00:00 2001 From: Camilo Zapata Date: Tue, 1 Oct 2024 19:17:17 -0500 Subject: [PATCH] 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"]