o ec@sddlmZddlmZmZddlZddlZddlm Z ddl m Z m Z ddl mZddlmZmZddlmZdd lmZGd d d ZGd d d ZdS))replace)OptionalTupleN) Detections) clip_boxespolygon_to_mask)Color) draw_polygon draw_text)Position)get_polygon_centerc@sJeZdZdZejfdejdee e fdefddZ de dejfd d Z d S) PolygonZonea| A class for defining a polygon-shaped zone within a frame for detecting objects. Attributes: polygon (np.ndarray): A polygon represented by a numpy array of shape `(N, 2)`, containing the `x`, `y` coordinates of the points. frame_resolution_wh (Tuple[int, int]): The frame resolution (width, height) triggering_position (Position): The position within the bounding box that triggers the zone (default: Position.BOTTOM_CENTER) current_count (int): The current count of detected objects within the zone mask (np.ndarray): The 2D bool mask for the polygon zone polygonframe_resolution_whtriggering_positioncCsD|t|_||_||_d|_|\}}t||d|dfd|_dS)Nr)r resolution_wh)astypeintrrr current_countrmask)selfrrrwidthheightrW/var/www/myenv/lib/python3.10/site-packages/supervision/detection/tools/polygon_zone.py__init__s  zPolygonZone.__init__ detectionsreturncCsvt|j|jd}t||d}t|j|jdt }|j |dddf|dddff}t t ||_ |t S)aS Determines if the detections are within the polygon zone. Parameters: detections (Detections): The detections to be checked against the polygon zone Returns: np.ndarray: A boolean numpy array indicating if each detection is within the polygon zone )xyxyr)r)anchorNrr)rrrrnpceilget_anchors_coordinatesrrrrsumrbool)rr clipped_xyxyclipped_detectionsclipped_anchors is_in_zonerrrtrigger-s   & zPolygonZone.triggerN) __name__ __module__ __qualname____doc__r BOTTOM_CENTERr!ndarrayrrrrr*rrrrr s  r c@sfeZdZdZdedddfdededed ed ed ed efd dZ dde j de e de j fddZdS)PolygonZoneAnnotatoraO A class for annotating a polygon-shaped zone within a frame with a count of detected objects. Attributes: zone (PolygonZone): The polygon zone to be annotated color (Color): The color to draw the polygon lines thickness (int): The thickness of the polygon lines, default is 2 text_color (Color): The color of the text on the polygon, default is black text_scale (float): The scale of the text on the polygon, default is 0.5 text_thickness (int): The thickness of the text on the polygon, default is 1 text_padding (int): The padding around the text on the polygon, default is 10 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 g?r zonecolor thickness text_color text_scaletext_thickness text_paddingcCsD||_||_||_||_||_||_||_tj|_ t |j d|_ dS)N)r) r4r5r6r7r8r9r:cv2FONT_HERSHEY_SIMPLEXfontr rcenter)rr4r5r6r7r8r9r:rrrrXs zPolygonZoneAnnotator.__init__Nscenelabelrc CsXt||jj|j|jd}t||durt|jjn||j|j|j |j |j |j |j d }|S)a Annotates the polygon zone within a frame with a count of detected objects. Parameters: scene (np.ndarray): The image on which the polygon zone will be annotated label (Optional[str]): An optional label for the count of detected objects within the polygon zone (default: None) Returns: np.ndarray: The image with the polygon zone and count of detected objects )r?rr5r6N) r?text text_anchorbackground_colorr7r8r9r: text_font)r r4rr5r6r strrr>r7r8r9r:r=)rr?r@annotated_framerrrannotatels$  zPolygonZoneAnnotator.annotate)N)r+r,r-r.rblackr rfloatrr!r0rrErGrrrrr1Fs. $r1) dataclassesrtypingrrr;numpyr! supervisionrsupervision.detection.utilsrrsupervision.draw.colorrsupervision.draw.utilsr r supervision.geometry.corer supervision.geometry.utilsr r r1rrrrs     7