finish logic, add imports

roboflow · Dec 4, 2023 · 21bf374 · 21bf374
1 parent b4fca50
commit 21bf374
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/supervision/__init__.py b/supervision/__init__.py
@@ -20,6 +20,7 @@
     MaskAnnotator,
     PolygonAnnotator,
     TraceAnnotator,
+    ColorAnnotator
 )
 from supervision.annotators.utils import ColorLookup
 from supervision.classification.core import Classifications

diff --git a/supervision/annotators/core.py b/supervision/annotators/core.py
@@ -1205,7 +1205,7 @@ class ColorAnnotator(BaseAnnotator):
     using provided detections.
     """
 
-    def __init__(self, color: Color = Color[255, 255, 255]):
+    def __init__(self, color: Color = Color(r=0, g=0, b=0)):
         """
         Args:
             color (Color): The color to replace the regions with.