dgenerate-ultralytics-headless 8.3.141__py3-none-any.whl → 8.3.144__py3-none-any.whl

ultralytics/data/augment.py

@@ -31,10 +31,10 @@ class BaseTransform:
     compatible with both classification and semantic segmentation tasks.
 
     Methods:
-        apply_image: Applies image transformations to labels.
-        apply_instances: Applies transformations to object instances in labels.
-        apply_semantic: Applies semantic segmentation to an image.
-        __call__: Applies all label transformations to an image, instances, and semantic masks.
+        apply_image: Apply image transformations to labels.
+        apply_instances: Apply transformations to object instances in labels.
+        apply_semantic: Apply semantic segmentation to an image.
+        __call__: Apply all label transformations to an image, instances, and semantic masks.
 
     Examples:
         >>> transform = BaseTransform()
@@ -44,7 +44,7 @@ class BaseTransform:
 
     def __init__(self) -> None:
         """
-        Initializes the BaseTransform object.
+        Initialize the BaseTransform object.
 
         This constructor sets up the base transformation object, which can be extended for specific image
         processing tasks. It is designed to be compatible with both classification and semantic segmentation.
@@ -56,7 +56,7 @@ class BaseTransform:
 
     def apply_image(self, labels):
         """
-        Applies image transformations to labels.
+        Apply image transformations to labels.
 
         This method is intended to be overridden by subclasses to implement specific image transformation
         logic. In its base form, it returns the input labels unchanged.
@@ -79,7 +79,7 @@ class BaseTransform:
 
     def apply_instances(self, labels):
         """
-        Applies transformations to object instances in labels.
+        Apply transformations to object instances in labels.
 
         This method is responsible for applying various transformations to object instances within the given
         labels. It is designed to be overridden by subclasses to implement specific instance transformation
@@ -100,7 +100,7 @@ class BaseTransform:
 
     def apply_semantic(self, labels):
         """
-        Applies semantic segmentation transformations to an image.
+        Apply semantic segmentation transformations to an image.
 
         This method is intended to be overridden by subclasses to implement specific semantic segmentation
         transformations. In its base form, it does not perform any operations.
@@ -120,7 +120,7 @@ class BaseTransform:
 
     def __call__(self, labels):
         """
-        Applies all label transformations to an image, instances, and semantic masks.
+        Apply all label transformations to an image, instances, and semantic masks.
 
         This method orchestrates the application of various transformations defined in the BaseTransform class
         to the input labels. It sequentially calls the apply_image and apply_instances methods to process the
@@ -151,12 +151,12 @@ class Compose:
         transforms (List[Callable]): A list of transformation functions to be applied sequentially.
 
     Methods:
-        __call__: Applies a series of transformations to input data.
-        append: Appends a new transform to the existing list of transforms.
-        insert: Inserts a new transform at a specified index in the list of transforms.
-        __getitem__: Retrieves a specific transform or a set of transforms using indexing.
-        __setitem__: Sets a specific transform or a set of transforms using indexing.
-        tolist: Converts the list of transforms to a standard Python list.
+        __call__: Apply a series of transformations to input data.
+        append: Append a new transform to the existing list of transforms.
+        insert: Insert a new transform at a specified index in the list of transforms.
+        __getitem__: Retrieve a specific transform or a set of transforms using indexing.
+        __setitem__: Set a specific transform or a set of transforms using indexing.
+        tolist: Convert the list of transforms to a standard Python list.
 
     Examples:
         >>> transforms = [RandomFlip(), RandomPerspective(30)]
@@ -168,7 +168,7 @@ class Compose:
 
     def __init__(self, transforms):
         """
-        Initializes the Compose object with a list of transforms.
+        Initialize the Compose object with a list of transforms.
 
         Args:
             transforms (List[Callable]): A list of callable transform objects to be applied sequentially.
@@ -182,8 +182,9 @@ class Compose:
 
     def __call__(self, data):
         """
-        Applies a series of transformations to input data. This method sequentially applies each transformation in the
-        Compose object's list of transforms to the input data.
+        Apply a series of transformations to input data.
+
+        This method sequentially applies each transformation in the Compose object's transforms to the input data.
 
         Args:
             data (Any): The input data to be transformed. This can be of any type, depending on the
@@ -203,7 +204,7 @@ class Compose:
 
     def append(self, transform):
         """
-        Appends a new transform to the existing list of transforms.
+        Append a new transform to the existing list of transforms.
 
         Args:
             transform (BaseTransform): The transformation to be added to the composition.
@@ -216,7 +217,7 @@ class Compose:
 
     def insert(self, index, transform):
         """
-        Inserts a new transform at a specified index in the existing list of transforms.
+        Insert a new transform at a specified index in the existing list of transforms.
 
         Args:
             index (int): The index at which to insert the new transform.
@@ -232,7 +233,7 @@ class Compose:
 
     def __getitem__(self, index: Union[list, int]) -> "Compose":
         """
