Calculate the intersection over box2 area given box1 and box2. Boxes are in x1y1x2y2 format. Args: box1 (np.array): A numpy array of

Compute the Intersection-Over-Union of a bounding box with respect to an array of other bounding boxes. Args: box1 (torch.Tensor): (

Calculate the Intersection over Union (IoU) between pairs of bounding boxes. Args: box1 (np.array): A numpy array of shape (n, 4) re

Initialize Detect() biases, WARNING: requires stride availability.

initialize conv/fc bias value according to a given probability value.

Generate point grids for all crop layers.

Build YOLO Dataset Args: img_path (str): Path to the folder containing images. mode (str): `train` mode or `val` mode

Build YOLO Dataset. Args: img_path (str): Path to the folder containing images. mode (str): `train` mode or

Constructs an optimizer for the given model, based on the specified optimizer name, learning rate, momentum, weight decay, and number

Generate a 2D grid of evenly spaced points in the range [0,1]x[0,1].

Users can custom augmentations here like: if self.augment: # Training transforms return Compose([]

Builds and appends transforms to the list.

Cache images to memory or disk.

Cache dataset labels, check images and read shapes. Args: path (Path): path where to save the cache file (default: Path('./labels.

Computes the stability score for a batch of masks. The stability score is the IoU between the binary masks obtained by thresholding the p

This function checks the PyTorch Automatic Mixed Precision (AMP) functionality of a YOLOv8 model. If the checks fail, it means there are anom

Check image caching requirements vs available memory.

Check if there is sufficient disk space to download and store a file. Args: url (str, optional): The URL to the file. Defaults to 'h

Returns the latest version of a PyPI package without downloading or installing it. Parameters: package_name (str): The name of the p

Checks if a new version of the ultralytics package is available on PyPI. Returns: (bool): True if an update is available, False othe

Check current python version against the required minimum version. Args: minimum (str): Required minimum version of python. Ret

Check if resume checkpoint exists and update arguments accordingly.

Check source type and return corresponding flag values.

Checks statistics.

Checks the installed versions of PyTorch and Torchvision to ensure they're compatible. This function checks the installed versions of PyTorc

Check YOLO training batch size using the autobatch() function. Args: model (torch.nn.Module): YOLO model to check batch size for.

Return a human-readable YOLO software and hardware summary.

Replace legacy YOLOv5 filenames with updated YOLOv5u filenames.

YOLOv8 classification Albumentations (optional, only used if package is installed).

Clip line coordinates to the image boundaries. Args: coords (torch.Tensor | numpy.ndarray): A list of line coordinates. shap

Converts 91-index COCO class IDs to 80-index COCO class IDs. Returns: (list): A list of 91 class IDs where the index represents the 80-in

Compresses a single image file to reduced size while preserving its aspect ratio and quality using either the Python Imaging Library (PIL) or

Compute the average precision (AP) given the recall and precision curves. Arguments: recall (list): The recall curve. precis

设置不同类别的固定颜色

设置不同类别的固定颜色

设置不同类别的固定颜色

Converts COCO dataset annotations to a format suitable for training YOLOv5 models. Args: labels_dir (str, optional): Path to directory co

Copies attributes from object 'b' to object 'a', with options to include/exclude certain attributes.

Return a copy of the Results object with all tensors on GPU memory.

Delete temp file if created.

Deletes all ".DS_store" files under a specified directory. Args: path (str, optional): The directory path where the ".DS_store" file

This function is the ultralytics package entrypoint, it's responsible for parsing the command line arguments passed to the package. This

Evaluate and return JSON format of prediction statistics.

Returns exif-corrected PIL size.

Export a YOLOv model to a specific format.

YOLOv8 CoreML export.

YOLOv8 Edge TPU export https://coral.ai/docs/edgetpu/models-intro/.

YOLOv8 TensorRT export https://developer.nvidia.com/tensorrt.

YOLOv8 ncnn export using PNNX https://github.com/pnnx/pnnx.

YOLOv8 OpenVINO export.

YOLOv8 Paddle export.

YOLOv8 TensorFlow GraphDef *.pb export https://github.com/leimao/Frozen_Graph_TensorFlow.

YOLOv8 TensorFlow SavedModel export.

YOLOv8 TensorFlow.js export.

YOLOv8 TensorFlow Lite export.

YOLOv8 TorchScript model export.

Performs final evaluation and validation for object detection YOLO model.

Finalizes and returns all metrics.

Finds a free port on localhost. It is useful in single-node training when we don't want to connect to a real main node but have to set the `M

Forward pass of the model on a single scale. Wrapper for `_forward_once` method. Args: x (torch.Tensor | dict):

Runs inference on the YOLOv8 MultiBackend model. Args: im (torch.Tensor): The image tensor to perform inference on.

Fuse Conv2d() and BatchNorm2d() layers https://tehnokv.com/posts/fusing-batchnorm-and-conv/.

Fuse ConvTranspose2d() and BatchNorm2d() layers.

Segment the whole image. Args: im (torch.Tensor): The preprocessed image, (N, C, H, W). crop_n_layers (int): If >0, m

Generates a list of crop boxes of different sizes. Each layer has (2**i)**2 boxes for the ith layer.

Generates and returns command for distributed training.

Generates a DDP file and returns its file name.

Retrieves the URL of the best quality MP4 video stream from a given YouTube video. This function uses the pafy or yt_dlp library to extract

Compute alignment metric given predicted and ground truth bounding boxes.

Get contrastive denoising training group. This function creates a contrastive denoising training group with positive and negative samples fro

Get data loader from dataset path and batch size.

Get train, val path from data dict if it exists. Returns None if data format is not recognized.

Returns a dictionary of default arguments for a function. Args: func (callable): The function to inspect. Returns: (dict): A

Get description of the YOLO model.

Get the match indices for denoising. Args: dn_pos_idx (List[torch.Tensor]): A list includes positive indices of denoising.

Retrieves the origin URL of a git repository. Returns: (str | None): The origin URL of the git repository.