hZSSKJr SSKJr SSKrSSKJrJrJ r SSK J r "SS\5r g)) annotations)AnyN) BaseSolutionSolutionAnnotatorSolutionResults)colorsc^^\rSrSrSrSU4SjjrS SjrSrS SjrSr U=r $) RegionCounter a 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 |>[TU]"S0UD6 SSSSSS.Ul0Ul/UlUR 5 g)zSInitialize the RegionCounter for real-time object counting in user-defined regions.zDefault RegionNrrr)rrr)namepolygoncounts region_color text_color)super__init__region_template region_countscounting_regionsinitialize_regions)selfkwargs __class__s ^/home/james-whalen/.local/lib/python3.13/site-packages/ultralytics/solutions/region_counter.pyrRegionCounter.__init__)sJ "6"$+#    " !cURR5nURUURU5UUS.5 URR U5 U$)a 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_regionRegionCounter.add_region7sY&%%**, <<7 ,(    $$V, r c hURcUR5 [UR[5(dSUR0Ul[ URR 55H<unup#UR X#[US5S5nURUS5US'M> g)zInitialize regions only once.Nz Region#01Tr rprepared_polygon) r'initialize_region isinstancedict enumerateitemsr(rprep)rirptsr's rr RegionCounter.initialize_regionsVs ;;   " " $$++t,,& 4DK' (9(9(;>SWS]S](^ $Ch   +@+@H+U]cdlnr]s  tZZ#a&3q6/Q!6Q#a&A8M NOF//,-66v>>8$)$9?9ID&&vf~60)_++F)$Ds5"((4==+?+?rxx"PQRC#&t}}#7T]]__9M"N!ORS!S HRhr  ! !#n'=tQR?R S  $ $ &*+^, .* %  !F8 ,""$ G$wS=P`d`r`rssr )rr'rr)rrreturnNone) rr\r&z list[tuple]rtuple[int, int, int]rrprnzdict[str, Any])r`z np.ndarrayrnr) __name__ __module__ __qualname____firstlineno____doc__rr(rrl__static_attributes__ __classcell__)rs@rr r sT6 "$+  )   >F(t(tr r ) __future__rtypingrnumpyrQultralytics.solutions.solutionsrrrultralytics.utils.plottingrr rr rr}s)#\\-{tL{tr