ai.djl.modality.cv.translator

Class BigGANTranslator

java.lang.Object

ai.djl.modality.cv.translator.BigGANTranslator

All Implemented Interfaces:

PostProcessor<Image[]>, PreProcessor<int[]>, Translator<int[],Image[]>

public final class BigGANTranslator
extends java.lang.Object
implements Translator<int[],Image[]>

Built-in Translator that provides preprocessing and postprocessing for BigGAN.

Constructor Summary

Constructors

Constructor and Description
BigGANTranslator(float truncation) Constructs a translator for BigGAN.

Method Summary

Modifier and Type	Method and Description
Batchifier	getBatchifier() Gets the Batchifier.
NDList	processInput(TranslatorContext ctx, int[] input) Processes the input and converts it to NDList.
Image[]	processOutput(TranslatorContext ctx, NDList list) Processes the output NDList to the corresponding output object.

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

prepare

Constructor Detail

BigGANTranslator

public BigGANTranslator(float truncation)

Constructs a translator for BigGAN.

Parameters:

truncation - value used to scale the normal seed for BigGAN

Method Detail

processOutput

public Image[] processOutput(TranslatorContext ctx,
                             NDList list)

Processes the output NDList to the corresponding output object.

Specified by:

processOutput in interface PostProcessor<Image[]>

Parameters:

ctx - the toolkit used for post-processing

list - the output NDList after inference

Returns:

the output object of expected type

processInput

public NDList processInput(TranslatorContext ctx,
                           int[] input)
                    throws java.lang.Exception

Processes the input and converts it to NDList.

Specified by:

processInput in interface PreProcessor<int[]>

Parameters:

ctx - the toolkit for creating the input NDArray

input - the input object

Returns:

the NDList after pre-processing

Throws:

java.lang.Exception - if an error occurs during processing input

getBatchifier

public Batchifier getBatchifier()

Gets the Batchifier.

Specified by:

getBatchifier in interface Translator<int[],Image[]>

Returns:

the Batchifier