.hZddlmZddlmZddlZddlmZmZm Z ddl m Z GddeZ y)) annotations)AnyN) BaseSolutionSolutionAnnotatorSolutionResults)colorscNeZdZdZdfd Z ddZdZddZxZS) RegionCountera A class for real-time counting of objects within user-defined regions in a video stream. This class inherits from `BaseSolution` and provides functionality to define polygonal regions in a video frame, track objects, and count those objects that pass through each defined region. Useful for applications requiring counting in specified areas, such as monitoring zones or segmented sections. Attributes: region_template (dict): Template for creating new counting regions with default attributes including name, polygon coordinates, and display colors. counting_regions (list): List storing all defined regions, where each entry is based on `region_template` and includes specific region settings like name, coordinates, and color. region_counts (dict): Dictionary storing the count of objects for each named region. Methods: add_region: Add a new counting region with specified attributes. process: Process video frames to count objects in each region. initialize_regions: Initialize zones to count the objects in each one. Zones could be multiple as well. Examples: Initialize a RegionCounter and add a counting region >>> counter = RegionCounter() >>> counter.add_region("Zone1", [(100, 100), (200, 100), (200, 200), (100, 200)], (255, 0, 0), (255, 255, 255)) >>> results = counter.process(frame) >>> print(f"Total tracks: {results.total_tracks}") c zt|di|dddddd|_i|_g|_|j y)zSInitialize the RegionCounter for real-time object counting in user-defined regions.zDefault RegionNrr r )rrr)namepolygoncounts region_color text_color)super__init__region_template region_countscounting_regionsinitialize_regions)selfkwargs __class__s b/var/www/html/ai-service/venv/lib/python3.12/site-packages/ultralytics/solutions/region_counter.pyrzRegionCounter.__init__)sJ "6"$+#    " !c|jj}|j||j|||d|jj ||S)aM Add a new region to the counting list based on the provided template with specific attributes. Args: name (str): Name assigned to the new region. polygon_points (list[tuple]): List of (x, y) coordinates defining the region's polygon. region_color (tuple[int, int, int]): BGR color for region visualization. text_color (tuple[int, int, int]): BGR color for the text within the region. Returns: (dict[str, any]): Returns a dictionary including the region information i.e. name, region_color etc. )rrrr)rcopyupdatePolygonrappend)rrpolygon_pointsrrregions r add_regionzRegionCounter.add_region7sY&%%**, <<7 ,(    $$V, rc \|j|jt|jtsd|ji|_t |jj D]=\}\}}|j ||t|dd}|j|d|d<?y)zInitialize regions only once.Nz Region#01Tr rprepared_polygon) r%initialize_region isinstancedict enumerateitemsr&rprep)rirptsr%s rrz RegionCounter.initialize_regionsVs ;;   " " $$++t,& 4DK' (9(9(;< FNA{c__T3q$QF)-6)3D)EF% & Frc |j|t||j}t|j|j |j |jD]\}}}}|j||j|||t|d|j|d|dzdz |d|dzdz f}|jD]9}|dj|s|d xxdz cc<|d |j|d <;|jD]}|d } tt!t"t%j&| j(j*t$j, } t/| j0j2t/| j0j4fgdz\\} } \} }|j7| |d |jdz|j9| | | |gt;|d |d |d|jdzdd|d <|j=}|j?|tA|tC|j |jS)a Process the input frame to detect and count objects within each defined region. Args: im0 (np.ndarray): Input image frame where objects and regions are annotated. Returns: (SolutionResults): Contains processed image `plot_im`, 'total_tracks' (int, total number of tracked objects), and 'region_counts' (dict, counts of objects per region). ) line_widthT)labelcolorrr(rrr)dtyperrrect)r3r4 txt_colormarginshape)plot_im total_tracksr)"extract_tracksrr2zipboxesclss track_idsconfs box_labeladjust_box_labelrPointrcontainsrlistmaptuplenparrayexteriorcoordsint32intcentroidxy draw_regionadaptive_labelstrresultdisplay_outputrlen)rim0 annotatorboxclstrack_idconfcenterr%polyr0x1y1x2y2r>s rprocesszRegionCounter.process`sK C %cdooF (+DJJ 4>>SWS]S](^ J $Ch   4+@+@dH+U]cdlnr]s  tZZ#a&3q6/Q!6Q#a&A8M NOF// J,-66v>8$)$9?9ID&&vf~6 J J++ !F)$Ds5"((4==+?+?rxx"PQRC#&t}}#7T]]__9M"N!ORS!S HRhr2  ! !#vn'=tQR?R S  $ $RR &*+^, .* %  !F8  !""$ G$wS=P`d`r`rssr)rrreturnNone) rrXr$z list[tuple]rtuple[int, int, int]rrkrizdict[str, Any])r\z np.ndarrayrir) __name__ __module__ __qualname____doc__rr&rrh __classcell__)rs@rr r sN6 "$+  )   >F(trr ) __future__rtypingrnumpyrMultralytics.solutions.solutionsrrrultralytics.utils.plottingrr rrrrvs)#\\-{tL{tr