Shortcuts

FiftyOneSegmentationLabelsOutput

class flash.image.segmentation.output.FiftyOneSegmentationLabelsOutput(labels_map=None, visualize=False, return_filepath=True)[source]

A Output which converts the model outputs to FiftyOne segmentation format.

Parameters
  • labels_map (Optional[Dict[int, Tuple[int, int, int]]]) – A dictionary that map the labels ids to pixel intensities.

  • visualize (bool) – whether to visualize the image 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).