Loop through the registered actions and fire all callbacks Args: hook The name of the hook to check, defaults to all

save the metrics to the logging dictionary arguments: log_dict (Dict) -- metrics/media to be logged in current step

Creates or loads a YOLOv5 model Arguments: name (str): model name 'yolov5s' or path 'path/to/best.pt' pretrained (bool): load pre

Returns True if this track is confirmed.

Runs Non-Maximum Suppression (NMS) on inference results Returns: list of detections, on (n,6) tensor per image [xyxy, conf, cls]

Perform Kalman filter measurement update step and update the feature cache. Parameters ---------- kf : kalman_filter.

Return intersection-over-union (Jaccard index) of boxes. Both sets of boxes are expected to be in (x1, y1, x2, y2) format. Arguments:

Args: frame_id (int): Returns: bool: true if frame_id is recognized

Get current position in bounding box format `(top left x, top left y, width, height)`. Returns ------- ndarray

Convert bounding box to format `(center x, center y, aspect ratio, height)`, where the aspect ratio is `width / height`.

Decorator to make all processes in distributed training wait for each local_master to do something.

source - https://gist.github.com/simon-weber/7853144 A context manager that will prevent any logging messages triggered during the body from bein

Args: frame_id: bbox_id: Returns: bool: if bbox exists in frame bboxes list

Returns the intersection over box2 area given box1, box2. Boxes are x1y1x2y2 box1: np.array of shape(4) box2: np.array of shape(n

Calculates the relative bounding box from absolute pixel values.

Check if the dataset format is compatible and upload it as W&B artifact arguments: opt (namespace)-- Commandline arguments f

Simple function that adds fixed color depending on the class

Create and return W&B artifact containing W&B Table of the dataset. arguments: dataset -- instance of LoadImagesAndLabels cl

download the model checkpoint artifact if the path starts with WANDB_ARTIFACT_PREFIX arguments: path -- path of the dataset

Transpose a PIL image accordingly if it has an EXIF Orientation tag. Inplace version of https://github.com/python-pillow/Pillow/blob/master/s