ai.djl.modality.cv.translator

Class StyleTransferTranslatorFactory

java.lang.Object

ai.djl.modality.cv.translator.StyleTransferTranslatorFactory

All Implemented Interfaces:

TranslatorFactory

public class StyleTransferTranslatorFactory
extends java.lang.Object
implements TranslatorFactory

A TranslatorFactory that creates a StyleTransferTranslator instance.

Constructor Summary

Constructors

Constructor and Description
StyleTransferTranslatorFactory()

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

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 Detail

StyleTransferTranslatorFactory

public StyleTransferTranslatorFactory()

Method Detail

getSupportedTypes

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

Returns supported input/output classes.

Specified by:

getSupportedTypes in interface TranslatorFactory

Returns:

a set of supported input/output classes

newInstance

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

Specified by:

newInstance in interface TranslatorFactory

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