ai.djl.translate

Interface TranslatorFactory

All Known Implementing Classes:

BigGANTranslatorFactory, DefaultTranslatorFactory, ImageClassificationTranslatorFactory, InstanceSegmentationTranslatorFactory, ObjectDetectionTranslatorFactory, ServingTranslatorFactory, SimplePoseTranslatorFactory, SingleShotDetectionTranslatorFactory, StyleTransferTranslatorFactory, YoloTranslatorFactory, YoloV5TranslatorFactory

public interface TranslatorFactory

A utility class creates Translator instances.

Method Summary

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

Method Detail

getSupportedTypes

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

Returns supported input/output classes.

Returns:

a set of supported input/output classes

isSupported

default boolean isSupported(java.lang.Class<?> input,
                            java.lang.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

Translator<?,?> newInstance(java.lang.Class<?> input,
                            java.lang.Class<?> output,
                            Model model,
                            java.util.Map<java.lang.String,?> arguments)
                     throws TranslateException

Returns a new instance of the Translator class.

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