dgenerate-ultralytics-headless 8.3.194__py3-none-any.whl → 8.3.196__py3-none-any.whl

ultralytics/solutions/analytics.py

@@ -29,12 +29,12 @@ class Analytics(BaseSolution):
         fontsize (int): Font size for text display.
         color_cycle (cycle): Cyclic iterator for chart colors.
         total_counts (int): Total count of detected objects (used for line charts).
-        clswise_count (Dict[str, int]): Dictionary for class-wise object counts.
+        clswise_count (dict[str, int]): Dictionary for class-wise object counts.
         fig (Figure): Matplotlib figure object for the chart.
         ax (Axes): Matplotlib axes object for the chart.
         canvas (FigureCanvasAgg): Canvas for rendering the chart.
         lines (dict): Dictionary to store line objects for area charts.
-        color_mapping (Dict[str, str]): Dictionary mapping class labels to colors for consistent visualization.
+        color_mapping (dict[str, str]): Dictionary mapping class labels to colors for consistent visualization.
 
     Methods:
         process: Process image data and update the chart.
@@ -144,7 +144,7 @@ class Analytics(BaseSolution):
 
         Args:
             frame_number (int): The current frame number.
-            count_dict (Dict[str, int], optional): Dictionary with class names as keys and counts as values for
+            count_dict (dict[str, int], optional): Dictionary with class names as keys and counts as values for
                 multiple classes. If None, updates a single line graph.
             plot (str): Type of the plot. Options are 'line', 'bar', 'pie', or 'area'.
 

ultralytics/solutions/config.py

@@ -20,20 +20,20 @@ class SolutionConfig:
     Attributes:
         source (str, optional): Path to the input source (video, RTSP, etc.). Only usable with Solutions CLI.
         model (str, optional): Path to the Ultralytics YOLO model to be used for inference.
-        classes (List[int], optional): List of class indices to filter detections.
+        classes (list[int], optional): List of class indices to filter detections.
         show_conf (bool): Whether to show confidence scores on the visual output.
         show_labels (bool): Whether to display class labels on visual output.
-        region (List[Tuple[int, int]], optional): Polygonal region or line for object counting.
+        region (list[tuple[int, int]], optional): Polygonal region or line for object counting.
         colormap (int, optional): OpenCV colormap constant for visual overlays (e.g., cv2.COLORMAP_JET).
         show_in (bool): Whether to display count number for objects entering the region.
         show_out (bool): Whether to display count number for objects leaving the region.
         up_angle (float): Upper angle threshold used in pose-based workouts monitoring.
         down_angle (int): Lower angle threshold used in pose-based workouts monitoring.
-        kpts (List[int]): Keypoint indices to monitor, e.g., for pose analytics.
+        kpts (list[int]): Keypoint indices to monitor, e.g., for pose analytics.
         analytics_type (str): Type of analytics to perform ("line", "area", "bar", "pie", etc.).
-        figsize (Tuple[int, int], optional): Size of the matplotlib figure used for analytical plots (width, height).
+        figsize (tuple[int, int], optional): Size of the matplotlib figure used for analytical plots (width, height).
         blur_ratio (float): Ratio used to blur objects in the video frames (0.0 to 1.0).
-        vision_point (Tuple[int, int]): Reference point for directional tracking or perspective drawing.
+        vision_point (tuple[int, int]): Reference point for directional tracking or perspective drawing.
         crop_dir (str): Directory path to save cropped detection images.
         json_file (str): Path to a JSON file containing data for parking areas.
         line_width (int): Width for visual display i.e. bounding boxes, keypoints, counts.

ultralytics/solutions/distance_calculation.py

@@ -18,8 +18,8 @@ class DistanceCalculation(BaseSolution):
 
     Attributes:
         left_mouse_count (int): Counter for left mouse button clicks.
