dgenerate-ultralytics-headless 8.3.222__py3-none-any.whl → 8.3.225__py3-none-any.whl

ultralytics/solutions/region_counter.py

@@ -11,8 +11,7 @@ from ultralytics.utils.plotting import colors
 
 
 class RegionCounter(BaseSolution):
-    """
-    A class for real-time counting of objects within user-defined regions in a video stream.
+    """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
@@ -21,8 +20,8 @@ class RegionCounter(BaseSolution):
     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.
+        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:
@@ -59,8 +58,7 @@ class RegionCounter(BaseSolution):
         region_color: tuple[int, int, int],
         text_color: tuple[int, int, int],
     ) -> dict[str, Any]:
-        """
-        Add a new region to the counting list based on the provided template with specific attributes.
+        """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.
@@ -94,15 +92,14 @@ class RegionCounter(BaseSolution):
             region["prepared_polygon"] = self.prep(region["polygon"])
 
     def process(self, im0: np.ndarray) -> SolutionResults:
-        """
-        Process the input frame to detect and count objects within each defined region.
+        """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).
+            (SolutionResults): Contains processed image `plot_im`, 'total_tracks' (int, total number of tracked
+                objects), and 'region_counts' (dict, counts of objects per region).
         """
         self.extract_tracks(im0)
         annotator = SolutionAnnotator(im0, line_width=self.line_width)

ultralytics/solutions/security_alarm.py

@@ -8,8 +8,7 @@ from ultralytics.utils.plotting import colors
 
 
 class SecurityAlarm(BaseSolution):
-    """
-    A class to manage security alarm functionalities for real-time monitoring.
+    """A class to manage security alarm functionalities for real-time monitoring.
 
     This class extends the BaseSolution class and provides features to monitor objects in a frame, send email
     notifications when specific thresholds are exceeded for total detections, and annotate the output frame for
@@ -35,8 +34,7 @@ class SecurityAlarm(BaseSolution):
     """
 
     def __init__(self, **kwargs: Any) -> None:
-        """
-        Initialize the SecurityAlarm class with parameters for real-time object monitoring.
+        """Initialize the SecurityAlarm class with parameters for real-time object monitoring.
 
         Args:
             **kwargs (Any): Additional keyword arguments passed to the parent class.
@@ -49,16 +47,15 @@ class SecurityAlarm(BaseSolution):
         self.from_email = ""
 
     def authenticate(self, from_email: str, password: str, to_email: str) -> None:
-        """
-        Authenticate the email server for sending alert notifications.
+        """Authenticate the email server for sending alert notifications.
+
+        This method initializes a secure connection with the SMTP server and logs in using the provided credentials.
 
         Args:
             from_email (str): Sender's email address.
             password (str): Password for the sender's email account.
             to_email (str): Recipient's email address.
 
-        This method initializes a secure connection with the SMTP server and logs in using the provided credentials.
-
         Examples:
             >>> alarm = SecurityAlarm()
             >>> alarm.authenticate("sender@example.com", "password123", "recipient@example.com")
@@ -72,16 +69,15 @@ class SecurityAlarm(BaseSolution):
         self.from_email = from_email
 
     def send_email(self, im0, records: int = 5) -> None:
-        """
-        Send an email notification with an image attachment indicating the number of objects detected.
+        """Send an email notification with an image attachment indicating the number of objects detected.
+
+        This method encodes the input image, composes the email message with details about the detection, and sends it
+        to the specified recipient.
 
         Args:
             im0 (np.ndarray): The input image or frame to be attached to the email.
             records (int, optional): The number of detected objects to be included in the email message.
 
-        This method encodes the input image, composes the email message with details about the detection, and sends it
-        to the specified recipient.
-
         Examples:
             >>> alarm = SecurityAlarm()
             >>> frame = cv2.imread("path/to/image.jpg")
@@ -117,8 +113,11 @@ class SecurityAlarm(BaseSolution):
             LOGGER.error(f"Failed to send email: {e}")
 
     def process(self, im0) -> SolutionResults:
