Shortcuts

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 None

  • store_logits (bool) – Boolean determining whether to store logits in the FiftyOne labels

  • return_filepath (bool) – Boolean determining whether to return a dict containing filepath and FiftyOne labels (True) or only a list of FiftyOne labels (False)

Read the Docs v: latest
Versions
latest
stable
0.7.4
0.7.3
0.7.2
0.7.1
0.7.0
0.6.0
0.5.2
0.5.1
0.5.0
0.4.0
0.3.2
0.3.1
0.3.0
0.2.3
0.2.2
0.2.1
0.2.0
0.1.0post1
docs-fix_typing
Downloads
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.