-        Retrieves a specific transform or a set of transforms using indexing.
+        Retrieve a specific transform or a set of transforms using indexing.
 
         Args:
             index (int | List[int]): Index or list of indices of the transforms to retrieve.
@@ -255,7 +256,7 @@ class Compose:
 
     def __setitem__(self, index: Union[list, int], value: Union[list, int]) -> None:
         """
-        Sets one or more transforms in the composition using indexing.
+        Set one or more transforms in the composition using indexing.
 
         Args:
             index (int | List[int]): Index or list of indices to set transforms at.
@@ -282,7 +283,7 @@ class Compose:
 
     def tolist(self):
         """
-        Converts the list of transforms to a standard Python list.
+        Convert the list of transforms to a standard Python list.
 
         Returns:
             (list): A list containing all the transform objects in the Compose instance.
@@ -298,7 +299,7 @@ class Compose:
 
     def __repr__(self):
         """
-        Returns a string representation of the Compose object.
+        Return a string representation of the Compose object.
 
         Returns:
             (str): A string representation of the Compose object, including the list of transforms.
@@ -328,10 +329,10 @@ class BaseMixTransform:
         p (float): Probability of applying the mix transformation.
 
     Methods:
-        __call__: Applies the mix transformation to the input labels.
+        __call__: Apply the mix transformation to the input labels.
         _mix_transform: Abstract method to be implemented by subclasses for specific mix operations.
         get_indexes: Abstract method to get indexes of images to be mixed.
-        _update_label_text: Updates label text for mixed images.
+        _update_label_text: Update label text for mixed images.
 
     Examples:
         >>> class CustomMixTransform(BaseMixTransform):
@@ -348,7 +349,7 @@ class BaseMixTransform:
 
     def __init__(self, dataset, pre_transform=None, p=0.0) -> None:
         """
-        Initializes the BaseMixTransform object for mix transformations like CutMix, MixUp and Mosaic.
+        Initialize the BaseMixTransform object for mix transformations like CutMix, MixUp and Mosaic.
 
         This class serves as a base for implementing mix transformations in image processing pipelines.
 
@@ -368,7 +369,7 @@ class BaseMixTransform:
 
     def __call__(self, labels):
         """
-        Applies pre-processing transforms and cutmix/mixup/mosaic transforms to labels data.
+        Apply pre-processing transforms and cutmix/mixup/mosaic transforms to labels data.
 
         This method determines whether to apply the mix transform based on a probability factor. If applied, it
         selects additional images, applies pre-transforms if specified, and then performs the mix transform.
@@ -408,7 +409,7 @@ class BaseMixTransform:
 
     def _mix_transform(self, labels):
         """
-        Applies CutMix, MixUp or Mosaic augmentation to the label dictionary.
+        Apply CutMix, MixUp or Mosaic augmentation to the label dictionary.
 
         This method should be implemented by subclasses to perform specific mix transformations like CutMix, MixUp or
         Mosaic. It modifies the input label dictionary in-place with the augmented data.
@@ -429,7 +430,7 @@ class BaseMixTransform:
 
     def get_indexes(self):
         """
-        Gets a list of shuffled indexes for mosaic augmentation.
+        Get a list of shuffled indexes for mosaic augmentation.
 
         Returns:
             (List[int]): A list of shuffled indexes from the dataset.
@@ -444,7 +445,7 @@ class BaseMixTransform:
     @staticmethod
     def _update_label_text(labels):
         """
-        Updates label text and class IDs for mixed labels in image augmentation.
+        Update label text and class IDs for mixed labels in image augmentation.
 
         This method processes the 'texts' and 'cls' fields of the input labels dictionary and any mixed labels,
         creating a unified set of text labels and updating class IDs accordingly.
@@ -502,13 +503,13 @@ class Mosaic(BaseMixTransform):
         border (Tuple[int, int]): Border size for width and height.
 
     Methods:
-        get_indexes: Returns a list of random indexes from the dataset.
-        _mix_transform: Applies mixup transformation to the input image and labels.
-        _mosaic3: Creates a 1x3 image mosaic.
-        _mosaic4: Creates a 2x2 image mosaic.
-        _mosaic9: Creates a 3x3 image mosaic.
-        _update_labels: Updates labels with padding.
-        _cat_labels: Concatenates labels and clips mosaic border instances.
+        get_indexes: Return a list of random indexes from the dataset.
+        _mix_transform: Apply mixup transformation to the input image and labels.
+        _mosaic3: Create a 1x3 image mosaic.
+        _mosaic4: Create a 2x2 image mosaic.
+        _mosaic9: Create a 3x3 image mosaic.
+        _update_labels: Update labels with padding.
+        _cat_labels: Concatenate labels and clips mosaic border instances.
 
     Examples:
         >>> from ultralytics.data.augment import Mosaic