-        """
-        Monitor the frame, process object detections, and trigger alerts if thresholds are exceeded.
+        """Monitor the frame, process object detections, and trigger alerts if thresholds are exceeded.
+
+        This method processes the input frame, extracts detections, annotates the frame with bounding boxes, and sends
+        an email notification if the number of detected objects surpasses the specified threshold and an alert has not
+        already been sent.
 
         Args:
             im0 (np.ndarray): The input image or frame to be processed and annotated.
@@ -127,10 +126,6 @@ class SecurityAlarm(BaseSolution):
             (SolutionResults): Contains processed image `plot_im`, 'total_tracks' (total number of tracked objects) and
                 'email_sent' (whether an email alert was triggered).
 
-        This method processes the input frame, extracts detections, annotates the frame with bounding boxes, and sends
-        an email notification if the number of detected objects surpasses the specified threshold and an alert has not
-        already been sent.
-
         Examples:
             >>> alarm = SecurityAlarm()
             >>> frame = cv2.imread("path/to/image.jpg")

ultralytics/solutions/similarity_search.py

@@ -18,8 +18,7 @@ os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"  # Avoid OpenMP conflict on some sys
 
 
 class VisualAISearch:
-    """
-    A semantic image search system that leverages OpenCLIP for generating high-quality image and text embeddings and
+    """A semantic image search system that leverages OpenCLIP for generating high-quality image and text embeddings and
     FAISS for fast similarity-based retrieval.
 
     This class aligns image and text embeddings in a shared semantic space, enabling users to search large collections
@@ -85,8 +84,7 @@ class VisualAISearch:
         return self.model.encode_text(self.model.tokenize([text])).detach().cpu().numpy()
 
     def load_or_build_index(self) -> None:
-        """
-        Load existing FAISS index or build a new one from image features.
+        """Load existing FAISS index or build a new one from image features.
 
         Checks if FAISS index and image paths exist on disk. If found, loads them directly. Otherwise, builds a new
         index by extracting features from all images in the data directory, normalizes the features, and saves both the
@@ -130,8 +128,7 @@ class VisualAISearch:
         LOGGER.info(f"Indexed {len(self.image_paths)} images.")
 
     def search(self, query: str, k: int = 30, similarity_thresh: float = 0.1) -> list[str]:
-        """
-        Return top-k semantically similar images to the given query.
+        """Return top-k semantically similar images to the given query.
 
         Args:
             query (str): Natural language text query to search for.
@@ -167,11 +164,10 @@ class VisualAISearch:
 
 
 class SearchApp:
-    """
-    A Flask-based web interface for semantic image search with natural language queries.
+    """A Flask-based web interface for semantic image search with natural language queries.
 
-    This class provides a clean, responsive frontend that enables users to input natural language queries and
-    instantly view the most relevant images retrieved from the indexed database.
+    This class provides a clean, responsive frontend that enables users to input natural language queries and instantly
+    view the most relevant images retrieved from the indexed database.
 
     Attributes:
         render_template: Flask template rendering function.
@@ -190,8 +186,7 @@ class SearchApp:
     """
 
     def __init__(self, data: str = "images", device: str | None = None) -> None:
-        """
-        Initialize the SearchApp with VisualAISearch backend.
+        """Initialize the SearchApp with VisualAISearch backend.
 
         Args:
             data (str, optional): Path to directory containing images to index and search.

ultralytics/solutions/solutions.py

@@ -18,8 +18,7 @@ from ultralytics.utils.plotting import Annotator
 
 
 class BaseSolution:
-    """
-    A base class for managing Ultralytics Solutions.
+    """A base class for managing Ultralytics Solutions.
 
     This class provides core functionality for various Ultralytics Solutions, including model loading, object tracking,
     and region initialization. It serves as the foundation for implementing specific computer vision solutions such as
@@ -73,8 +72,7 @@ class BaseSolution:
     """
 
     def __init__(self, is_cli: bool = False, **kwargs: Any) -> None:
-        """
-        Initialize the BaseSolution class with configuration settings and YOLO model.
+        """Initialize the BaseSolution class with configuration settings and YOLO model.
 
         Args:
             is_cli (bool): Enable CLI mode if set to True.
@@ -138,12 +136,11 @@ class BaseSolution:
         )
 
     def adjust_box_label(self, cls: int, conf: float, track_id: int | None = None) -> str | None:
