|hZddlmZmZmZmZddlZddlmZm Z m Z ddl m Z GddeZ y))AnyDictListTupleN) BaseSolutionSolutionAnnotatorSolutionResults)colorsc eZdZdZdeddffd Zdedeedee e e fd ee e e fde eeff d Z d Z d e jdefd ZxZS) 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}") kwargsreturnNc zt|di|dddddd|_i|_g|_|j y)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)selfr __class__s c/var/www/html/test/engine/venv/lib/python3.12/site-packages/ultralytics/solutions/region_counter.pyrzRegionCounter.__init__'sJ "6"$+#    " !rpolygon_pointsrrc|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)rrr"rrregions r add_regionzRegionCounter.add_region5sY&%%**, <<7 ,(    $$V, r!c \|j|jt|jtsd|ji|_t |jj D]=\}\}}|j ||t|dd}|j|d|d<?y)zInitialize regions only once.Nz Region#01Trrprepared_polygon) r(initialize_region isinstancedict enumerateitemsr)r prep)rirptsr(s r rz RegionCounter.initialize_regionsTs ;;   " " $$++t,& 4DK' (9(9(;< FNA{c__T3q$QF)-6)3D)EF% & Fr!im0c |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]}tt |d j"\} } } } | | f| | f| | f| | fg} |j%| |d |jdz|j'| | | | gt)|d |d |d |jdzd|d <|j+}|j-|t/|t1|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+rrrrr)r7r8 txt_colormargin)plot_im total_tracksr)extract_tracksrr6zipboxesclss track_idsconfs box_labeladjust_box_labelr Pointrcontainsrmapintbounds draw_region text_labelstrresultdisplay_outputr len)rr4 annotatorboxclstrack_idconfcenterr(x1y1x2y2r3r?s r processzRegionCounter.process^s 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 fY&7&>&>?NBB8b"XBx"b:C  ! !#vn'=tQR?R S  RR &*+^, .* !  !F8  !""$ G$wS=P`d`r`rssr!)__name__ __module__ __qualname____doc__rrrPrrrLrr)rnpndarrayr r^ __classcell__)rs@r r r s6 " " "U CcM*  #sC-(  c3h >F&t2::&t/&tr!r )typingrrrrnumpyrcultralytics.solutions.solutionsrrr ultralytics.utils.plottingr r rr!r rjs)*)\\-ytLytr!