@@ -519,7 +520,7 @@ class Mosaic(BaseMixTransform):
 
     def __init__(self, dataset, imgsz=640, p=1.0, n=4):
         """
-        Initializes the Mosaic augmentation object.
+        Initialize the Mosaic augmentation object.
 
         This class performs mosaic augmentation by combining multiple (4 or 9) images into a single mosaic image.
         The augmentation is applied to a dataset with a given probability.
@@ -545,7 +546,7 @@ class Mosaic(BaseMixTransform):
 
     def get_indexes(self):
         """
-        Returns a list of random indexes from the dataset for mosaic augmentation.
+        Return a list of random indexes from the dataset for mosaic augmentation.
 
         This method selects random image indexes either from a buffer or from the entire dataset, depending on
         the 'buffer' parameter. It is used to choose images for creating mosaic augmentations.
@@ -566,7 +567,7 @@ class Mosaic(BaseMixTransform):
 
     def _mix_transform(self, labels):
         """
-        Applies mosaic augmentation to the input image and labels.
+        Apply mosaic augmentation to the input image and labels.
 
         This method combines multiple images (3, 4, or 9) into a single mosaic image based on the 'n' attribute.
         It ensures that rectangular annotations are not present and that there are other images available for
@@ -595,7 +596,7 @@ class Mosaic(BaseMixTransform):
 
     def _mosaic3(self, labels):
         """
-        Creates a 1x3 image mosaic by combining three images.
+        Create a 1x3 image mosaic by combining three images.
 
         This method arranges three images in a horizontal layout, with the main image in the center and two
         additional images on either side. It's part of the Mosaic augmentation technique used in object detection.
@@ -654,7 +655,7 @@ class Mosaic(BaseMixTransform):
 
     def _mosaic4(self, labels):
         """
-        Creates a 2x2 image mosaic from four input images.
+        Create a 2x2 image mosaic from four input images.
 
         This method combines four images into a single mosaic image by placing them in a 2x2 grid. It also
         updates the corresponding labels for each image in the mosaic.
@@ -712,7 +713,7 @@ class Mosaic(BaseMixTransform):
 
     def _mosaic9(self, labels):
         """
-        Creates a 3x3 image mosaic from the input image and eight additional images.
+        Create a 3x3 image mosaic from the input image and eight additional images.
 
         This method combines nine images into a single mosaic image. The input image is placed at the center,
         and eight additional images from the dataset are placed around it in a 3x3 grid pattern.
@@ -785,7 +786,7 @@ class Mosaic(BaseMixTransform):
     @staticmethod
     def _update_labels(labels, padw, padh):
         """
-        Updates label coordinates with padding values.
+        Update label coordinates with padding values.
 
         This method adjusts the bounding box coordinates of object instances in the labels by adding padding
         values. It also denormalizes the coordinates if they were previously normalized.
@@ -811,7 +812,7 @@ class Mosaic(BaseMixTransform):
 
     def _cat_labels(self, mosaic_labels):
         """
-        Concatenates and processes labels for mosaic augmentation.
+        Concatenate and process labels for mosaic augmentation.
 
         This method combines labels from multiple images used in mosaic augmentation, clips instances to the
         mosaic border, and removes zero-area boxes.
@@ -863,7 +864,7 @@ class Mosaic(BaseMixTransform):
 
 class MixUp(BaseMixTransform):
     """
-    Applies MixUp augmentation to image datasets.
+    Apply MixUp augmentation to image datasets.
 
     This class implements the MixUp augmentation technique as described in the paper [mixup: Beyond Empirical Risk
     Minimization](https://arxiv.org/abs/1710.09412). MixUp combines two images and their labels using a random weight.
@@ -874,7 +875,7 @@ class MixUp(BaseMixTransform):
         p (float): Probability of applying MixUp augmentation.
 
     Methods:
-        _mix_transform: Applies MixUp augmentation to the input labels.
+        _mix_transform: Apply MixUp augmentation to the input labels.
 
     Examples:
         >>> from ultralytics.data.augment import MixUp
@@ -885,7 +886,7 @@ class MixUp(BaseMixTransform):
 
     def __init__(self, dataset, pre_transform=None, p=0.0) -> None:
         """
-        Initializes the MixUp augmentation object.
+        Initialize the MixUp augmentation object.
 
         MixUp is an image augmentation technique that combines two images by taking a weighted sum of their pixel
         values and labels. This implementation is designed for use with the Ultralytics YOLO framework.
@@ -904,7 +905,7 @@ class MixUp(BaseMixTransform):
 
     def _mix_transform(self, labels):
         """
-        Applies MixUp augmentation to the input labels.
+        Apply MixUp augmentation to the input labels.
 
         This method implements the MixUp augmentation technique as described in the paper
         "mixup: Beyond Empirical Risk Minimization" (https://arxiv.org/abs/1710.09412).
@@ -929,7 +930,7 @@ class MixUp(BaseMixTransform):
 
 class CutMix(BaseMixTransform):
     """
-    Applies CutMix augmentation to image datasets as described in the paper https://arxiv.org/abs/1905.04899.
+    Apply CutMix augmentation to image datasets as described in the paper https://arxiv.org/abs/1905.04899.
 
     CutMix combines two images by replacing a random rectangular region of one image with the corresponding region from another image,
     and adjusts the labels proportionally to the area of the mixed region.
@@ -938,12 +939,12 @@ class CutMix(BaseMixTransform):
         dataset (Any): The dataset to which CutMix augmentation will be applied.
         pre_transform (Callable | None): Optional transform to apply before CutMix.
         p (float): Probability of applying CutMix augmentation.
-        beta (float): Beta distribution parameter for sampling the mixing ratio (default=1.0).
-        num_areas (int): Number of areas to try to cut and mix (default=3).
+        beta (float): Beta distribution parameter for sampling the mixing ratio.
+        num_areas (int): Number of areas to try to cut and mix.
 
     Methods:
-        _mix_transform: Applies CutMix augmentation to the input labels.
-        _rand_bbox: Generates random bounding box coordinates for the cut region.
+        _mix_transform: Apply CutMix augmentation to the input labels.
+        _rand_bbox: Generate random bounding box coordinates for the cut region.
 
     Examples:
         >>> from ultralytics.data.augment import CutMix
@@ -954,14 +955,14 @@ class CutMix(BaseMixTransform):
 
     def __init__(self, dataset, pre_transform=None, p=0.0, beta=1.0, num_areas=3) -> None:
         """
-        Initializes the CutMix augmentation object.
+        Initialize the CutMix augmentation object.
 
         Args:
             dataset (Any): The dataset to which CutMix augmentation will be applied.
             pre_transform (Callable | None): Optional transform to apply before CutMix.
             p (float): Probability of applying CutMix augmentation.
-            beta (float): Beta distribution parameter for sampling the mixing ratio (default=1.0).
-            num_areas (int): Number of areas to try to cut and mix (default=3).
+            beta (float): Beta distribution parameter for sampling the mixing ratio.
+            num_areas (int): Number of areas to try to cut and mix.
         """
         super().__init__(dataset=dataset, pre_transform=pre_transform, p=p)
         self.beta = beta
@@ -969,7 +970,7 @@ class CutMix(BaseMixTransform):
 
     def _rand_bbox(self, width, height):
         """
-        Generates random bounding box coordinates for the cut region.
+        Generate random bounding box coordinates for the cut region.
 
         Args:
             width (int): Width of the image.
@@ -999,7 +1000,7 @@ class CutMix(BaseMixTransform):
 
     def _mix_transform(self, labels):
         """
-        Applies CutMix augmentation to the input labels.
+        Apply CutMix augmentation to the input labels.
 
         Args:
             labels (dict): A dictionary containing the original image and label information.
@@ -1021,7 +1022,7 @@ class CutMix(BaseMixTransform):
             return labels
 
         labels2 = labels.pop("mix_labels")[0]
-        area = cut_areas[np.random.choice(idx)]  # randomle select one
+        area = cut_areas[np.random.choice(idx)]  # randomly select one
         ioa2 = bbox_ioa(area[None], labels2["instances"].bboxes).squeeze(0)
         indexes2 = np.nonzero(ioa2 >= (0.01 if len(labels["instances"].segments) else 0.1))[0]
         if len(indexes2) == 0:
@@ -1047,7 +1048,7 @@ class CutMix(BaseMixTransform):
 
 class RandomPerspective:
     """
-    Implements random perspective and affine transformations on images and corresponding annotations.
+    Implement random perspective and affine transformations on images and corresponding annotations.
 
     This class applies random rotations, translations, scaling, shearing, and perspective transformations
     to images and their associated bounding boxes, segments, and keypoints. It can be used as part of an
@@ -1063,12 +1064,12 @@ class RandomPerspective:
         pre_transform (Callable | None): Optional transform to apply before the random perspective.
 
     Methods:
-        affine_transform: Applies affine transformations to the input image.
-        apply_bboxes: Transforms bounding boxes using the affine matrix.
-        apply_segments: Transforms segments and generates new bounding boxes.
-        apply_keypoints: Transforms keypoints using the affine matrix.
-        __call__: Applies the random perspective transformation to images and annotations.
-        box_candidates: Filters transformed bounding boxes based on size and aspect ratio.
+        affine_transform: Apply affine transformations to the input image.
+        apply_bboxes: Transform bounding boxes using the affine matrix.
+        apply_segments: Transform segments and generate new bounding boxes.
+        apply_keypoints: Transform keypoints using the affine matrix.
+        __call__: Apply the random perspective transformation to images and annotations.
+        box_candidates: Filter transformed bounding boxes based on size and aspect ratio.
 
     Examples:
         >>> transform = RandomPerspective(degrees=10, translate=0.1, scale=0.1, shear=10)
@@ -1083,7 +1084,7 @@ class RandomPerspective:
         self, degrees=0.0, translate=0.1, scale=0.5, shear=0.0, perspective=0.0, border=(0, 0), pre_transform=None
     ):
         """
-        Initializes RandomPerspective object with transformation parameters.
+        Initialize RandomPerspective object with transformation parameters.
 
         This class implements random perspective and affine transformations on images and corresponding bounding boxes,
         segments, and keypoints. Transformations include rotation, translation, scaling, and shearing.
