Package ai.djl.modality.cv.translator

Class SemanticSegmentationTranslatorFactory

java.lang.Object

ai.djl.translate.ExpansionTranslatorFactory<Image,O>

ai.djl.modality.cv.translator.BaseImageTranslatorFactory<CategoryMask>

ai.djl.modality.cv.translator.SemanticSegmentationTranslatorFactory

All Implemented Interfaces:

TranslatorFactory, Serializable

public class SemanticSegmentationTranslatorFactory
extends BaseImageTranslatorFactory<CategoryMask>
implements Serializable

A TranslatorFactory that creates a SemanticSegmentationTranslator instance.

See Also:

• Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class ai.djl.translate.ExpansionTranslatorFactory

ExpansionTranslatorFactory.TranslatorExpansion<IbaseT,ObaseT>

Constructor Summary

Constructors

Constructor	Description
SemanticSegmentationTranslatorFactory()

Method Summary

Modifier and Type	Method	Description
protected Translator<Image,CategoryMask>	buildBaseTranslator(Model model, Map<String,?> arguments)	Builds the base translator that can be expanded.
Class<CategoryMask>	getBaseOutputType()	Returns the output type for the base translator.

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

getBaseInputType, getExpansions, getPreprocessorExpansions

Methods inherited from class ai.djl.translate.ExpansionTranslatorFactory

getPostprocessorExpansions, getSupportedTypes, newInstance, withTranslator

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

isSupported

Constructor Details

SemanticSegmentationTranslatorFactory

public SemanticSegmentationTranslatorFactory()

Method Details

buildBaseTranslator

protected Translator<Image,CategoryMask> buildBaseTranslator(Model model,
 Map<String,?> arguments)

Builds the base translator that can be expanded.

Specified by:

buildBaseTranslator in class ExpansionTranslatorFactory<Image,CategoryMask>

Parameters:

model - the Model that uses the Translator

arguments - the configurations for a new Translator instance

Returns:

a base translator that can be expanded to form the factory options

getBaseOutputType

public Class<CategoryMask> getBaseOutputType()

Returns the output type for the base translator.

Specified by:

getBaseOutputType in class ExpansionTranslatorFactory<Image,CategoryMask>

Returns:

the output type for the base translator