Shortcuts

FiftyOneLabels

class flash.core.classification.FiftyOneLabels(labels=None, multi_label=False, threshold=None, store_logits=False, return_filepath=False)[source]

A Serializer 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. If labels is not provided, will attempt to get them from the LabelsState.

  • 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.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
Downloads
pdf
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.