Check if a string is composed of only ASCII characters. Args: s (str): String to be checked. Returns: bool: True if the

Check if a string is composed of only ASCII characters. Args: s (str): String to be checked. Returns: bool: True if the

Check internet connectivity by attempting to connect to a known online host. Returns: (bool): True if connection is successful, Fals

Check internet connectivity by attempting to connect to a known online host. Returns: (bool): True if connection is successful, Fals

r"""Heuristic to determine if a slurm job is a batch job or not. Batch jobs will have a job name that is not a shell unless the user specifically

r"""Heuristic to determine if a slurm job is a batch job or not. Batch jobs will have a job name that is not a shell unless the user specifically

r"""Heuristic to determine if a slurm job is a batch job or not. Batch jobs will have a job name that is not a shell unless the user specifically

Combine two lists of stracks into a single one.

Combine two lists of stracks into a single one.

Returns a list of keys for accessing specific metrics.

Returns a list of keys for accessing specific metrics.

Transfers parameters with matching names and shapes from 'weights' to model.

Transfers parameters with matching names and shapes from 'weights' to model.

Loads 1 image from dataset index 'i', returns (im, original hw, resized hw).

Loads 1 image from dataset index 'i', returns (im, original hw, resized hw).

Normalize bounding boxes, segments, and keypoints to image dimensions.

Normalize bounding boxes, segments, and keypoints to image dimensions.

Convert model output to target format [batch_id, class_id, x, y, w, h, conf] for plotting.

Convert model output to target format [batch_id, class_id, x, y, w, h, conf] for plotting.

Plots the detection results on an input RGB image. Accepts a numpy array (cv2) or a PIL Image. Args: conf (bool): Whethe

Plots the detection results on an input RGB image. Accepts a numpy array (cv2) or a PIL Image. Args: conf (bool): Whethe

Plots a metric-confidence curve.

Plots a metric-confidence curve.

Pre-tranform input image before inference. Args: im (List(np.ndarray)): (N, 3, h, w) for tensor, [(h, w, 3) x N] for list.

Pre-tranform input image before inference. Args: im (List(np.ndarray)): (N, 3, h, w) for tensor, [(h, w, 3) x N] for list.

Perform a forward pass through the network. Args: x (torch.Tensor): The input tensor to the model. profile (

Perform a forward pass through the network. Args: x (torch.Tensor): The input tensor to the model. profile (

Predict masks for the given input prompts, using the currently set image. Input prompts are batched torch tensors and are expected to

Predict masks for the given input prompts, using the currently set image. Input prompts are batched torch tensors and are expected to

Preprocesses the target counts and matches with the input batch size to output a tensor.

Preprocesses the target counts and matches with the input batch size to output a tensor.

Process predicted results for object detection and update metrics.

Process predicted results for object detection and update metrics.

Resets the currently set image.

Resets the currently set image.

this might be similar with denormalize func but without normalized sign.

this might be similar with denormalize func but without normalized sign.

load/create/download model for any task.

Set up YOLO model with specified thresholds and device.

load/create/download model for any task.

Streams real-time inference on camera feed and saves results to file.

Streams real-time inference on camera feed and saves results to file.

Make sure we are sharing the input and output embeddings. Export to TorchScript can't handle parameter sharing so we are cloning them ins

Make sure we are sharing the input and output embeddings. Export to TorchScript can't handle parameter sharing so we are cloning them ins

Make sure we are sharing the input and output embeddings. Export to TorchScript can't handle parameter sharing so we are cloning them ins

Update features vector and smooth it using exponential moving average.

Update features vector and smooth it using exponential moving average.

Initialize instance variables and check for valid input.

Initialize ToTensor class for YOLOv5 image preprocessing.

Initialize instance variables and check for valid input.

Initialize ToTensor class for YOLOv5 image preprocessing.

Return a human-readable string representation of the object.

Return a human-readable string representation of the object.

Add metadata to *.tflite models per https://www.tensorflow.org/lite/models/convert/metadata.

Add metadata to *.tflite models per https://www.tensorflow.org/lite/models/convert/metadata.

Return labels with mosaic border instances clipped.

Return labels with mosaic border instances clipped.

r"""Evaluates a single checkpoint. Args: checkpoint_path: path of checkpoint writer: tensorboard writer object

r"""Evaluates a single checkpoint. Args: checkpoint_path: path of checkpoint writer: tensorboard writer object

r"""Evaluates a single checkpoint. Args: checkpoint_path: path of checkpoint writer: tensorboard writer object

Get losses

Get losses