ai.djl.modality.cv.translator

Class BaseImageTranslator<T>

java.lang.Object

ai.djl.modality.cv.translator.BaseImageTranslator<T>

Type Parameters:

T - the output object type

All Implemented Interfaces:

PostProcessor<T>, PreProcessor<Image>, Translator<Image,T>

Direct Known Subclasses:

ImageClassificationTranslator, InstanceSegmentationTranslator, ObjectDetectionTranslator, SimplePoseTranslator

public abstract class BaseImageTranslator<T>
extends java.lang.Object
implements Translator<Image,T>

Built-in Translator that provides default image pre-processing.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	BaseImageTranslator.BaseBuilder<T extends BaseImageTranslator.BaseBuilder> A builder to extend for all classes extending the BaseImageTranslator.
static class	BaseImageTranslator.ClassificationBuilder<T extends BaseImageTranslator.BaseBuilder> A Builder to construct a ImageClassificationTranslator.
protected static class	BaseImageTranslator.SynsetLoader

Constructor Summary

Constructors

Constructor and Description
BaseImageTranslator(BaseImageTranslator.BaseBuilder<?> builder) Constructs an ImageTranslator with the provided builder.

Method Summary

Modifier and Type	Method and Description
Batchifier	getBatchifier() Gets the Batchifier.
protected static boolean	getBooleanValue(java.util.Map<java.lang.String,?> arguments, java.lang.String key, boolean def)
protected static float	getFloatValue(java.util.Map<java.lang.String,?> arguments, java.lang.String key, float def)
protected static int	getIntValue(java.util.Map<java.lang.String,?> arguments, java.lang.String key, int def)
Pipeline	getPipeline() Gets the Pipeline applied to the input.
protected static java.lang.String	getStringValue(java.util.Map<java.lang.String,?> arguments, java.lang.String key, java.lang.String def)
NDList	processInput(TranslatorContext ctx, Image input) Processes the Image input and converts it to NDList.

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

Methods inherited from interface ai.djl.translate.PostProcessor

processOutput

Constructor Detail

BaseImageTranslator

public BaseImageTranslator(BaseImageTranslator.BaseBuilder<?> builder)

Constructs an ImageTranslator with the provided builder.

Parameters:

builder - the data to build with

Method Detail

getBatchifier

public Batchifier getBatchifier()

Gets the Batchifier.

Specified by:

getBatchifier in interface Translator<Image,T>

Returns:

the Batchifier

getPipeline

public Pipeline getPipeline()

Gets the Pipeline applied to the input.

Specified by:

getPipeline in interface PreProcessor<Image>

Returns:

the Pipeline

processInput

public NDList processInput(TranslatorContext ctx,
                           Image input)

Processes the Image input and converts it to NDList.

Specified by:

processInput in interface PreProcessor<Image>

Parameters:

ctx - the toolkit that helps create the input NDArray

input - the Image input

Returns:

a NDList

getStringValue

protected static java.lang.String getStringValue(java.util.Map<java.lang.String,?> arguments,
                                                 java.lang.String key,
                                                 java.lang.String def)

getIntValue

protected static int getIntValue(java.util.Map<java.lang.String,?> arguments,
                                 java.lang.String key,
                                 int def)

getFloatValue

protected static float getFloatValue(java.util.Map<java.lang.String,?> arguments,
                                     java.lang.String key,
                                     float def)

getBooleanValue

protected static boolean getBooleanValue(java.util.Map<java.lang.String,?> arguments,
                                         java.lang.String key,
                                         boolean def)