@@ -1112,7 +1113,7 @@ class RandomPerspective:
 
     def affine_transform(self, img, border):
         """
-        Applies a sequence of affine transformations centered around the image center.
+        Apply a sequence of affine transformations centered around the image center.
 
         This function performs a series of geometric transformations on the input image, including
         translation, perspective change, rotation, scaling, and shearing. The transformations are
@@ -1246,7 +1247,7 @@ class RandomPerspective:
 
     def apply_keypoints(self, keypoints, M):
         """
-        Applies affine transformation to keypoints.
+        Apply affine transformation to keypoints.
 
         This method transforms the input keypoints using the provided affine transformation matrix. It handles
         perspective rescaling if necessary and updates the visibility of keypoints that fall outside the image
@@ -1280,7 +1281,7 @@ class RandomPerspective:
 
     def __call__(self, labels):
         """
-        Applies random perspective and affine transformations to an image and its associated labels.
+        Apply random perspective and affine transformations to an image and its associated labels.
 
         This method performs a series of transformations including rotation, translation, scaling, shearing,
         and perspective distortion on the input image and adjusts the corresponding bounding boxes, segments,
@@ -1398,7 +1399,7 @@ class RandomPerspective:
 
 class RandomHSV:
     """
-    Randomly adjusts the Hue, Saturation, and Value (HSV) channels of an image.
+    Randomly adjust the Hue, Saturation, and Value (HSV) channels of an image.
 
     This class applies random HSV augmentation to images within predefined limits set by hgain, sgain, and vgain.
 
@@ -1408,7 +1409,7 @@ class RandomHSV:
         vgain (float): Maximum variation for value. Range is typically [0, 1].
 
     Methods:
-        __call__: Applies random HSV augmentation to an image.
+        __call__: Apply random HSV augmentation to an image.
 
     Examples:
         >>> import numpy as np
@@ -1422,7 +1423,7 @@ class RandomHSV:
 
     def __init__(self, hgain=0.5, sgain=0.5, vgain=0.5) -> None:
         """
-        Initializes the RandomHSV object for random HSV (Hue, Saturation, Value) augmentation.
+        Initialize the RandomHSV object for random HSV (Hue, Saturation, Value) augmentation.
 
         This class applies random adjustments to the HSV channels of an image within specified limits.
 
@@ -1441,7 +1442,7 @@ class RandomHSV:
 
     def __call__(self, labels):
         """
-        Applies random HSV augmentation to an image within predefined limits.
+        Apply random HSV augmentation to an image within predefined limits.
 
         This method modifies the input image by randomly adjusting its Hue, Saturation, and Value (HSV) channels.
         The adjustments are made within the limits set by hgain, sgain, and vgain during initialization.
@@ -1482,7 +1483,7 @@ class RandomHSV:
 
 class RandomFlip:
     """
-    Applies a random horizontal or vertical flip to an image with a given probability.
+    Apply a random horizontal or vertical flip to an image with a given probability.
 
     This class performs random image flipping and updates corresponding instance annotations such as
     bounding boxes and keypoints.
@@ -1493,7 +1494,7 @@ class RandomFlip:
         flip_idx (array-like): Index mapping for flipping keypoints, if applicable.
 
     Methods:
-        __call__: Applies the random flip transformation to an image and its annotations.
+        __call__: Apply the random flip transformation to an image and its annotations.
 
     Examples:
         >>> transform = RandomFlip(p=0.5, direction="horizontal")
@@ -1504,7 +1505,7 @@ class RandomFlip:
 
     def __init__(self, p=0.5, direction="horizontal", flip_idx=None) -> None:
         """
-        Initializes the RandomFlip class with probability and direction.
+        Initialize the RandomFlip class with probability and direction.
 
         This class applies a random horizontal or vertical flip to an image with a given probability.
         It also updates any instances (bounding boxes, keypoints, etc.) accordingly.
@@ -1530,7 +1531,7 @@ class RandomFlip:
 
     def __call__(self, labels):
         """
-        Applies random flip to an image and updates any instances like bounding boxes or keypoints accordingly.
+        Apply random flip to an image and update any instances like bounding boxes or keypoints accordingly.
 
         This method randomly flips the input image either horizontally or vertically based on the initialized
         probability and direction. It also updates the corresponding instances (bounding boxes, keypoints) to
@@ -1634,7 +1635,7 @@ class LetterBox:
 
     def __call__(self, labels=None, image=None):
         """
-        Resizes and pads an image for object detection, instance segmentation, or pose estimation tasks.
+        Resize and pad an image for object detection, instance segmentation, or pose estimation tasks.
 
         This method applies letterboxing to the input image, which involves resizing the image while maintaining its
         aspect ratio and adding padding to fit the new shape. It also updates any associated labels accordingly.
@@ -1711,7 +1712,7 @@ class LetterBox:
     @staticmethod
     def _update_labels(labels, ratio, padw, padh):
         """
-        Updates labels after applying letterboxing to an image.
+        Update labels after applying letterboxing to an image.
 
         This method modifies the bounding box coordinates of instances in the labels
         to account for resizing and padding applied during letterboxing.
@@ -1753,8 +1754,8 @@ class CopyPaste(BaseMixTransform):
         p (float): Probability of applying Copy-Paste augmentation.
 
     Methods:
-        _mix_transform: Applies Copy-Paste augmentation to the input labels.
-        __call__: Applies the Copy-Paste transformation to images and annotations.
+        _mix_transform: Apply Copy-Paste augmentation to the input labels.
+        __call__: Apply the Copy-Paste transformation to images and annotations.
 
     Examples:
         >>> from ultralytics.data.augment import CopyPaste
@@ -1764,18 +1765,18 @@ class CopyPaste(BaseMixTransform):
     """
 
     def __init__(self, dataset=None, pre_transform=None, p=0.5, mode="flip") -> None:
-        """Initializes CopyPaste object with dataset, pre_transform, and probability of applying MixUp."""
+        """Initialize CopyPaste object with dataset, pre_transform, and probability of applying MixUp."""
         super().__init__(dataset=dataset, pre_transform=pre_transform, p=p)
         assert mode in {"flip", "mixup"}, f"Expected `mode` to be `flip` or `mixup`, but got {mode}."
         self.mode = mode
 
     def _mix_transform(self, labels):
-        """Applies Copy-Paste augmentation to combine objects from another image into the current image."""
+        """Apply Copy-Paste augmentation to combine objects from another image into the current image."""
         labels2 = labels["mix_labels"][0]
         return self._transform(labels, labels2)
 
     def __call__(self, labels):
-        """Applies Copy-Paste augmentation to an image and its labels."""
+        """Apply Copy-Paste augmentation to an image and its labels."""
         if len(labels["instances"].segments) == 0 or self.p == 0:
             return labels
         if self.mode == "flip":
@@ -1802,7 +1803,7 @@ class CopyPaste(BaseMixTransform):
         return labels
 
     def _transform(self, labels1, labels2={}):
-        """Applies Copy-Paste augmentation to combine objects from another image into the current image."""
+        """Apply Copy-Paste augmentation to combine objects from another image into the current image."""
         im = labels1["img"]
         cls = labels1["cls"]
         h, w = im.shape[:2]
@@ -1851,7 +1852,7 @@ class Albumentations:
         contains_spatial (bool): Indicates if the transforms include spatial operations.
 
     Methods:
-        __call__: Applies the Albumentations transformations to the input labels.
+        __call__: Apply the Albumentations transformations to the input labels.
 
     Examples:
         >>> transform = Albumentations(p=0.5)
@@ -1979,7 +1980,7 @@ class Albumentations:
 
     def __call__(self, labels):
         """
-        Applies Albumentations transformations to input labels.
+        Apply Albumentations transformations to input labels.
 
         This method applies a series of image augmentations using the Albumentations library. It can perform both
         spatial and non-spatial transformations on the input image and its corresponding labels.
@@ -2052,9 +2053,9 @@ class Format:
         bgr (float): The probability to return BGR images.
 
     Methods:
-        __call__: Formats labels dictionary with image, classes, bounding boxes, and optionally masks and keypoints.
-        _format_img: Converts image from Numpy array to PyTorch tensor.
-        _format_segments: Converts polygon points to bitmap masks.
+        __call__: Format labels dictionary with image, classes, bounding boxes, and optionally masks and keypoints.
+        _format_img: Convert image from Numpy array to PyTorch tensor.
+        _format_segments: Convert polygon points to bitmap masks.
 
     Examples:
         >>> formatter = Format(bbox_format="xywh", normalize=True, return_mask=True)
@@ -2077,7 +2078,7 @@ class Format:
         bgr=0.0,
     ):
         """
-        Initializes the Format class with given parameters for image and instance annotation formatting.
+        Initialize the Format class with given parameters for image and instance annotation formatting.
 
         This class standardizes image and instance annotations for object detection, instance segmentation, and pose
         estimation tasks, preparing them for use in PyTorch DataLoader's `collate_fn`.
@@ -2121,7 +2122,7 @@ class Format:
 
     def __call__(self, labels):
         """
-        Formats image annotations for object detection, instance segmentation, and pose estimation tasks.
+        Format image annotations for object detection, instance segmentation, and pose estimation tasks.
 
         This method standardizes the image and instance annotations to be used by the `collate_fn` in PyTorch
         DataLoader. It processes the input labels dictionary, converting annotations to the specified format and
@@ -2188,7 +2189,7 @@ class Format:
 
     def _format_img(self, img):
         """
-        Formats an image for YOLO from a Numpy array to a PyTorch tensor.
+        Format an image for YOLO from a Numpy array to a PyTorch tensor.
 
         This function performs the following operations:
         1. Ensures the image has 3 dimensions (adds a channel dimension if needed).