-        """
-        Generate a formatted label for a bounding box.
+        """Generate a formatted label for a bounding box.
 
-        This method constructs a label string for a bounding box using the class index and confidence score.
-        Optionally includes the track ID if provided. The label format adapts based on the display settings
-        defined in `self.show_conf` and `self.show_labels`.
+        This method constructs a label string for a bounding box using the class index and confidence score. Optionally
+        includes the track ID if provided. The label format adapts based on the display settings defined in
+        `self.show_conf` and `self.show_labels`.
 
         Args:
             cls (int): The class index of the detected object.
@@ -157,8 +154,7 @@ class BaseSolution:
         return (f"{name} {conf:.2f}" if self.show_conf else name) if self.show_labels else None
 
     def extract_tracks(self, im0: np.ndarray) -> None:
-        """
-        Apply object tracking and extract tracks from an input image or frame.
+        """Apply object tracking and extract tracks from an input image or frame.
 
         Args:
             im0 (np.ndarray): The input image or frame.
@@ -185,11 +181,10 @@ class BaseSolution:
             self.boxes, self.clss, self.track_ids, self.confs = [], [], [], []
 
     def store_tracking_history(self, track_id: int, box) -> None:
-        """
-        Store the tracking history of an object.
+        """Store the tracking history of an object.
 
-        This method updates the tracking history for a given object by appending the center point of its
-        bounding box to the track line. It maintains a maximum of 30 points in the tracking history.
+        This method updates the tracking history for a given object by appending the center point of its bounding box to
+        the track line. It maintains a maximum of 30 points in the tracking history.
 
         Args:
             track_id (int): The unique identifier for the tracked object.
@@ -214,8 +209,8 @@ class BaseSolution:
         )  # region or line
 
     def display_output(self, plot_im: np.ndarray) -> None:
-        """
-        Display the results of the processing, which could involve showing frames, printing counts, or saving results.
+        """Display the results of the processing, which could involve showing frames, printing counts, or saving
+        results.
 
         This method is responsible for visualizing the output of the object detection and tracking process. It displays
         the processed frame with annotations, and allows for user interaction to close the display.
@@ -264,8 +259,7 @@ class BaseSolution:
 
 
 class SolutionAnnotator(Annotator):
-    """
-    A specialized annotator class for visualizing and analyzing computer vision tasks.
+    """A specialized annotator class for visualizing and analyzing computer vision tasks.
 
     This class extends the base Annotator class, providing additional methods for drawing regions, centroids, tracking
     trails, and visual annotations for Ultralytics Solutions. It offers comprehensive visualization capabilities for
@@ -310,8 +304,7 @@ class SolutionAnnotator(Annotator):
         pil: bool = False,
         example: str = "abc",
     ):
-        """
-        Initialize the SolutionAnnotator class with an image for annotation.
+        """Initialize the SolutionAnnotator class with an image for annotation.
 
         Args:
             im (np.ndarray): The image to be annotated.
@@ -329,8 +322,7 @@ class SolutionAnnotator(Annotator):
         color: tuple[int, int, int] = (0, 255, 0),
         thickness: int = 5,
     ):
-        """
-        Draw a region or line on the image.
+        """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).
@@ -350,8 +342,7 @@ class SolutionAnnotator(Annotator):
         region_color: tuple[int, int, int] = (255, 255, 255),
         txt_color: tuple[int, int, int] = (0, 0, 0),
     ):
-        """
-        Display queue counts on an image centered at the points with customizable font size and colors.
+        """Display queue counts on an image centered at the points with customizable font size and colors.
 
         Args:
             label (str): Queue counts label.
@@ -397,8 +388,7 @@ class SolutionAnnotator(Annotator):
         bg_color: tuple[int, int, int],
         margin: int,
     ):
-        """
-        Display the overall statistics for parking lots, object counter etc.
+        """Display the overall statistics for parking lots, object counter etc.
 
         Args:
             im0 (np.ndarray): Inference image.
@@ -428,8 +418,7 @@ class SolutionAnnotator(Annotator):
     @staticmethod
     @lru_cache(maxsize=256)
     def estimate_pose_angle(a: list[float], b: list[float], c: list[float]) -> float:
-        """
-        Calculate the angle between three points for workout monitoring.
+        """Calculate the angle between three points for workout monitoring.
 
         Args:
             a (list[float]): The coordinates of the first point.
@@ -450,8 +439,7 @@ class SolutionAnnotator(Annotator):
         radius: int = 2,
         conf_thresh: float = 0.25,
     ) -> np.ndarray:
-        """
-        Draw specific keypoints for gym steps counting.
+        """Draw specific keypoints for gym steps counting.
 
         Args:
             keypoints (list[list[float]]): Keypoints data to be plotted, each in format [x, y, confidence].
@@ -486,8 +474,7 @@ class SolutionAnnotator(Annotator):
         color: tuple[int, int, int] = (104, 31, 17),
         txt_color: tuple[int, int, int] = (255, 255, 255),
     ) -> int:
-        """
-        Draw workout text with a background on the image.
+        """Draw workout text with a background on the image.
 
         Args:
             display_text (str): The text to be displayed.
@@ -522,8 +509,7 @@ class SolutionAnnotator(Annotator):
         color: tuple[int, int, int] = (104, 31, 17),
         txt_color: tuple[int, int, int] = (255, 255, 255),
     ):
-        """
-        Plot the pose angle, count value, and step stage for workout monitoring.
+        """Plot the pose angle, count value, and step stage for workout monitoring.
 
         Args:
             angle_text (str): Angle value for workout monitoring.
@@ -554,8 +540,7 @@ class SolutionAnnotator(Annotator):
         line_color: tuple[int, int, int] = (104, 31, 17),
         centroid_color: tuple[int, int, int] = (255, 0, 255),
     ):
-        """
-        Plot the distance and line between two centroids on the frame.
+        """Plot the distance and line between two centroids on the frame.
 
         Args:
             pixels_distance (float): Pixels distance between two bbox centroids.
@@ -597,8 +582,7 @@ class SolutionAnnotator(Annotator):
         y_center: float,
         margin: int,
     ):
-        """
-        Display the bounding boxes labels in parking management app.
+        """Display the bounding boxes labels in parking management app.
 
         Args:
             im0 (np.ndarray): Inference image.
@@ -644,8 +628,7 @@ class SolutionAnnotator(Annotator):
         color: tuple[int, int, int] = (221, 0, 186),
         txt_color: tuple[int, int, int] = (255, 255, 255),
     ):
-        """
-        Draw a sweep annotation line and an optional label.
+        """Draw a sweep annotation line and an optional label.
 
         Args:
             line_x (int): The x-coordinate of the sweep line.
@@ -684,8 +667,7 @@ class SolutionAnnotator(Annotator):
         color: tuple[int, int, int] = (235, 219, 11),
         pin_color: tuple[int, int, int] = (255, 0, 255),
     ):
-        """
-        Perform pinpoint human-vision eye mapping and plotting.
+        """Perform pinpoint human-vision eye mapping and plotting.
 
         Args:
             box (list[float]): Bounding box coordinates in format [x1, y1, x2, y2].
@@ -707,8 +689,7 @@ class SolutionAnnotator(Annotator):
         shape: str = "rect",
         margin: int = 5,
     ):
-        """
-        Draw a label with a background rectangle or circle centered within a given bounding box.
+        """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).
@@ -757,12 +738,11 @@ class SolutionAnnotator(Annotator):
 
 
 class SolutionResults:
-    """
-    A class to encapsulate the results of Ultralytics Solutions.
+    """A class to encapsulate the results of Ultralytics Solutions.
 
     This class is designed to store and manage various outputs generated by the solution pipeline, including counts,
-    angles, workout stages, and other analytics data. It provides a structured way to access and manipulate results
-    from different computer vision solutions such as object counting, pose estimation, and tracking analytics.
+    angles, workout stages, and other analytics data. It provides a structured way to access and manipulate results from
+    different computer vision solutions such as object counting, pose estimation, and tracking analytics.
 
     Attributes:
         plot_im (np.ndarray): Processed image with counts, blurred, or other effects from solutions.
@@ -785,8 +765,7 @@ class SolutionResults:
     """
 
     def __init__(self, **kwargs):
-        """
-        Initialize a SolutionResults object with default or user-specified values.
+        """Initialize a SolutionResults object with default or user-specified values.
 
         Args:
             **kwargs (Any): Optional arguments to override default attribute values.
@@ -813,8 +792,7 @@ class SolutionResults:
         self.__dict__.update(kwargs)
 
     def __str__(self) -> str:
-        """
-        Return a formatted string representation of the SolutionResults object.
+        """Return a formatted string representation of the SolutionResults object.
 
         Returns:
             (str): A string representation listing non-null attributes.

ultralytics/solutions/speed_estimation.py

@@ -9,12 +9,11 @@ from ultralytics.utils.plotting import colors
 
 
 class SpeedEstimator(BaseSolution):
-    """
-    A class to estimate the speed of objects in a real-time video stream based on their tracks.
+    """A class to estimate the speed of objects in a real-time video stream based on their tracks.
 
-    This class extends the BaseSolution class and provides functionality for estimating object speeds using
-    tracking data in video streams. Speed is calculated based on pixel displacement over time and converted
-    to real-world units using a configurable meters-per-pixel scale factor.
+    This class extends the BaseSolution class and provides functionality for estimating object speeds using tracking
+    data in video streams. Speed is calculated based on pixel displacement over time and converted to real-world units
+    using a configurable meters-per-pixel scale factor.
 
     Attributes:
         fps (float): Video frame rate for time calculations.
@@ -42,8 +41,7 @@ class SpeedEstimator(BaseSolution):
     """
 
     def __init__(self, **kwargs: Any) -> None:
-        """
-        Initialize the SpeedEstimator object with speed estimation parameters and data structures.
+        """Initialize the SpeedEstimator object with speed estimation parameters and data structures.
 
         Args:
             **kwargs (Any): Additional keyword arguments passed to the parent class.
@@ -61,8 +59,7 @@ class SpeedEstimator(BaseSolution):
         self.max_speed = self.CFG["max_speed"]  # Maximum speed adjustment
 
     def process(self, im0) -> SolutionResults:
-        """
-        Process an input frame to estimate object speeds based on tracking data.
+        """Process an input frame to estimate object speeds based on tracking data.
 
         Args:
             im0 (np.ndarray): Input image for processing with shape (H, W, C) for RGB images.

ultralytics/solutions/streamlit_inference.py

@@ -16,8 +16,7 @@ torch.classes.__path__ = []  # Torch module __path__._path issue: https://github
 
 
 class Inference:
-    """
-    A class to perform object detection, image classification, image segmentation and pose estimation inference.
+    """A class to perform object detection, image classification, image segmentation and pose estimation inference.
 
     This class provides functionalities for loading models, configuring settings, uploading video files, and performing
     real-time inference using Streamlit and Ultralytics YOLO models.
@@ -54,8 +53,7 @@ class Inference:
     """
 
     def __init__(self, **kwargs: Any) -> None:
-        """
-        Initialize the Inference class, checking Streamlit requirements and setting up the model path.
+        """Initialize the Inference class, checking Streamlit requirements and setting up the model path.
 
         Args:
             **kwargs (Any): Additional keyword arguments for model configuration.

ultralytics/solutions/trackzone.py

@@ -10,8 +10,7 @@ from ultralytics.utils.plotting import colors
 
 
 class TrackZone(BaseSolution):
-    """
-    A class to manage region-based object tracking in a video stream.
+    """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.
@@ -37,8 +36,7 @@ class TrackZone(BaseSolution):
     """
 
     def __init__(self, **kwargs: Any) -> None:
-        """
-        Initialize the TrackZone class for tracking objects within a defined region in video streams.
+        """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.
@@ -49,18 +47,17 @@ class TrackZone(BaseSolution):
         self.mask = None
 
     def process(self, im0: np.ndarray) -> SolutionResults:
-        """
-        Process the input frame to track objects within a defined region.
+        """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.
+        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.
+            (SolutionResults): Contains processed image `plot_im` and `total_tracks` (int) representing the total number
+                of tracked objects within the defined region.
 
         Examples:
             >>> tracker = TrackZone()

ultralytics/solutions/vision_eye.py

@@ -7,11 +7,10 @@ from ultralytics.utils.plotting import colors
 
 
 class VisionEye(BaseSolution):
