From b92ffeff82c260adbd55ea6b449275fcd7c80341 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 16 Nov 2023 14:24:18 +0000 Subject: [PATCH] =?UTF-8?q?fix(pre=5Fcommit):=20=F0=9F=8E=A8=20auto=20form?= =?UTF-8?q?at=20pre-commit=20hooks?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- supervision/annotators/core.py | 235 +++++++++++++++++---------------- 1 file changed, 118 insertions(+), 117 deletions(-) diff --git a/supervision/annotators/core.py b/supervision/annotators/core.py index cc8bfe4c..f1ec48ea 100644 --- a/supervision/annotators/core.py +++ b/supervision/annotators/core.py @@ -19,10 +19,10 @@ class BoundingBoxAnnotator(BaseAnnotator): """ def __init__( - self, - color: Union[Color, ColorPalette] = ColorPalette.default(), - thickness: int = 2, - color_lookup: ColorLookup = ColorLookup.CLASS, + self, + color: Union[Color, ColorPalette] = ColorPalette.default(), + thickness: int = 2, + color_lookup: ColorLookup = ColorLookup.CLASS, ): """ Args: @@ -37,10 +37,10 @@ class BoundingBoxAnnotator(BaseAnnotator): self.color_lookup: ColorLookup = color_lookup def annotate( - self, - scene: np.ndarray, - detections: Detections, - custom_color_lookup: Optional[np.ndarray] = None, + self, + scene: np.ndarray, + detections: Detections, + custom_color_lookup: Optional[np.ndarray] = None, ) -> np.ndarray: """ Annotates the given scene with bounding boxes based on the provided detections. @@ -101,10 +101,10 @@ class MaskAnnotator(BaseAnnotator): """ def __init__( - self, - color: Union[Color, ColorPalette] = ColorPalette.default(), - opacity: float = 0.5, - color_lookup: ColorLookup = ColorLookup.CLASS, + self, + color: Union[Color, ColorPalette] = ColorPalette.default(), + opacity: float = 0.5, + color_lookup: ColorLookup = ColorLookup.CLASS, ): """ Args: @@ -119,10 +119,10 @@ class MaskAnnotator(BaseAnnotator): self.color_lookup: ColorLookup = color_lookup def annotate( - self, - scene: np.ndarray, - detections: Detections, - custom_color_lookup: Optional[np.ndarray] = None, + self, + scene: np.ndarray, + detections: Detections, + custom_color_lookup: Optional[np.ndarray] = None, ) -> np.ndarray: """ Annotates the given scene with masks based on the provided detections. @@ -185,10 +185,10 @@ class PolygonAnnotator(BaseAnnotator): """ def __init__( - self, - color: Union[Color, ColorPalette] = ColorPalette.default(), - thickness: int = 2, - color_lookup: ColorLookup = ColorLookup.CLASS, + self, + color: Union[Color, ColorPalette] = ColorPalette.default(), + thickness: int = 2, + color_lookup: ColorLookup = ColorLookup.CLASS, ): """ Args: @@ -203,10 +203,10 @@ class PolygonAnnotator(BaseAnnotator): self.color_lookup: ColorLookup = color_lookup def annotate( - self, - scene: np.ndarray, - detections: Detections, - custom_color_lookup: Optional[np.ndarray] = None, + self, + scene: np.ndarray, + detections: Detections, + custom_color_lookup: Optional[np.ndarray] = None, ) -> np.ndarray: """ Annotates the given scene with polygons based on the provided detections. @@ -267,10 +267,10 @@ class BoxMaskAnnotator(BaseAnnotator): """ def __init__( - self, - color: Union[Color, ColorPalette] = ColorPalette.default(), - opacity: float = 0.5, - color_lookup: ColorLookup = ColorLookup.CLASS, + self, + color: Union[Color, ColorPalette] = ColorPalette.default(), + opacity: float = 0.5, + color_lookup: ColorLookup = ColorLookup.CLASS, ): """ Args: @@ -285,10 +285,10 @@ class BoxMaskAnnotator(BaseAnnotator): self.opacity = opacity def annotate( - self, - scene: np.ndarray, - detections: Detections, - custom_color_lookup: Optional[np.ndarray] = None, + self, + scene: np.ndarray, + detections: Detections, + custom_color_lookup: Optional[np.ndarray] = None, ) -> np.ndarray: """ Annotates the given scene with box masks based on the provided detections. @@ -353,11 +353,11 @@ class HaloAnnotator(BaseAnnotator): """ def __init__( - self, - color: Union[Color, ColorPalette] = ColorPalette.default(), - opacity: float = 0.8, - kernel_size: int = 40, - color_lookup: ColorLookup = ColorLookup.CLASS, + self, + color: Union[Color, ColorPalette] = ColorPalette.default(), + opacity: float = 0.8, + kernel_size: int = 40, + color_lookup: ColorLookup = ColorLookup.CLASS, ): """ Args: @@ -375,10 +375,10 @@ class HaloAnnotator(BaseAnnotator): self.kernel_size: int = kernel_size def annotate( - self, - scene: np.ndarray, - detections: Detections, - custom_color_lookup: Optional[np.ndarray] = None, + self, + scene: np.ndarray, + detections: Detections, + custom_color_lookup: Optional[np.ndarray] = None, ) -> np.ndarray: """ Annotates the given scene with halos based on the provided detections. @@ -445,12 +445,12 @@ class EllipseAnnotator(BaseAnnotator): """ def __init__( - self, - color: Union[Color, ColorPalette] = ColorPalette.default(), - thickness: int = 2, - start_angle: int = -45, - end_angle: int = 235, - color_lookup: ColorLookup = ColorLookup.CLASS, + self, + color: Union[Color, ColorPalette] = ColorPalette.default(), + thickness: int = 2, + start_angle: int = -45, + end_angle: int = 235, + color_lookup: ColorLookup = ColorLookup.CLASS, ): """ Args: @@ -469,10 +469,10 @@ class EllipseAnnotator(BaseAnnotator): self.color_lookup: ColorLookup = color_lookup def annotate( - self, - scene: np.ndarray, - detections: Detections, - custom_color_lookup: Optional[np.ndarray] = None, + self, + scene: np.ndarray, + detections: Detections, + custom_color_lookup: Optional[np.ndarray] = None, ) -> np.ndarray: """ Annotates the given scene with ellipses based on the provided detections. @@ -535,11 +535,11 @@ class BoxCornerAnnotator(BaseAnnotator): """ def __init__( - self, - color: Union[Color, ColorPalette] = ColorPalette.default(), - thickness: int = 4, - corner_length: int = 15, - color_lookup: ColorLookup = ColorLookup.CLASS, + self, + color: Union[Color, ColorPalette] = ColorPalette.default(), + thickness: int = 4, + corner_length: int = 15, + color_lookup: ColorLookup = ColorLookup.CLASS, ): """ Args: @@ -556,10 +556,10 @@ class BoxCornerAnnotator(BaseAnnotator): self.color_lookup: ColorLookup = color_lookup def annotate( - self, - scene: np.ndarray, - detections: Detections, - custom_color_lookup: Optional[np.ndarray] = None, + self, + scene: np.ndarray, + detections: Detections, + custom_color_lookup: Optional[np.ndarray] = None, ) -> np.ndarray: """ Annotates the given scene with box corners based on the provided detections. @@ -621,10 +621,10 @@ class CircleAnnotator(BaseAnnotator): """ def __init__( - self, - color: Union[Color, ColorPalette] = ColorPalette.default(), - thickness: int = 2, - color_lookup: ColorLookup = ColorLookup.CLASS, + self, + color: Union[Color, ColorPalette] = ColorPalette.default(), + thickness: int = 2, + color_lookup: ColorLookup = ColorLookup.CLASS, ): """ Args: @@ -640,10 +640,10 @@ class CircleAnnotator(BaseAnnotator): self.color_lookup: ColorLookup = color_lookup def annotate( - self, - scene: np.ndarray, - detections: Detections, - custom_color_lookup: Optional[np.ndarray] = None, + self, + scene: np.ndarray, + detections: Detections, + custom_color_lookup: Optional[np.ndarray] = None, ) -> np.ndarray: """ Annotates the given scene with circles based on the provided detections. @@ -705,11 +705,11 @@ class DotAnnotator(BaseAnnotator): """ def __init__( - self, - color: Union[Color, ColorPalette] = ColorPalette.default(), - radius: int = 4, - position: Position = Position.CENTER, - color_lookup: ColorLookup = ColorLookup.CLASS, + self, + color: Union[Color, ColorPalette] = ColorPalette.default(), + radius: int = 4, + position: Position = Position.CENTER, + color_lookup: ColorLookup = ColorLookup.CLASS, ): """ Args: @@ -726,10 +726,10 @@ class DotAnnotator(BaseAnnotator): self.color_lookup: ColorLookup = color_lookup def annotate( - self, - scene: np.ndarray, - detections: Detections, - custom_color_lookup: Optional[np.ndarray] = None, + self, + scene: np.ndarray, + detections: Detections, + custom_color_lookup: Optional[np.ndarray] = None, ) -> np.ndarray: """ Annotates the given scene with dots based on the provided detections. @@ -781,14 +781,14 @@ class LabelAnnotator: """ def __init__( - self, - color: Union[Color, ColorPalette] = ColorPalette.default(), - text_color: Color = Color.black(), - text_scale: float = 0.5, - text_thickness: int = 1, - text_padding: int = 10, - text_position: Position = Position.TOP_LEFT, - color_lookup: ColorLookup = ColorLookup.CLASS, + self, + color: Union[Color, ColorPalette] = ColorPalette.default(), + text_color: Color = Color.black(), + text_scale: float = 0.5, + text_thickness: int = 1, + text_padding: int = 10, + text_position: Position = Position.TOP_LEFT, + color_lookup: ColorLookup = ColorLookup.CLASS, ): """ Args: @@ -813,9 +813,9 @@ class LabelAnnotator: @staticmethod def resolve_text_background_xyxy( - center_coordinates: Tuple[int, int], - text_wh: Tuple[int, int], - position: Position, + center_coordinates: Tuple[int, int], + text_wh: Tuple[int, int], + position: Position, ) -> Tuple[int, int, int, int]: center_x, center_y = center_coordinates text_w, text_h = text_wh @@ -829,14 +829,14 @@ class LabelAnnotator: center_x - text_w // 2, center_y - text_h, center_x + text_w // 2, - center_y + center_y, ) elif position == Position.CENTER or position == Position.CENTER_OF_MASS: return ( center_x - text_w // 2, center_y - text_h // 2, center_x + text_w // 2, - center_y + text_h // 2 + center_y + text_h // 2, ) elif position == Position.BOTTOM_LEFT: return center_x, center_y, center_x + text_w, center_y + text_h @@ -847,15 +847,15 @@ class LabelAnnotator: center_x - text_w // 2, center_y, center_x + text_w // 2, - center_y + text_h + center_y + text_h, ) def annotate( - self, - scene: np.ndarray, - detections: Detections, - labels: List[str] = None, - custom_color_lookup: Optional[np.ndarray] = None, + self, + scene: np.ndarray, + detections: Detections, + labels: List[str] = None, + custom_color_lookup: Optional[np.ndarray] = None, ) -> np.ndarray: """ Annotates the given scene with labels based on the provided detections. @@ -889,7 +889,8 @@ class LabelAnnotator: """ font = cv2.FONT_HERSHEY_SIMPLEX anchors_coordinates = detections.get_anchors_coordinates( - anchor=self.text_anchor) + anchor=self.text_anchor + ) for center_coordinates, detection_idx in enumerate(anchors_coordinates): color = resolve_color( color=self.color, @@ -955,9 +956,9 @@ class BlurAnnotator(BaseAnnotator): self.kernel_size: int = kernel_size def annotate( - self, - scene: np.ndarray, - detections: Detections, + self, + scene: np.ndarray, + detections: Detections, ) -> np.ndarray: """ Annotates the given scene by blurring regions based on the provided detections. @@ -1011,12 +1012,12 @@ class TraceAnnotator: """ def __init__( - self, - color: Union[Color, ColorPalette] = ColorPalette.default(), - position: Position = Position.CENTER, - trace_length: int = 30, - thickness: int = 2, - color_lookup: ColorLookup = ColorLookup.CLASS, + self, + color: Union[Color, ColorPalette] = ColorPalette.default(), + position: Position = Position.CENTER, + trace_length: int = 30, + thickness: int = 2, + color_lookup: ColorLookup = ColorLookup.CLASS, ): """ Args: @@ -1037,10 +1038,10 @@ class TraceAnnotator: self.color_lookup: ColorLookup = color_lookup def annotate( - self, - scene: np.ndarray, - detections: Detections, - custom_color_lookup: Optional[np.ndarray] = None, + self, + scene: np.ndarray, + detections: Detections, + custom_color_lookup: Optional[np.ndarray] = None, ) -> np.ndarray: """ Draws trace paths on the frame based on the detection coordinates provided. @@ -1114,13 +1115,13 @@ class HeatMapAnnotator: """ def __init__( - self, - position: Position = Position.BOTTOM_CENTER, - opacity: float = 0.2, - radius: int = 40, - kernel_size: int = 25, - top_hue: int = 0, - low_hue: int = 125, + self, + position: Position = Position.BOTTOM_CENTER, + opacity: float = 0.2, + radius: int = 40, + kernel_size: int = 25, + top_hue: int = 0, + low_hue: int = 125, ): """ Args: