Package ai.djl.modality.cv.translator

Class SemanticSegmentationTranslator

java.lang.Object

ai.djl.modality.cv.translator.BaseImageTranslator<Image>

ai.djl.modality.cv.translator.SemanticSegmentationTranslator

All Implemented Interfaces:

PostProcessor<Image>, PreProcessor<Image>, Translator<Image,Image>

public class SemanticSegmentationTranslator
extends BaseImageTranslator<Image>

A BaseImageTranslator that post-process the NDArray into DetectedObjects with boundaries at the detailed pixel level.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	SemanticSegmentationTranslator.Builder	The builder for Semantic Segmentation translator.

Nested classes/interfaces inherited from class ai.djl.modality.cv.translator.BaseImageTranslator

BaseImageTranslator.BaseBuilder<T extends BaseImageTranslator.BaseBuilder>, BaseImageTranslator.ClassificationBuilder<T extends BaseImageTranslator.BaseBuilder>, BaseImageTranslator.SynsetLoader

Field Summary

Fields inherited from class ai.djl.modality.cv.translator.BaseImageTranslator

pipeline

Constructor Summary

Constructors

Constructor	Description
SemanticSegmentationTranslator(SemanticSegmentationTranslator.Builder builder)	Creates the Semantic Segmentation translator from the given builder.

Method Summary

Modifier and Type	Method	Description
static SemanticSegmentationTranslator.Builder	builder()	Creates a builder to build a SemanticSegmentationTranslator.
static SemanticSegmentationTranslator.Builder	builder(java.util.Map<java.lang.String,?> arguments)	Creates a builder to build a SemanticSegmentationTranslator with specified arguments.
NDList	processInput(TranslatorContext ctx, Image image)	Processes the input and converts it to NDList.
Image	processOutput(TranslatorContext ctx, NDList list)	Processes the output NDList to the corresponding output object.

Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator

getBatchifier

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface ai.djl.translate.Translator

prepare

Constructor Detail

SemanticSegmentationTranslator

public SemanticSegmentationTranslator(SemanticSegmentationTranslator.Builder builder)

Creates the Semantic Segmentation translator from the given builder.

Parameters:

builder - the builder for the translator

Method Detail

processInput

public NDList processInput(TranslatorContext ctx,
                           Image image)

Processes the input and converts it to NDList.

Specified by:

processInput in interface PreProcessor<Image>

Overrides:

processInput in class BaseImageTranslator<Image>

Parameters:

ctx - the toolkit for creating the input NDArray

image - the input object

Returns:

the NDList after pre-processing

processOutput

public Image processOutput(TranslatorContext ctx,
                           NDList list)

Processes the output NDList to the corresponding output object.

Parameters:

ctx - the toolkit used for post-processing

list - the output NDList after inference

Returns:

the output object of expected type

builder

public static SemanticSegmentationTranslator.Builder builder()

Creates a builder to build a SemanticSegmentationTranslator.

Returns:

a new builder

builder

public static SemanticSegmentationTranslator.Builder builder(java.util.Map<java.lang.String,?> arguments)

Creates a builder to build a SemanticSegmentationTranslator with specified arguments.

Parameters:

arguments - arguments to specify builder options

Returns:

a new builder