fiftyone.utils.torch.TorchSamplesMixin

class documentation

class TorchSamplesMixin(fom.SamplesMixin): (source)

Known subclasses: fiftyone.utils.sam.SegmentAnythingModel

View In Hierarchy

Undocumented

Method predict Performs prediction on the given data.

Inherited from SamplesMixin:

Method __init__ Undocumented
Method needs_fields.setter Undocumented
Method predict_all Performs prediction on the given iterable of data.
Property needs_fields A dict mapping model-specific keys to sample field names.
Instance Variable _fields Undocumented
def predict(self, img, sample=None): (source) ΒΆ
overrides fiftyone.core.models.SamplesMixin.predict

Performs prediction on the given data.

Image models should support, at minimum, processing arg values that are uint8 numpy arrays (HWC).

Video models should support, at minimum, processing arg values that are eta.core.video.VideoReader instances.

Parameters
imgUndocumented
sample:Nonethe fiftyone.core.sample.Sample associated with the data
argthe data
Returns
a fiftyone.core.labels.Label instance or dict of fiftyone.core.labels.Label instances containing the predictions