Shortcuts

SegmentationLabelsOutput

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

A Output which converts the model outputs to the label of the argmax classification per pixel in the image for semantic segmentation tasks.

Parameters
static labels_to_image(img_labels, labels_map)[source]

Function that given an image with labels ids and their pixel intensity mapping, creates an RGB representation for visualisation purposes.

Return type

Tensor

Read the Docs v: latest
Versions
latest
stable
0.7.5
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.