fiftyone.utils.ultralytics¶
Utilities for working with Ultralytics.
Classes:
|
Converts Ultralytics PyTorch Hub model outputs to FiftyOne format. |
Functions:
|
Converts |
|
Converts |
|
Converts |
|
Converts |
-
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.