-        selected_boxes (Dict[int, List[float]]): Dictionary to store selected bounding boxes and their track IDs.
-        centroids (List[List[int]]): List to store centroids of selected bounding boxes.
+        selected_boxes (dict[int, list[float]]): Dictionary to store selected bounding boxes and their track IDs.
+        centroids (list[list[int]]): List to store centroids of selected bounding boxes.
 
     Methods:
         mouse_event_for_distance: Handle mouse events for selecting objects in the video stream.

ultralytics/solutions/heatmap.py

@@ -57,7 +57,7 @@ class Heatmap(ObjectCounter):
         Efficiently calculate heatmap area and effect location for applying colormap.
 
         Args:
-            box (List[float]): Bounding box coordinates [x0, y0, x1, y1].
+            box (list[float]): Bounding box coordinates [x0, y0, x1, y1].
         """
         x0, y0, x1, y1 = map(int, box)
         radius_squared = (min(x1 - x0, y1 - y0) // 2) ** 2

ultralytics/solutions/instance_segmentation.py

@@ -16,10 +16,10 @@ class InstanceSegmentation(BaseSolution):
     Attributes:
         model (str): The segmentation model to use for inference.
         line_width (int): Width of the bounding box and text lines.
-        names (Dict[int, str]): Dictionary mapping class indices to class names.
-        clss (List[int]): List of detected class indices.
-        track_ids (List[int]): List of track IDs for detected instances.
-        masks (List[np.ndarray]): List of segmentation masks for detected instances.
+        names (dict[int, str]): Dictionary mapping class indices to class names.
+        clss (list[int]): List of detected class indices.
+        track_ids (list[int]): List of track IDs for detected instances.
+        masks (list[np.ndarray]): List of segmentation masks for detected instances.
         show_conf (bool): Whether to display confidence scores.
         show_labels (bool): Whether to display class labels.
         show_boxes (bool): Whether to display bounding boxes.

ultralytics/solutions/object_counter.py

@@ -19,8 +19,8 @@ class ObjectCounter(BaseSolution):
     Attributes:
         in_count (int): Counter for objects moving inward.
         out_count (int): Counter for objects moving outward.
-        counted_ids (List[int]): List of IDs of objects that have been counted.
-        classwise_counts (Dict[str, Dict[str, int]]): Dictionary for counts, categorized by object class.
+        counted_ids (list[int]): List of IDs of objects that have been counted.
+        classwise_counts (dict[str, dict[str, int]]): Dictionary for counts, categorized by object class.
         region_initialized (bool): Flag indicating whether the counting region has been initialized.
         show_in (bool): Flag to control display of inward count.
         show_out (bool): Flag to control display of outward count.
@@ -63,9 +63,9 @@ class ObjectCounter(BaseSolution):
         Count objects within a polygonal or linear region based on their tracks.
 
         Args:
-            current_centroid (Tuple[float, float]): Current centroid coordinates (x, y) in the current frame.
+            current_centroid (tuple[float, float]): Current centroid coordinates (x, y) in the current frame.
             track_id (int): Unique identifier for the tracked object.
-            prev_position (Tuple[float, float], optional): Last frame position coordinates (x, y) of the track.
+            prev_position (tuple[float, float], optional): Last frame position coordinates (x, y) of the track.
             cls (int): Class index for classwise count updates.
 
         Examples:

ultralytics/solutions/parking_management.py

@@ -28,8 +28,8 @@ class ParkingPtsSelection:
         canvas (tk.Canvas): The canvas widget for displaying the image and drawing bounding boxes.
         image (PIL.Image.Image): The uploaded image.
         canvas_image (ImageTk.PhotoImage): The image displayed on the canvas.
-        rg_data (List[List[Tuple[int, int]]]): List of bounding boxes, each defined by 4 points.
-        current_box (List[Tuple[int, int]]): Temporary storage for the points of the current bounding box.
+        rg_data (list[list[tuple[int, int]]]): List of bounding boxes, each defined by 4 points.
+        current_box (list[tuple[int, int]]): Temporary storage for the points of the current bounding box.
         imgw (int): Original width of the uploaded image.
         imgh (int): Original height of the uploaded image.
         canvas_max_width (int): Maximum width of the canvas.
@@ -186,11 +186,11 @@ class ParkingManagement(BaseSolution):
 
     Attributes:
         json_file (str): Path to the JSON file containing parking region details.
-        json (List[Dict]): Loaded JSON data containing parking region information.
-        pr_info (Dict[str, int]): Dictionary storing parking information (Occupancy and Available spaces).
-        arc (Tuple[int, int, int]): RGB color tuple for available region visualization.
-        occ (Tuple[int, int, int]): RGB color tuple for occupied region visualization.
-        dc (Tuple[int, int, int]): RGB color tuple for centroid visualization of detected objects.
+        json (list[dict]): Loaded JSON data containing parking region information.
+        pr_info (dict[str, int]): Dictionary storing parking information (Occupancy and Available spaces).
+        arc (tuple[int, int, int]): RGB color tuple for available region visualization.
+        occ (tuple[int, int, int]): RGB color tuple for occupied region visualization.
+        dc (tuple[int, int, int]): RGB color tuple for centroid visualization of detected objects.
 
     Methods:
         process: Process the input image for parking lot management and visualization.

ultralytics/solutions/queue_management.py

@@ -15,10 +15,10 @@ class QueueManager(BaseSolution):
 
     Attributes:
         counts (int): The current count of objects in the queue.
-        rect_color (Tuple[int, int, int]): RGB color tuple for drawing the queue region rectangle.
+        rect_color (tuple[int, int, int]): RGB color tuple for drawing the queue region rectangle.
         region_length (int): The number of points defining the queue region.
-        track_line (List[Tuple[int, int]]): List of track line coordinates.
-        track_history (Dict[int, List[Tuple[int, int]]]): Dictionary storing tracking history for each object.
+        track_line (list[tuple[int, int]]): List of track line coordinates.
+        track_history (dict[int, list[tuple[int, int]]]): Dictionary storing tracking history for each object.
 
     Methods:
         initialize_region: Initialize the queue region.

ultralytics/solutions/region_counter.py

@@ -64,12 +64,12 @@ class RegionCounter(BaseSolution):
 
         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.
+            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.
+            (dict[str, any]): Returns a dictionary including the region information i.e. name, region_color etc.
         """
         region = self.region_template.copy()
         region.update(

ultralytics/solutions/similarity_search.py

@@ -33,7 +33,7 @@ class VisualAISearch:
         data_dir (Path): Path object for the data directory.
         model: Loaded CLIP model.
         index: FAISS index for similarity search.
-        image_paths (List[str]): List of image file paths.
+        image_paths (list[str]): List of image file paths.
 
     Methods:
         extract_image_feature: Extract CLIP embedding from an image.
@@ -138,7 +138,7 @@ class VisualAISearch:
             similarity_thresh (float, optional): Minimum similarity threshold for filtering results.
 
         Returns:
-            (List[str]): List of image filenames ranked by similarity score.
+            (list[str]): List of image filenames ranked by similarity score.
 
         Examples:
             Search for images matching a query

ultralytics/solutions/solutions.py

@@ -30,31 +30,31 @@ class BaseSolution:
         Polygon: Class for creating polygon geometries from shapely.
         Point: Class for creating point geometries from shapely.
         prep: Prepared geometry function from shapely for optimized spatial operations.
-        CFG (Dict[str, Any]): Configuration dictionary loaded from YAML file and updated with kwargs.
+        CFG (dict[str, Any]): Configuration dictionary loaded from YAML file and updated with kwargs.
         LOGGER: Logger instance for solution-specific logging.
         annotator: Annotator instance for drawing on images.
         tracks: YOLO tracking results from the latest inference.
         track_data: Extracted tracking data (boxes or OBB) from tracks.
-        boxes (List): Bounding box coordinates from tracking results.
-        clss (List[int]): Class indices from tracking results.
-        track_ids (List[int]): Track IDs from tracking results.
-        confs (List[float]): Confidence scores from tracking results.
+        boxes (list): Bounding box coordinates from tracking results.
+        clss (list[int]): Class indices from tracking results.
+        track_ids (list[int]): Track IDs from tracking results.
+        confs (list[float]): Confidence scores from tracking results.
         track_line: Current track line for storing tracking history.
         masks: Segmentation masks from tracking results.
         r_s: Region or line geometry object for spatial operations.
         frame_no (int): Current frame number for logging purposes.
-        region (List[Tuple[int, int]]): List of coordinate tuples defining region of interest.
+        region (list[tuple[int, int]]): List of coordinate tuples defining region of interest.
         line_width (int): Width of lines used in visualizations.
         model (YOLO): Loaded YOLO model instance.
-        names (Dict[int, str]): Dictionary mapping class indices to class names.
-        classes (List[int]): List of class indices to track.
+        names (dict[int, str]): Dictionary mapping class indices to class names.
+        classes (list[int]): List of class indices to track.
         show_conf (bool): Flag to show confidence scores in annotations.
         show_labels (bool): Flag to show class labels in annotations.
         device (str): Device for model inference.
-        track_add_args (Dict[str, Any]): Additional arguments for tracking configuration.
+        track_add_args (dict[str, Any]): Additional arguments for tracking configuration.
         env_check (bool): Flag indicating whether environment supports image display.
         track_history (defaultdict): Dictionary storing tracking history for each object.
-        profilers (Tuple): Profiler instances for performance monitoring.
+        profilers (tuple): Profiler instances for performance monitoring.
 
     Methods:
         adjust_box_label: Generate formatted label for bounding box.
@@ -193,7 +193,7 @@ class BaseSolution:
 
         Args:
             track_id (int): The unique identifier for the tracked object.
-            box (List[float]): The bounding box coordinates of the object in the format [x1, y1, x2, y2].
+            box (list[float]): The bounding box coordinates of the object in the format [x1, y1, x2, y2].
 
         Examples:
             >>> solution = BaseSolution()
@@ -333,8 +333,8 @@ class SolutionAnnotator(Annotator):
         Draw a region or line on the image.
 
         Args:
-            reg_pts (List[Tuple[int, int]], optional): Region points (for line 2 points, for region 4+ points).
-            color (Tuple[int, int, int]): RGB color value for the region.
+            reg_pts (list[tuple[int, int]], optional): Region points (for line 2 points, for region 4+ points).
+            color (tuple[int, int, int]): RGB color value for the region.
             thickness (int): Line thickness for drawing the region.
         """
         cv2.polylines(self.im, [np.array(reg_pts, dtype=np.int32)], isClosed=True, color=color, thickness=thickness)
@@ -355,9 +355,9 @@ class SolutionAnnotator(Annotator):
 
         Args:
             label (str): Queue counts label.
-            points (List[Tuple[int, int]], optional): Region points for center point calculation to display text.
-            region_color (Tuple[int, int, int]): RGB queue region color.
-            txt_color (Tuple[int, int, int]): RGB text display color.
+            points (list[tuple[int, int]], optional): Region points for center point calculation to display text.
+            region_color (tuple[int, int, int]): RGB queue region color.
+            txt_color (tuple[int, int, int]): RGB text display color.
         """
         x_values = [point[0] for point in points]
         y_values = [point[1] for point in points]
@@ -402,9 +402,9 @@ class SolutionAnnotator(Annotator):
 
         Args:
             im0 (np.ndarray): Inference image.
-            text (Dict[str, Any]): Labels dictionary.
-            txt_color (Tuple[int, int, int]): Display color for text foreground.
-            bg_color (Tuple[int, int, int]): Display color for text background.
+            text (dict[str, Any]): Labels dictionary.
+            txt_color (tuple[int, int, int]): Display color for text foreground.
+            bg_color (tuple[int, int, int]): Display color for text background.
             margin (int): Gap between text and rectangle for better display.
         """
         horizontal_gap = int(im0.shape[1] * 0.02)
@@ -432,9 +432,9 @@ class SolutionAnnotator(Annotator):
         Calculate the angle between three points for workout monitoring.
 
         Args:
-            a (List[float]): The coordinates of the first point.
-            b (List[float]): The coordinates of the second point (vertex).
-            c (List[float]): The coordinates of the third point.
+            a (list[float]): The coordinates of the first point.
+            b (list[float]): The coordinates of the second point (vertex).
+            c (list[float]): The coordinates of the third point.
 
         Returns:
             (float): The angle in degrees between the three points.
@@ -454,8 +454,8 @@ class SolutionAnnotator(Annotator):
         Draw specific keypoints for gym steps counting.
 
         Args:
-            keypoints (List[List[float]]): Keypoints data to be plotted, each in format [x, y, confidence].
-            indices (List[int], optional): Keypoint indices to be plotted.
+            keypoints (list[list[float]]): Keypoints data to be plotted, each in format [x, y, confidence].
+            indices (list[int], optional): Keypoint indices to be plotted.
             radius (int): Keypoint radius.
             conf_thresh (float): Confidence threshold for keypoints.
 
@@ -491,9 +491,9 @@ class SolutionAnnotator(Annotator):
 
         Args:
             display_text (str): The text to be displayed.
-            position (Tuple[int, int]): Coordinates (x, y) on the image where the text will be placed.
-            color (Tuple[int, int, int]): Text background color.
-            txt_color (Tuple[int, int, int]): Text foreground color.
+            position (tuple[int, int]): Coordinates (x, y) on the image where the text will be placed.
+            color (tuple[int, int, int]): Text background color.
+            txt_color (tuple[int, int, int]): Text foreground color.
 
         Returns:
             (int): The height of the text.
@@ -529,9 +529,9 @@ class SolutionAnnotator(Annotator):
             angle_text (str): Angle value for workout monitoring.
             count_text (str): Counts value for workout monitoring.
             stage_text (str): Stage decision for workout monitoring.
-            center_kpt (List[int]): Centroid pose index for workout monitoring.
-            color (Tuple[int, int, int]): Text background color.
-            txt_color (Tuple[int, int, int]): Text foreground color.
+            center_kpt (list[int]): Centroid pose index for workout monitoring.
+            color (tuple[int, int, int]): Text background color.
+            txt_color (tuple[int, int, int]): Text foreground color.
         """
         # Format text
         angle_text, count_text, stage_text = f" {angle_text:.2f}", f"Steps : {count_text}", f" {stage_text}"
@@ -559,9 +559,9 @@ class SolutionAnnotator(Annotator):
 
         Args:
             pixels_distance (float): Pixels distance between two bbox centroids.
-            centroids (List[Tuple[int, int]]): Bounding box centroids data.
-            line_color (Tuple[int, int, int]): Distance line color.
-            centroid_color (Tuple[int, int, int]): Bounding box centroid color.
+            centroids (list[tuple[int, int]]): Bounding box centroids data.
+            line_color (tuple[int, int, int]): Distance line color.
+            centroid_color (tuple[int, int, int]): Bounding box centroid color.
         """
         # Get the text size
         text = f"Pixels Distance: {pixels_distance:.2f}"
@@ -603,8 +603,8 @@ class SolutionAnnotator(Annotator):
         Args:
             im0 (np.ndarray): Inference image.
             text (str): Object/class name.
-            txt_color (Tuple[int, int, int]): Display color for text foreground.
-            bg_color (Tuple[int, int, int]): Display color for text background.
+            txt_color (tuple[int, int, int]): Display color for text foreground.
+            bg_color (tuple[int, int, int]): Display color for text background.
             x_center (float): The x position center point for bounding box.
             y_center (float): The y position center point for bounding box.
             margin (int): The gap between text and rectangle for better display.
@@ -651,8 +651,8 @@ class SolutionAnnotator(Annotator):
             line_x (int): The x-coordinate of the sweep line.
             line_y (int): The y-coordinate limit of the sweep line.
             label (str, optional): Text label to be drawn in center of sweep line. If None, no label is drawn.
-            color (Tuple[int, int, int]): RGB color for the line and label background.
-            txt_color (Tuple[int, int, int]): RGB color for the label text.
+            color (tuple[int, int, int]): RGB color for the line and label background.
+            txt_color (tuple[int, int, int]): RGB color for the label text.
         """
         # Draw the sweep line
         cv2.line(self.im, (line_x, 0), (line_x, line_y), color, self.tf * 2)
@@ -688,10 +688,10 @@ class SolutionAnnotator(Annotator):
         Perform pinpoint human-vision eye mapping and plotting.
 
         Args:
-            box (List[float]): Bounding box coordinates in format [x1, y1, x2, y2].
-            center_point (Tuple[int, int]): Center point for vision eye view.
-            color (Tuple[int, int, int]): Object centroid and line color.
-            pin_color (Tuple[int, int, int]): Visioneye point color.
+            box (list[float]): Bounding box coordinates in format [x1, y1, x2, y2].
+            center_point (tuple[int, int]): Center point for vision eye view.
+            color (tuple[int, int, int]): Object centroid and line color.
+            pin_color (tuple[int, int, int]): Visioneye point color.
         """
         center_bbox = int((box[0] + box[2]) / 2), int((box[1] + box[3]) / 2)
         cv2.circle(self.im, center_point, self.tf * 2, pin_color, -1)
@@ -711,10 +711,10 @@ class SolutionAnnotator(Annotator):
         Draw a label with a background rectangle or circle centered within a given bounding box.
 
         Args:
-            box (Tuple[float, float, float, float]): The bounding box coordinates (x1, y1, x2, y2).
+            box (tuple[float, float, float, float]): The bounding box coordinates (x1, y1, x2, y2).
             label (str): The text label to be displayed.
-            color (Tuple[int, int, int]): The background color of the rectangle (B, G, R).
-            txt_color (Tuple[int, int, int]): The color of the text (R, G, B).
+            color (tuple[int, int, int]): The background color of the rectangle (B, G, R).
+            txt_color (tuple[int, int, int]): The color of the text (R, G, B).
             shape (str): The shape of the label i.e "circle" or "rect"
             margin (int): The margin between the text and the rectangle border.
         """
@@ -768,7 +768,7 @@ class SolutionResults:
         plot_im (np.ndarray): Processed image with counts, blurred, or other effects from solutions.
         in_count (int): The total number of "in" counts in a video stream.
         out_count (int): The total number of "out" counts in a video stream.
-        classwise_count (Dict[str, int]): A dictionary containing counts of objects categorized by class.
+        classwise_count (dict[str, int]): A dictionary containing counts of objects categorized by class.
         queue_count (int): The count of objects in a queue or waiting area.
         workout_count (int): The count of workout repetitions.
         workout_angle (float): The angle calculated during a workout exercise.
@@ -778,10 +778,10 @@ class SolutionResults:
         filled_slots (int): The number of filled slots in a monitored area.
         email_sent (bool): A flag indicating whether an email notification was sent.
         total_tracks (int): The total number of tracked objects.
-        region_counts (Dict[str, int]): The count of objects within a specific region.
-        speed_dict (Dict[str, float]): A dictionary containing speed information for tracked objects.
+        region_counts (dict[str, int]): The count of objects within a specific region.
+        speed_dict (dict[str, float]): A dictionary containing speed information for tracked objects.
         total_crop_objects (int): Total number of cropped objects using ObjectCropper class.
-        speed (Dict[str, float]): Performance timing information for tracking and solution processing.
+        speed (dict[str, float]): Performance timing information for tracking and solution processing.
     """
 
     def __init__(self, **kwargs):

ultralytics/solutions/streamlit_inference.py

@@ -34,7 +34,7 @@ class Inference:
         org_frame (Any): Container for the original frame to be displayed.
         ann_frame (Any): Container for the annotated frame to be displayed.
         vid_file_name (str | int): Name of the uploaded video file or webcam index.
-        selected_ind (List[int]): List of selected class indices for detection.
+        selected_ind (list[int]): List of selected class indices for detection.
 
     Methods:
         web_ui: Set up the Streamlit web interface with custom HTML elements.

ultralytics/solutions/trackzone.py

@@ -19,10 +19,10 @@ class TrackZone(BaseSolution):
     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.
+        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.

ultralytics/solutions/vision_eye.py

@@ -14,7 +14,7 @@ class VisionEye(BaseSolution):
     mapping vision points, and annotating results with bounding boxes and labels.
 
     Attributes:
-        vision_point (Tuple[int, int]): Coordinates (x, y) where vision will view objects and draw tracks.
+        vision_point (tuple[int, int]): Coordinates (x, y) where vision will view objects and draw tracks.
 
     Methods:
         process: Process the input image to detect objects, annotate them, and apply vision mapping.

ultralytics/trackers/byte_tracker.py

@@ -58,7 +58,7 @@ class STrack(BaseTrack):
         Initialize a new STrack instance.
 
         Args:
-            xywh (List[float]): Bounding box coordinates and dimensions in the format (x, y, w, h, [a], idx), where
+            xywh (list[float]): Bounding box coordinates and dimensions in the format (x, y, w, h, [a], idx), where
                 (x, y) is the center, (w, h) are width and height, [a] is optional aspect ratio, and idx is the id.
             score (float): Confidence score of the detection.
             cls (Any): Class label for the detected object.
@@ -246,9 +246,9 @@ class BYTETracker:
     predicting the new object locations, and performs data association.
 
     Attributes:
-        tracked_stracks (List[STrack]): List of successfully activated tracks.
-        lost_stracks (List[STrack]): List of lost tracks.
-        removed_stracks (List[STrack]): List of removed tracks.
+        tracked_stracks (list[STrack]): List of successfully activated tracks.
+        lost_stracks (list[STrack]): List of lost tracks.
+        removed_stracks (list[STrack]): List of removed tracks.
         frame_id (int): The current frame ID.
         args (Namespace): Command-line arguments.
         max_time_lost (int): The maximum frames for a track to be considered as 'lost'.
@@ -286,9 +286,9 @@ class BYTETracker:
             >>> args = Namespace(track_buffer=30)
             >>> tracker = BYTETracker(args, frame_rate=30)
         """
-        self.tracked_stracks = []  # type: List[STrack]
-        self.lost_stracks = []  # type: List[STrack]
-        self.removed_stracks = []  # type: List[STrack]
+        self.tracked_stracks = []  # type: list[STrack]
+        self.lost_stracks = []  # type: list[STrack]
+        self.removed_stracks = []  # type: list[STrack]
 
         self.frame_id = 0
         self.args = args
@@ -320,7 +320,7 @@ class BYTETracker:
         detections = self.init_track(results, feats_keep)
         # Add newly detected tracklets to tracked_stracks
         unconfirmed = []
-        tracked_stracks = []  # type: List[STrack]
+        tracked_stracks = []  # type: list[STrack]
         for track in self.tracked_stracks:
             if not track.is_activated:
                 unconfirmed.append(track)
@@ -439,9 +439,9 @@ class BYTETracker:
 
     def reset(self):
         """Reset the tracker by clearing all tracked, lost, and removed tracks and reinitializing the Kalman filter."""
-        self.tracked_stracks = []  # type: List[STrack]
-        self.lost_stracks = []  # type: List[STrack]
-        self.removed_stracks = []  # type: List[STrack]
+        self.tracked_stracks = []  # type: list[STrack]
+        self.lost_stracks = []  # type: list[STrack]
+        self.removed_stracks = []  # type: list[STrack]
         self.frame_id = 0
         self.kalman_filter = self.get_kalmanfilter()
         self.reset_id()

ultralytics/trackers/utils/gmc.py

@@ -21,7 +21,7 @@ class GMC:
         method (str): The tracking method to use. Options include 'orb', 'sift', 'ecc', 'sparseOptFlow', 'none'.
         downscale (int): Factor by which to downscale the frames for processing.
         prevFrame (np.ndarray): Previous frame for tracking.
-        prevKeyPoints (List): Keypoints from the previous frame.
+        prevKeyPoints (list): Keypoints from the previous frame.
         prevDescriptors (np.ndarray): Descriptors from the previous frame.
         initializedFirstFrame (bool): Flag indicating if the first frame has been processed.
 
@@ -96,7 +96,7 @@ class GMC:
 
         Args:
             raw_frame (np.ndarray): The raw frame to be processed, with shape (H, W, C).
-            detections (List, optional): List of detections to be used in the processing.
+            detections (list, optional): List of detections to be used in the processing.
 
         Returns:
             (np.ndarray): Transformation matrix with shape (2, 3).
@@ -163,7 +163,7 @@ class GMC:
 
         Args:
             raw_frame (np.ndarray): The raw frame to be processed, with shape (H, W, C).
-            detections (List, optional): List of detections to be used in the processing.
+            detections (list, optional): List of detections to be used in the processing.
 
         Returns:
             (np.ndarray): Transformation matrix with shape (2, 3).

ultralytics/trackers/utils/matching.py

@@ -66,8 +66,8 @@ def iou_distance(atracks: list, btracks: list) -> np.ndarray:
     Compute cost based on Intersection over Union (IoU) between tracks.
 
     Args:
-        atracks (List[STrack] | List[np.ndarray]): List of tracks 'a' or bounding boxes.
-        btracks (List[STrack] | List[np.ndarray]): List of tracks 'b' or bounding boxes.
+        atracks (list[STrack] | list[np.ndarray]): List of tracks 'a' or bounding boxes.
+        btracks (list[STrack] | list[np.ndarray]): List of tracks 'b' or bounding boxes.
 
     Returns:
         (np.ndarray): Cost matrix computed based on IoU with shape (len(atracks), len(btracks)).
@@ -106,8 +106,8 @@ def embedding_distance(tracks: list, detections: list, metric: str = "cosine") -
     Compute distance between tracks and detections based on embeddings.
 
     Args:
-        tracks (List[STrack]): List of tracks, where each track contains embedding features.
-        detections (List[BaseTrack]): List of detections, where each detection contains embedding features.
+        tracks (list[STrack]): List of tracks, where each track contains embedding features.
+        detections (list[BaseTrack]): List of detections, where each detection contains embedding features.
         metric (str): Metric for distance computation. Supported metrics include 'cosine', 'euclidean', etc.
 
     Returns:
@@ -137,7 +137,7 @@ def fuse_score(cost_matrix: np.ndarray, detections: list) -> np.ndarray:
 
     Args:
         cost_matrix (np.ndarray): The matrix containing cost values for assignments, with shape (N, M).
-        detections (List[BaseTrack]): List of detections, each containing a score attribute.
+        detections (list[BaseTrack]): List of detections, each containing a score attribute.
 
     Returns:
         (np.ndarray): Fused similarity matrix with shape (N, M).