heVSSKJr SSKrSSKrSSKJrJrJr SSK J r "SS\5r g))AnyN) BaseSolutionSolutionAnnotatorSolutionResults)colorsc^^\rSrSrSrS\SS4U4SjjrS\RS\ 4Sjr S r U=r $) TrackZone ab A class to manage region-based object tracking in a video stream. This class extends the BaseSolution class and provides functionality for tracking objects within a specific region defined by a polygonal area. Objects outside the region are excluded from tracking. Attributes: region (np.ndarray): The polygonal region for tracking, represented as a convex hull of points. line_width (int): Width of the lines used for drawing bounding boxes and region boundaries. names (list[str]): List of class names that the model can detect. boxes (list[np.ndarray]): Bounding boxes of tracked objects. track_ids (list[int]): Unique identifiers for each tracked object. clss (list[int]): Class indices of tracked objects. Methods: process: Process each frame of the video, applying region-based tracking. extract_tracks: Extract tracking information from the input frame. display_output: Display the processed output. Examples: >>> tracker = TrackZone() >>> frame = cv2.imread("frame.jpg") >>> results = tracker.process(frame) >>> cv2.imshow("Tracked Frame", results.plot_im) kwargsreturnNc >[TU]"S0UD6 /SQn[R"[R "UR =(d U[RS95UlSUlg)z Initialize the TrackZone class for tracking objects within a defined region in video streams. Args: **kwargs (Any): Additional keyword arguments passed to the parent class. ))Kr)5r)r)rr)dtypeN) super__init__cv2 convexHullnparrayregionint32mask)selfr default_region __class__s Y/home/james-whalen/.local/lib/python3.13/site-packages/ultralytics/solutions/trackzone.pyrTrackZone.__init__'sL "6"EnnRXXdkk.K^SUS[S[%\]  im0c [XRS9nURcR[R"USS2SS2S45Ul[ R "URUR/S5 [ R"XURS9nURU5 [ R"XR/SSURS-S 9 [URURURUR5H.upEpgUR!X@R#XgUS 9[%US5S 9 M0 UR'5nUR)U5 [+U[-UR5S 9$) a Process the input frame to track objects within a defined region. This method initializes the annotator, creates a mask for the specified region, extracts tracks only from the masked area, and updates tracking information. Objects outside the region are ignored. Args: im0 (np.ndarray): The input image or frame to be processed. Returns: (SolutionResults): Contains processed image `plot_im` and `total_tracks` (int) representing the total number of tracked objects within the defined region. Examples: >>> tracker = TrackZone() >>> frame = cv2.imread("path/to/image.jpg") >>> results = tracker.process(frame) ) line_widthNr)rT)r%r%r%)isClosedcolor thickness)track_id)labelr()plot_im total_tracks)rr$rr zeros_likerfillPolyr bitwise_andextract_tracks polylineszipboxes track_idsclssconfs box_labeladjust_box_labelrresultdisplay_outputrlen) rr" annotator masked_frameboxr*clsconfr,s rprocessTrackZone.process3s1&&cooF 99  c!Q'l3DI LLT[[M3 7sdii@  L) cKK=4Z^ZiZilmZmn),DJJ SWS]S](^ $C3   00X0NV\]egkVl  )_ ""$ G$wS=PQQr!)rr) __name__ __module__ __qualname____firstlineno____doc__rrrndarrayrrB__static_attributes__ __classcell__)rs@rr r s;4   (R2::(R/(R(Rr!r ) typingrrnumpyrultralytics.solutions.solutionsrrrultralytics.utils.plottingrr rr!rrPs) \\-OR ORr!