@@ -2219,7 +2220,7 @@ class Format:
 
     def _format_segments(self, instances, cls, w, h):
         """
-        Converts polygon segments to bitmap masks.
+        Convert polygon segments to bitmap masks.
 
         Args:
             instances (Instances): Object containing segment information.
@@ -2250,7 +2251,7 @@ class Format:
 
 
 class LoadVisualPrompt:
-    """Creates visual prompts from bounding boxes or masks for model input."""
+    """Create visual prompts from bounding boxes or masks for model input."""
 
     def __init__(self, scale_factor=1 / 8):
         """
@@ -2307,8 +2308,8 @@ class LoadVisualPrompt:
         Args:
             category (int | np.ndarray | torch.Tensor): The category labels for the objects.
             shape (tuple): The shape of the image (height, width).
-            bboxes (np.ndarray | torch.Tensor, optional): Bounding boxes for the objects, xyxy format. Defaults to None.
-            masks (np.ndarray | torch.Tensor, optional): Masks for the objects. Defaults to None.
+            bboxes (np.ndarray | torch.Tensor, optional): Bounding boxes for the objects, xyxy format.
+            masks (np.ndarray | torch.Tensor, optional): Masks for the objects.
 
         Returns:
             (torch.Tensor): A tensor containing the visual masks for each category.
@@ -2344,7 +2345,7 @@ class LoadVisualPrompt:
 
 class RandomLoadText:
     """
-    Randomly samples positive and negative texts and updates class indices accordingly.
+    Randomly sample positive and negative texts and update class indices accordingly.
 
     This class is responsible for sampling texts from a given set of class texts, including both positive
     (present in the image) and negative (not present in the image) samples. It updates the class indices
@@ -2358,7 +2359,7 @@ class RandomLoadText:
         padding_value (str): The text used for padding when padding is True.
 
     Methods:
-        __call__: Processes the input labels and returns updated classes and texts.
+        __call__: Process the input labels and return updated classes and texts.
 
     Examples:
         >>> loader = RandomLoadText(prompt_format="Object: {}", neg_samples=(5, 10), max_samples=20)
@@ -2377,21 +2378,21 @@ class RandomLoadText:
         padding_value: List[str] = [""],
     ) -> None:
         """
-        Initializes the RandomLoadText class for randomly sampling positive and negative texts.
+        Initialize the RandomLoadText class for randomly sampling positive and negative texts.
 
         This class is designed to randomly sample positive texts and negative texts, and update the class
         indices accordingly to the number of samples. It can be used for text-based object detection tasks.
 
         Args:
-            prompt_format (str): Format string for the prompt. Default is '{}'. The format string should
+            prompt_format (str): Format string for the prompt. The format string should
                 contain a single pair of curly braces {} where the text will be inserted.
             neg_samples (Tuple[int, int]): A range to randomly sample negative texts. The first integer
                 specifies the minimum number of negative samples, and the second integer specifies the
-                maximum. Default is (80, 80).
-            max_samples (int): The maximum number of different text samples in one image. Default is 80.
+                maximum.
+            max_samples (int): The maximum number of different text samples in one image.
             padding (bool): Whether to pad texts to max_samples. If True, the number of texts will always
-                be equal to max_samples. Default is False.
-            padding_value (str): The padding text to use when padding is True. Default is an empty string.
+                be equal to max_samples.
+            padding_value (str): The padding text to use when padding is True.
 
         Attributes:
             prompt_format (str): The format string for the prompt.
@@ -2417,7 +2418,7 @@ class RandomLoadText:
 
     def __call__(self, labels: dict) -> dict:
         """
-        Randomly samples positive and negative texts and updates class indices accordingly.
+        Randomly sample positive and negative texts and update class indices accordingly.
 
         This method samples positive texts based on the existing class labels in the image, and randomly
         selects negative texts from the remaining classes. It then updates the class indices to match the
@@ -2483,7 +2484,7 @@ class RandomLoadText:
 
 def v8_transforms(dataset, imgsz, hyp, stretch=False):
     """
-    Applies a series of image transformations for training.
+    Apply a series of image transformations for training.
 
     This function creates a composition of image augmentation techniques to prepare images for YOLO training.
     It includes operations such as mosaic, copy-paste, random perspective, mixup, and various color adjustments.
@@ -2558,7 +2559,7 @@ def classify_transforms(
     crop_fraction=None,
 ):
     """
-    Creates a composition of image transforms for classification tasks.
+    Create a composition of image transforms for classification tasks.
 
     This function generates a sequence of torchvision transforms suitable for preprocessing images
     for classification models during evaluation or inference. The transforms include resizing,
@@ -2618,7 +2619,7 @@ def classify_augmentations(
     interpolation="BILINEAR",
 ):
     """
-    Creates a composition of image augmentation transforms for classification tasks.
+    Create a composition of image augmentation transforms for classification tasks.
 
     This function generates a set of image transformations suitable for training classification models. It includes
     options for resizing, flipping, color jittering, auto augmentation, and random erasing.
