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

Field Summary

Fields

Modifier and Type	Field and Description
protected Pipeline	pipeline

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.
NDList	processInput(TranslatorContext ctx, Image input) Processes the 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

Field Detail

pipeline

protected Pipeline pipeline

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

processInput

public NDList processInput(TranslatorContext ctx,
                           Image input)

Processes the input and converts it to NDList.

Specified by:

processInput in interface PreProcessor<Image>

Parameters:

ctx - the toolkit for creating the input NDArray

input - the input object

Returns:

the NDList after pre-processing