fiftyone.utils.ultralytics

module documentation

(source)

Utilities for working with Ultralytics.

Class	`FiftyOneRTDETRModel`	FiftyOne wrapper around an `ultralytics.RTDETR` model.
Class	`FiftyOneRTDETRModelConfig`	Configuration for a `FiftyOneRTDETRModel`.
Class	`FiftyOneYOLOClassificationModel`	FiftyOne wrapper around an Ultralytics YOLO classification model.
Class	`FiftyOneYOLOClassificationModelConfig`	Undocumented
Class	`FiftyOneYOLODetectionModel`	FiftyOne wrapper around an Ultralytics YOLO detection model.
Class	`FiftyOneYOLODetectionModelConfig`	Undocumented
Class	`FiftyOneYOLOModel`	FiftyOne wrapper around an `ultralytics.YOLO` model.
Class	`FiftyOneYOLOModelConfig`	Configuration for a `FiftyOneYOLOModel`.
Class	`FiftyOneYOLOOBBModel`	FiftyOne wrapper around an Ultralytics YOLO OBB detection model.
Class	`FiftyOneYOLOOBBModelConfig`	Undocumented
Class	`FiftyOneYOLOPoseModel`	FiftyOne wrapper around an Ultralytics YOLO pose model.
Class	`FiftyOneYOLOPoseModelConfig`	Undocumented
Class	`FiftyOneYOLOSegmentationModel`	FiftyOne wrapper around an Ultralytics YOLO segmentation model.
Class	`FiftyOneYOLOSegmentationModelConfig`	Undocumented
Class	`UltralyticsOutputProcessor`	Converts Ultralytics PyTorch Hub model outputs to FiftyOne format.
Function	`convert_ultralytics_model`	Converts the given Ultralytics model into a FiftyOne model.
Function	`obb_to_polylines`	Converts `ultralytics.YOLO` instance segmentations to FiftyOne format.
Function	`to_detections`	Converts `ultralytics.YOLO` boxes to FiftyOne format.
Function	`to_instances`	Converts `ultralytics.YOLO` instance segmentations to FiftyOne format.
Function	`to_keypoints`	Converts `ultralytics.YOLO` keypoints to FiftyOne format.
Function	`to_polylines`	Converts `ultralytics.YOLO` instance segmentations to FiftyOne format.
Variable	`torch`	Undocumented
Variable	`ultralytics`	Undocumented
Function	`_convert_yolo_classification_model`	Undocumented
Function	`_convert_yolo_detection_model`	Undocumented
Function	`_convert_yolo_obb_model`	Undocumented
Function	`_convert_yolo_pose_model`	Undocumented
Function	`_convert_yolo_segmentation_model`	Undocumented
Function	`_extract_track_ids`	Get ultralytics track ids if present, else use Nones
Function	`_obb_to_polylines`	Undocumented
Function	`_to_detections`	Undocumented
Function	`_to_instances`	Undocumented
Function	`_to_keypoints`	Undocumented
Function	`_to_polylines`	Undocumented

def convert_ultralytics_model(model): (source) ¶

Converts the given Ultralytics model into a FiftyOne model.

Parameters
model	an `ultralytics.YOLO` model
Returns
a `fiftyone.core.models.Model`
Raises
`ValueError`	if the model could not be converted

def obb_to_polylines(results, confidence_thresh=None, filled=False): (source) ¶

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
filled:`False`	whether the polyline should be filled
Returns
a single or list of `fiftyone.core.labels.PolyLines`

def to_detections(results, confidence_thresh=None): (source) ¶

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`

def to_instances(results, confidence_thresh=None): (source) ¶

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`

def to_keypoints(results, confidence_thresh=None): (source) ¶

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`

def to_polylines(results, confidence_thresh=None, tolerance=2, filled=True): (source) ¶

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`

torch = (source) ¶

Undocumented

ultralytics = (source) ¶

Undocumented

def _convert_yolo_classification_model(model): (source) ¶

Undocumented

def _convert_yolo_detection_model(model): (source) ¶

Undocumented

def _convert_yolo_obb_model(model): (source) ¶

Undocumented

def _convert_yolo_pose_model(model): (source) ¶

Undocumented

def _convert_yolo_segmentation_model(model): (source) ¶

Undocumented

def _extract_track_ids(result): (source) ¶

Get ultralytics track ids if present, else use Nones

def _obb_to_polylines(result, filled, confidence_thresh=None): (source) ¶

Undocumented

def _to_detections(result, confidence_thresh=None): (source) ¶

Undocumented

def _to_instances(result, confidence_thresh=None): (source) ¶

Undocumented

def _to_keypoints(result, confidence_thresh=None): (source) ¶

Undocumented

def _to_polylines(result, tolerance, filled, confidence_thresh=None): (source) ¶

Undocumented