round scores

landing-ai · Apr 24, 2024 · 0fc5d14 · 0fc5d14
1 parent 669a88b
commit 0fc5d14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -916,7 +916,7 @@ def __call__(self, image: str) -> dict:
  ]
  box = normalize_bbox(box, image_size)
  output["bboxes"].append(box)
- output["scores"].append(det["score"])
+ output["scores"].append(round(det["score"], 2))
  return output