Shortcuts

FiftyOneDetectionLabelsOutput

class flash.image.detection.output.FiftyOneDetectionLabelsOutput(labels=None, threshold=None, return_filepath=True)[source]

A Output which converts model outputs to FiftyOne detection format.

Parameters
  • labels (Optional[List[str]]) – A list of labels, assumed to map the class index to the label for that class.

  • threshold (Optional[float]) – a score threshold to apply to candidate detections.

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