fiftyone.utils.ultralytics

Utilities for working with Ultralytics.

Copyright 2017-2023, Voxel51, Inc.

voxel51.com

Classes:

UltralyticsOutputProcessor([classes])

Converts Ultralytics PyTorch Hub model outputs to FiftyOne format.

Functions:

to_detections(results[, confidence_thresh])	Converts ultralytics.YOLO boxes to FiftyOne format.
to_instances(results[, confidence_thresh])	Converts ultralytics.YOLO instance segmentations to FiftyOne format.
to_keypoints(results[, confidence_thresh])	Converts ultralytics.YOLO keypoints to FiftyOne format.
to_polylines(results[, confidence_thresh, …])	Converts ultralytics.YOLO instance segmentations to FiftyOne format.

fiftyone.utils.ultralytics.to_detections(results, confidence_thresh=None)

Converts ultralytics.YOLO boxes to FiftyOne format.

Parameters

• results – a single or list of ultralytics.engine.results.Results

• confidence_thresh (None) – a confidence threshold to filter boxes

Returns

a single or list of fiftyone.core.labels.Detections

fiftyone.utils.ultralytics.to_instances(results, confidence_thresh=None)

Converts ultralytics.YOLO instance segmentations to FiftyOne format.

Parameters

• results – a single or list of ultralytics.engine.results.Results

• confidence_thresh (None) – a confidence threshold to filter boxes

Returns

a single or list of fiftyone.core.labels.Detections

fiftyone.utils.ultralytics.to_polylines(results, confidence_thresh=None, tolerance=2, filled=True)

Converts ultralytics.YOLO instance segmentations to FiftyOne format.

Parameters

• results – a single or list of ultralytics.engine.results.Results

• confidence_thresh (None) – a confidence threshold to filter boxes

• tolerance (2) – a tolerance, in pixels, when generating approximate polylines for instance masks. Typical values are 1-3 pixels

• filled (True) – whether the polyline should be filled

Returns

a single or list of fiftyone.core.labels.Polylines

fiftyone.utils.ultralytics.to_keypoints(results, confidence_thresh=None)

Converts ultralytics.YOLO keypoints to FiftyOne format.

Parameters

• results – a single or list of ultralytics.engine.results.Results

• confidence_thresh (None) – a confidence threshold to filter keypoints

Returns

a single or list of fiftyone.core.labels.Keypoints

class fiftyone.utils.ultralytics.UltralyticsOutputProcessor(classes=None, **kwargs)

Bases: fiftyone.utils.torch.OutputProcessor

Converts Ultralytics PyTorch Hub model outputs to FiftyOne format.