fix color issue

landing-ai · Apr 18, 2024 · 81f3cf3 · 81f3cf3
1 parent d1d3268
commit 81f3cf3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vision_agent/image_utils.py b/vision_agent/image_utils.py
@@ -103,7 +103,7 @@ def overlay_bboxes(
  elif isinstance(image, np.ndarray):
  image = Image.fromarray(image)
 
- color = {label: COLORS[i % len(COLORS)] for i, label in enumerate(bboxes["labels"])}
+ color = {label: COLORS[i % len(COLORS)] for i, label in enumerate(set(bboxes["labels"]))}
 
  width, height = image.size
  fontsize = max(12, int(min(width, height) / 40))