fix: labels should start at 1 instead of 0 (#254)

labels should start at 1 instead of 0
landing-ai · Oct 2, 2024 · c32bc65 · c32bc65
1 parent d6705bb
commit c32bc65
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"]