diff --git a/supervision/annotators/core.py b/supervision/annotators/core.py index 52b89833..b5395737 100644 --- a/supervision/annotators/core.py +++ b/supervision/annotators/core.py @@ -9,10 +9,11 @@ from PIL import Image, ImageDraw, ImageFont from supervision.annotators.base import BaseAnnotator, ImageType from supervision.annotators.utils import ( + PENDING_TRACK_ID, ColorLookup, Trace, resolve_color, - resolve_text_background_xyxy, PENDING_TRACK_ID, + resolve_text_background_xyxy, ) from supervision.config import CLASS_NAME_DATA_FIELD, ORIENTED_BOX_COORDINATES from supervision.detection.core import Detections diff --git a/supervision/annotators/utils.py b/supervision/annotators/utils.py index 501f6535..74d25a65 100644 --- a/supervision/annotators/utils.py +++ b/supervision/annotators/utils.py @@ -10,6 +10,7 @@ from supervision.geometry.core import Position PENDING_TRACK_COLOR = Color.GREY PENDING_TRACK_ID = -1 + class ColorLookup(Enum): """ Enumeration class to define strategies for mapping colors to annotations. diff --git a/supervision/config.py b/supervision/config.py index 050ad6e6..b18d2e20 100644 --- a/supervision/config.py +++ b/supervision/config.py @@ -1,2 +1,2 @@ CLASS_NAME_DATA_FIELD = "class_name" -ORIENTED_BOX_COORDINATES = "xyxyxyxy" \ No newline at end of file +ORIENTED_BOX_COORDINATES = "xyxyxyxy"