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: stable
Versions
latest
stable
0.8.1
0.8.0
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
Downloads
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.