Package ai.djl.modality.cv.translator

Class BaseImageTranslatorFactory<O>

java.lang.Object

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

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

Type Parameters:

O - the output type for the TranslatorFactory.

All Implemented Interfaces:

TranslatorFactory

Direct Known Subclasses:

ImageClassificationTranslatorFactory, ImageFeatureExtractorFactory, ObjectDetectionTranslatorFactory, SemanticSegmentationTranslatorFactory, SimplePoseTranslatorFactory, StyleTransferTranslatorFactory

public abstract class BaseImageTranslatorFactory<O>
extends ExpansionTranslatorFactory<Image,O>

A helper to create a TranslatorFactory with the BaseImageTranslator.

Nested Class Summary

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

ExpansionTranslatorFactory.TranslatorExpansion<IbaseT,ObaseT>

Constructor Summary

Constructors

Constructor	Description
BaseImageTranslatorFactory()

Method Summary

Modifier and Type	Method	Description
Class<Image>	getBaseInputType()	Returns the input type for the base translator.
protected Map<ai.djl.util.Pair<Type,Type>,ExpansionTranslatorFactory.TranslatorExpansion<Image,O>>	getExpansions()	Returns the possible expansions of this factory.
protected Map<Type,Function<PreProcessor<Image>,PreProcessor<?>>>	getPreprocessorExpansions()	Returns the possible expansions of this factory.

Methods inherited from class ai.djl.translate.ExpansionTranslatorFactory

buildBaseTranslator, getBaseOutputType, 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

BaseImageTranslatorFactory

public BaseImageTranslatorFactory()

Method Details

getPreprocessorExpansions

protected Map<Type,Function<PreProcessor<Image>,PreProcessor<?>>> getPreprocessorExpansions()

Returns the possible expansions of this factory.

Overrides:

getPreprocessorExpansions in class ExpansionTranslatorFactory<Image,O>

Returns:

the possible expansions of this factory

getExpansions

protected Map<ai.djl.util.Pair<Type,Type>,ExpansionTranslatorFactory.TranslatorExpansion<Image,O>> getExpansions()

Returns the possible expansions of this factory.

Overrides:

getExpansions in class ExpansionTranslatorFactory<Image,O>

Returns:

the possible expansions of this factory

getBaseInputType

public Class<Image> getBaseInputType()

Returns the input type for the base translator.

Specified by:

getBaseInputType in class ExpansionTranslatorFactory<Image,O>

Returns:

the input type for the base translator