class documentation

class FiftyOneYOLOSegmentationModel(FiftyOneYOLOModel): (source)

Constructor: FiftyOneYOLOSegmentationModel(config)

View In Hierarchy

FiftyOne wrapper around an Ultralytics YOLO segmentation model.

Parameters
configa FiftyOneYOLOSegmentationModelConfig
Property ragged_batches True/False whether transforms may return tensors of different sizes. If True, then passing ragged lists of data to predict_all is not allowed.
Method _format_predictions Undocumented

Inherited from FiftyOneYOLOModel:

Method __init__ Undocumented
Method predict Performs prediction on the given data.
Instance Variable config Undocumented
Instance Variable device Undocumented
Instance Variable model Undocumented
Property media_type The media type processed by the model.
Property preprocess Whether to apply transforms during inference (True) or to assume that they have already been applied (False).
Property transforms The preprocessing function that will/must be applied to each input before prediction, or None if no preprocessing is performed.
Method _load_model Undocumented

Inherited from Model (via FiftyOneYOLOModel):

Method __enter__ Undocumented
Method __exit__ Undocumented
Method predict_all Performs prediction on the given iterable of data.
Method preprocess.setter Undocumented
Property can_embed_prompts Whether this instance can generate prompt embeddings.
Property has_embeddings Whether this instance can generate embeddings.
Property has_logits Whether this instance can generate logits for its predictions.
@property
ragged_batches = (source)

True/False whether transforms may return tensors of different sizes. If True, then passing ragged lists of data to predict_all is not allowed.

def _format_predictions(self, predictions): (source)