fiftyone.utils.torch.ClassifierOutputProcessor

class documentation

class ClassifierOutputProcessor(OutputProcessor): (source)

Constructor: ClassifierOutputProcessor(classes, store_logits)

View In Hierarchy

Output processor for single label classifiers.

Parameters
classesthe list of class labels for the model
store_logitswhether to store logits in the model outputs
Method __call__ Parses the model output.
Method __init__ Undocumented
Instance Variable classes Undocumented
Instance Variable store_logits Undocumented
def __call__(self, output, _, confidence_thresh=None): (source)
overrides fiftyone.utils.torch.OutputProcessor.__call__

Parses the model output.

Parameters
outputeither a FloatTensor[N, M] containing the logits for N images and M classes, or a dict with a "logits" key containing the logits
_unused argument
confidence_thresh:Nonean optional confidence threshold to use to filter any applicable predictions
Returns
a list of fiftyone.core.labels.Classification instances
def __init__(self, classes=None, store_logits=False): (source)
overrides fiftyone.utils.torch.OutputProcessor.__init__

Undocumented

classes: None = (source)
overrides fiftyone.utils.torch.OutputProcessor.classes

Undocumented

store_logits: False = (source)

Undocumented