@@ -2719,7 +2720,7 @@ class ClassifyLetterBox:
         stride (int): The stride value, used when 'auto' is True.
 
     Methods:
-        __call__: Applies the letterbox transformation to an input image.
+        __call__: Apply the letterbox transformation to an input image.
 
     Examples:
         >>> transform = ClassifyLetterBox(size=(640, 640), auto=False, stride=32)
@@ -2731,7 +2732,7 @@ class ClassifyLetterBox:
 
     def __init__(self, size=(640, 640), auto=False, stride=32):
         """
-        Initializes the ClassifyLetterBox object for image preprocessing.
+        Initialize the ClassifyLetterBox object for image preprocessing.
 
         This class is designed to be part of a transformation pipeline for image classification tasks. It resizes and
         pads images to a specified size while maintaining the original aspect ratio.
@@ -2739,8 +2740,8 @@ class ClassifyLetterBox:
         Args:
             size (int | Tuple[int, int]): Target size for the letterboxed image. If an int, a square image of
                 (size, size) is created. If a tuple, it should be (height, width).
-            auto (bool): If True, automatically calculates the short side based on stride. Default is False.
-            stride (int): The stride value, used when 'auto' is True. Default is 32.
+            auto (bool): If True, automatically calculates the short side based on stride.
+            stride (int): The stride value, used when 'auto' is True.
 
         Attributes:
             h (int): Target height of the letterboxed image.
@@ -2762,7 +2763,7 @@ class ClassifyLetterBox:
 
     def __call__(self, im):
         """
-        Resizes and pads an image using the letterbox method.
+        Resize and pad an image using the letterbox method.
 
         This method resizes the input image to fit within the specified dimensions while maintaining its aspect ratio,
         then pads the resized image to match the target size.
@@ -2798,7 +2799,7 @@ class ClassifyLetterBox:
 # NOTE: keep this class for backward compatibility
 class CenterCrop:
     """
-    Applies center cropping to images for classification tasks.
+    Apply center cropping to images for classification tasks.
 
     This class performs center cropping on input images, resizing them to a specified size while maintaining the aspect
     ratio. It is designed to be part of a transformation pipeline, e.g., T.Compose([CenterCrop(size), ToTensor()]).
@@ -2808,7 +2809,7 @@ class CenterCrop:
         w (int): Target width of the cropped image.
 
     Methods:
-        __call__: Applies the center crop transformation to an input image.
+        __call__: Apply the center crop transformation to an input image.
 
     Examples:
         >>> transform = CenterCrop(640)
@@ -2820,7 +2821,7 @@ class CenterCrop:
 
     def __init__(self, size=640):
         """
-        Initializes the CenterCrop object for image preprocessing.
+        Initialize the CenterCrop object for image preprocessing.
 
         This class is designed to be part of a transformation pipeline, e.g., T.Compose([CenterCrop(size), ToTensor()]).
         It performs a center crop on input images to a specified size.
@@ -2844,7 +2845,7 @@ class CenterCrop:
 
     def __call__(self, im):
         """
-        Applies center cropping to an input image.
+        Apply center cropping to an input image.
 
         This method resizes and crops the center of the image using a letterbox method. It maintains the aspect
         ratio of the original image while fitting it into the specified dimensions.
@@ -2873,7 +2874,7 @@ class CenterCrop:
 # NOTE: keep this class for backward compatibility
 class ToTensor:
     """
-    Converts an image from a numpy array to a PyTorch tensor.
+    Convert an image from a numpy array to a PyTorch tensor.
 
     This class is designed to be part of a transformation pipeline, e.g., T.Compose([LetterBox(size), ToTensor()]).
 
@@ -2881,7 +2882,7 @@ class ToTensor:
         half (bool): If True, converts the image to half precision (float16).
 
     Methods:
-        __call__: Applies the tensor conversion to an input image.
+        __call__: Apply the tensor conversion to an input image.
 
     Examples:
         >>> transform = ToTensor(half=True)
@@ -2897,14 +2898,14 @@ class ToTensor:
 
     def __init__(self, half=False):
         """
-        Initializes the ToTensor object for converting images to PyTorch tensors.
+        Initialize the ToTensor object for converting images to PyTorch tensors.
 
         This class is designed to be used as part of a transformation pipeline for image preprocessing in the
         Ultralytics YOLO framework. It converts numpy arrays or PIL Images to PyTorch tensors, with an option
         for half-precision (float16) conversion.
 
         Args:
-            half (bool): If True, converts the tensor to half precision (float16). Default is False.
+            half (bool): If True, converts the tensor to half precision (float16).
 
         Examples:
             >>> transform = ToTensor(half=True)
@@ -2918,7 +2919,7 @@ class ToTensor:
 
     def __call__(self, im):
         """
-        Transforms an image from a numpy array to a PyTorch tensor.
+        Transform an image from a numpy array to a PyTorch tensor.
 
         This method converts the input image from a numpy array to a PyTorch tensor, applying optional
         half-precision conversion and normalization. The image is transposed from HWC to CHW format and