class documentation

class FiftyOneYOLOClassificationModel(FiftyOneYOLOModel): (source)

Constructor: FiftyOneYOLOClassificationModel(config)

View In Hierarchy

FiftyOne wrapper around an Ultralytics YOLO classification model.

Parameters
configa FiftyOneYOLOClassificationModelConfig
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 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.
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.
def _format_predictions(self, predictions): (source) ΒΆ