xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

🖤 Make black happy

This commit is contained in:
SkalskiP 2023-04-10 23:16:04 +02:00
parent c1ccff474f
commit ddc8a8ccce
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
supervision/detection/annotate.py
View File

@ -121,11 +121,12 @@ class BoxAnnotator:
class MaskAnnotator:
"""
A class for overlaying masks on an image using detections provided.
A class for overlaying masks on an image using detections provided.
Attributes:
color (Union[Color, ColorPalette]): The color to fill the mask, can be a single color or a color palette
"""
Attributes:
color (Union[Color, ColorPalette]): The color to fill the mask, can be a single color or a color palette
"""
def __init__(
self,
color: Union[Color, ColorPalette] = ColorPalette.default(),

2
supervision/detection/tools/polygon_zone.py
View File

@ -23,6 +23,7 @@ class PolygonZone:
current_count (int): The current count of detected objects within the zone
mask (np.ndarray): The 2D bool mask for the polygon zone
"""
def __init__(
self,
polygon: np.ndarray,
@ -77,6 +78,7 @@ class PolygonZoneAnnotator:
font (int): The font type for the text on the polygon, default is cv2.FONT_HERSHEY_SIMPLEX
center (Tuple[int, int]): The center of the polygon for text placement
"""
def __init__(
self,
zone: PolygonZone,