FiftyOneSegmentationLabelsOutput

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

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).