-    """
-    A class to manage object detection and vision mapping in images or video streams.
+    """A class to manage object detection and vision mapping in images or video streams.
 
-    This class extends the BaseSolution class and provides functionality for detecting objects,
-    mapping vision points, and annotating results with bounding boxes and labels.
+    This class extends the BaseSolution class and provides functionality for detecting objects, 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.
@@ -27,8 +26,7 @@ class VisionEye(BaseSolution):
     """
 
     def __init__(self, **kwargs: Any) -> None:
-        """
-        Initialize the VisionEye class for detecting objects and applying vision mapping.
+        """Initialize the VisionEye class for detecting objects and applying vision mapping.
 
         Args:
             **kwargs (Any): Keyword arguments passed to the parent class and for configuring vision_point.
@@ -38,8 +36,7 @@ class VisionEye(BaseSolution):
         self.vision_point = self.CFG["vision_point"]
 
     def process(self, im0) -> SolutionResults:
-        """
-        Perform object detection, vision mapping, and annotation on the input image.
+        """Perform object detection, vision mapping, and annotation on the input image.
 
         Args:
             im0 (np.ndarray): The input image for detection and annotation.

ultralytics/trackers/basetrack.py

@@ -8,8 +8,7 @@ import numpy as np
 
 
 class TrackState:
-    """
-    Enumeration class representing the possible states of an object being tracked.
+    """Enumeration class representing the possible states of an object being tracked.
 
     Attributes:
         New (int): State when the object is newly detected.
@@ -30,8 +29,7 @@ class TrackState:
 
 
 class BaseTrack:
-    """
-    Base class for object tracking, providing foundational attributes and methods.
+    """Base class for object tracking, providing foundational attributes and methods.
 
     Attributes:
         _count (int): Class-level counter for unique track IDs.

ultralytics/trackers/bot_sort.py

@@ -19,8 +19,7 @@ from .utils.kalman_filter import KalmanFilterXYWH
 
 
 class BOTrack(STrack):
-    """
-    An extended version of the STrack class for YOLO, adding object tracking features.
+    """An extended version of the STrack class for YOLO, adding object tracking features.
 
     This class extends the STrack class to include additional functionalities for object tracking, such as feature
     smoothing, Kalman filter prediction, and reactivation of tracks.
@@ -57,8 +56,7 @@ class BOTrack(STrack):
     def __init__(
         self, xywh: np.ndarray, score: float, cls: int, feat: np.ndarray | None = None, feat_history: int = 50
     ):
-        """
-        Initialize a BOTrack object with temporal parameters, such as feature history, alpha, and current features.
+        """Initialize a BOTrack object with temporal parameters, such as feature history, alpha, and current features.
 
         Args:
             xywh (np.ndarray): Bounding box coordinates in xywh format (center x, center y, width, height).
@@ -154,8 +152,7 @@ class BOTrack(STrack):
 
 
 class BOTSORT(BYTETracker):
-    """
-    An extended version of the BYTETracker class for YOLO, designed for object tracking with ReID and GMC algorithm.
+    """An extended version of the BYTETracker class for YOLO, designed for object tracking with ReID and GMC algorithm.
 
     Attributes:
         proximity_thresh (float): Threshold for spatial proximity (IoU) between tracks and detections.
@@ -177,13 +174,12 @@ class BOTSORT(BYTETracker):
         >>> bot_sort.init_track(dets, scores, cls, img)
         >>> bot_sort.multi_predict(tracks)
 
-    Note:
+    Notes:
         The class is designed to work with a YOLO object detection model and supports ReID only if enabled via args.
     """
 
     def __init__(self, args: Any, frame_rate: int = 30):
-        """
-        Initialize BOTSORT object with ReID module and GMC algorithm.
+        """Initialize BOTSORT object with ReID module and GMC algorithm.
 
         Args:
             args (Any): Parsed command-line arguments containing tracking parameters.
@@ -253,8 +249,7 @@ class ReID:
     """YOLO model as encoder for re-identification."""
 
     def __init__(self, model: str):
-        """
-        Initialize encoder for re-identification.
+        """Initialize encoder for re-identification.
 
         Args:
             model (str): Path to the YOLO model for re-identification.