Package ai.djl.translate

Interface TranslatorFactory

All Known Implementing Classes:

BaseImageTranslatorFactory, BigGANTranslatorFactory, DefaultTranslatorFactory, DeferredTranslatorFactory, ExpansionTranslatorFactory, ImageClassificationTranslatorFactory, ImageFeatureExtractorFactory, InstanceSegmentationTranslatorFactory, NoopServingTranslatorFactory, ObjectDetectionTranslatorFactory, SemanticSegmentationTranslatorFactory, ServingTranslatorFactory, SimplePoseTranslatorFactory, SingleShotDetectionTranslatorFactory, SpeechRecognitionTranslatorFactory, StyleTransferTranslatorFactory, YoloTranslatorFactory, YoloV5TranslatorFactory, YoloV8TranslatorFactory

public interface TranslatorFactory

A utility class creates Translator instances.

Method Summary

Modifier and Type	Method	Description
Set<ai.djl.util.Pair<Type,Type>>	getSupportedTypes()	Returns supported input/output classes.
default boolean	isSupported(Class<?> input, Class<?> output)	Returns if the input/output is supported by the TranslatorFactory.
<I, O> Translator<I,O>	newInstance(Class<I> input, Class<O> output, Model model, Map<String,?> arguments)	Returns a new instance of the Translator class.

Method Details

getSupportedTypes

Set<ai.djl.util.Pair<Type,Type>> getSupportedTypes()

Returns supported input/output classes.

Returns:

a set of supported input/output classes

isSupported

default boolean isSupported(Class<?> input,
 Class<?> output)

Returns if the input/output is supported by the TranslatorFactory.

Parameters:

input - the input class

output - the output class

Returns:

true if the input/output type is supported

newInstance

<I,
O> Translator<I,O> newInstance(Class<I> input,
 Class<O> output,
 Model model,
 Map<String,?> arguments)
                           throws TranslateException

Returns a new instance of the Translator class.

Type Parameters:

I - the input data type

O - the output data type

Parameters:

input - the input class

output - the output class

model - the Model that uses the Translator

arguments - the configurations for a new Translator instance

Returns:

a new instance of the Translator class

Throws:

TranslateException - if failed to create Translator instance