fiftyone.utils.ultralytics.UltralyticsOutputProcessor

class documentation

class UltralyticsOutputProcessor(fout.OutputProcessor): (source)

Constructor: UltralyticsOutputProcessor(classes, **kwargs)

View In Hierarchy

Converts Ultralytics PyTorch Hub model outputs to FiftyOne format.

Method __call__ Parses the model output.
Method _to_detections Undocumented

Inherited from OutputProcessor:

Method __init__ Undocumented
def __call__(self, result, frame_size, confidence_thresh=None): (source)
overrides fiftyone.utils.torch.OutputProcessor.__call__

Parses the model output.

Parameters
resultUndocumented
frame_sizethe (width, height) of the frames in the batch
confidence_thresh:Nonean optional confidence threshold to use to filter any applicable predictions
outputthe model output for the batch of predictions
Returns
a list of fiftyone.core.labels.Label instances
def _to_detections(self, df): (source)

Undocumented