FiftyOneLabelsOutput¶
- class flash.core.classification.FiftyOneLabelsOutput(labels=None, multi_label=False, threshold=None, store_logits=False, return_filepath=True)[source]¶
A
Output
which converts the model outputs to FiftyOne classification format.- Parameters
labels¶ (
Optional
[List
[str
]]) – A list of labels, assumed to map the class index to the label for that class.multi_label¶ (
bool
) – If true, treats outputs as multi label logits.threshold¶ (
Optional
[float
]) – A threshold to use to filter candidate labels. In the single label case, predictions below this threshold will be replaced with Nonestore_logits¶ (
bool
) – Boolean determining whether to store logits in the FiftyOne labelsreturn_filepath¶ (
bool
) – Boolean determining whether to return a dict containing filepath and FiftyOne labels (True) or only a list of FiftyOne labels (False)