^j4ddlmZddlmZddlmZmZddlZddlZ ddl mZ ddl m Z ddlmZddlmZddlmZmZmZdd lmZdd lmZGd d ZGd dZy)) annotations)Iterable)AnycastN) Detections)polygon_to_mask)Color)draw_filled_polygon draw_polygon draw_text)Position)get_polygon_centerc@eZdZdZej ff ddZddZy) PolygonZoneaV A class for defining a polygon-shaped zone within a frame for detecting objects. !!! warning PolygonZone uses the `tracker_id`. Read [here](/latest/trackers/) to learn how to plug tracking into your inference pipeline. Attributes: polygon: A polygon represented by a numpy array of shape `(N, 2)`, containing the `x`, `y` coordinates of the points. triggering_anchors: A list of positions specifying which anchors of the detections bounding box to consider when deciding on whether the detection fits within the PolygonZone (default: (sv.Position.BOTTOM_CENTER,)). current_count: The current count of detected objects within the zone mask: The 2D bool mask for the polygon zone Example: ```pycon >>> import numpy as np >>> import supervision as sv >>> polygon = np.array([[100, 200], [200, 100], [300, 200], [200, 300]]) >>> polygon_zone = sv.PolygonZone(polygon=polygon) >>> detections = sv.Detections( ... xyxy=np.array([[180, 100, 220, 200], [400, 400, 450, 500]]) ... ) >>> is_detections_in_zone = polygon_zone.trigger(detections) >>> is_detections_in_zone array([ True, False]) >>> polygon_zone.current_count 1 ``` c|jt|_||_t |js t dd|_tj|d\}}t||dz|dzf|_ y)Nz#Triggering anchors cannot be empty.raxis)polygon resolution_wh) astypeintrtriggering_anchorslist ValueError current_countnpmaxrmask)selfrrx_maxy_maxs s/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/supervision/detection/tools/polygon_zone.py__init__zPolygonZone.__init__8sp ~~c* "4D++,BC CvvgA. u#EAIuqy+A  c t|dk(r"d|_tjgtStj|j Dcgc]9}tj |j|jt;c}}|jj\}}|dddddf|dddddf}}|dk\|dk\z||kz||kz}tj|d|dz } tj|d|dz } tj||j| | fzd} ttj| |_| jtScc}w)a Determines if the detections are within the polygon zone. Anchor points are calculated from original (unclipped) detection boxes to avoid per-zone clipping shifting anchor positions. This prevents a single detection from being counted in multiple non-overlapping zones due to clipping artifacts, although overlapping zones may still legitimately contain the same detection. Args: detections: The detections to be checked against the polygon zone Returns: A boolean numpy array indicating if each detection is within the polygon zone r)dtypeNr)lenrrarrayboolrceilget_anchors_coordinatesrrrshapeclipallsum) r detectionsanchors all_anchorsmask_hmask_wxy in_boundsx_safey_safe is_in_zones r#triggerzPolygonZone.triggerIsD" z?a !"D 88Bd+ +hh $66  ::7CDKKCP  1a7#[Aq%91!VQ'1v:6!f*E Avz*Avz*VVI &&.(AAJ   !34  && s>E0N)rznpt.NDArray[np.int64]rzIterable[Position])r2rreturnznpt.NDArray[np.bool_])__name__ __module__ __qualname____doc__r BOTTOM_CENTERr$r=r%r#rrs2#P3;2H2H1J & / "#'r%rceZdZdZej dej dddddf d dZ d d d Zy )PolygonZoneAnnotatora A class for annotating a polygon-shaped zone within a frame with a count of detected objects. Attributes: zone: The polygon zone to be annotated color: The color to draw the polygon lines, default is white thickness: The thickness of the polygon lines, default is 2 text_color: The color of the text on the polygon, default is black text_scale: The scale of the text on the polygon, default is 0.5 text_thickness: The thickness of the text on the polygon, default is 1 text_padding: The padding around the text on the polygon, default is 10 font: The font type for the text on the polygon, default is cv2.FONT_HERSHEY_SIMPLEX center: The center of the polygon for text placement display_in_zone_count: Show the label of the zone or not. Default is True opacity: The opacity of zone filling when drawn on the scene. Default is 0 rg?r( Trc ||_||_||_||_||_||_||_tj|_ t|j|_ ||_ | |_y)N)r)zonecolor thickness text_color text_scaletext_thickness text_paddingcv2FONT_HERSHEY_SIMPLEXfontrrcenterdisplay_in_zone_countopacity) r rIrJrKrLrMrNrOrTrUs r#r$zPolygonZoneAnnotator.__init__sc  "$$,(,, (> %:" r%Nc |jdk(r8t||jj|j|j }n|t |j|jj|j|j}t||jj|j|j }|jr{t||t|jjn||j|j|j|j|j|j |j" }t%t&j(t*|S)a Annotates the polygon zone within a frame with a count of detected objects. Args: scene: The image on which the polygon zone will be annotated label: A label for the count of detected objects within the polygon zone (default: None) Returns: The image with the polygon zone and count of detected objects r)scenerrJrK)rWrrJrU) rWtext text_anchorbackground_colorrLrMrNrO text_font)rUr rIrrJrKr copyrTr strrrSrLrMrNrOrRrnptNDArrayr)r rWlabelannotated_frames r#annotatezPolygonZoneAnnotator.annotates <<1 * ))jj.. O2jjl ))jj O +% ))jj.. O  % %'%5:]S001 KK!%????#22!..)) OCKK$o66r%)rIrrJr rKrrLr rMfloatrNrrOrrTr+rUrc)N)rWnpt.NDArray[Any]r`z str | Noner>rd) r?r@rArBr WHITEBLACKr$rbrDr%r#rFrFos,{{!KK&*      $2<@07%07.807 07r%rF) __future__rcollections.abcrtypingrrrPnumpyr numpy.typingr^ supervisionr&supervision.detection.utils.convertersrsupervision.draw.colorr supervision.draw.utilsr r r supervision.geometry.corer supervision.geometry.utilsrrrFrDr%r#rrsE"$ "B(OO.9Z'Z'z\7\7r%