From 2abb76e167a5becdcc6e968dd03684eaa992ea97 Mon Sep 17 00:00:00 2001 From: Brad Dwyer Date: Wed, 17 Jul 2024 17:15:51 -0700 Subject: [PATCH] Change CropAnnotator scale_factor from int to float --- supervision/annotators/core.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/supervision/annotators/core.py b/supervision/annotators/core.py index a3f65b77..6d34ae5c 100644 --- a/supervision/annotators/core.py +++ b/supervision/annotators/core.py @@ -2124,7 +2124,7 @@ class CropAnnotator(BaseAnnotator): def __init__( self, position: Position = Position.TOP_CENTER, - scale_factor: int = 2, + scale_factor: float = 2.0, border_color: Union[Color, ColorPalette] = ColorPalette.DEFAULT, border_thickness: int = 2, border_color_lookup: ColorLookup = ColorLookup.CLASS, @@ -2133,7 +2133,7 @@ class CropAnnotator(BaseAnnotator): Args: position (Position): The anchor position for placing the cropped and scaled part of the detection in the scene. - scale_factor (int): The factor by which to scale the cropped image part. A + scale_factor (float): The factor by which to scale the cropped image part. A factor of 2, for example, would double the size of the cropped area, allowing for a closer view of the detection. border_color (Union[Color, ColorPalette]): The color or color palette to @@ -2143,7 +2143,7 @@ class CropAnnotator(BaseAnnotator): annotations. Options are `INDEX`, `CLASS`, `TRACK`. """ self.position: Position = position - self.scale_factor: int = scale_factor + self.scale_factor: float = scale_factor self.border_color: Union[Color, ColorPalette] = border_color self.border_thickness: int = border_thickness self.border_color_lookup: ColorLookup = border_color_lookup