Package 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, ImageFeatureExtractor, InstanceSegmentationTranslator, ObjectDetectionTranslator, SemanticSegmentationTranslator, SimplePoseTranslator, YoloPoseTranslator

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

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

Nested Class Summary

Nested Classes

Modifier and Type	Class	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 final class	BaseImageTranslator.SynsetLoader

Field Summary

Fields

Modifier and Type	Field	Description
protected int	height
protected Pipeline	pipeline
protected int	width

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	Description
Batchifier	getBatchifier()	Returns 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.PostProcessor

processOutput

Methods inherited from interface ai.djl.translate.Translator

batchProcessInput, batchProcessOutput, getExpansions, prepare

Field Details

pipeline

protected Pipeline pipeline

width

protected int width

height

protected int height

Constructor Details

BaseImageTranslator

public BaseImageTranslator(BaseImageTranslator.BaseBuilder<?> builder)

Constructs an ImageTranslator with the provided builder.

Parameters:

builder - the data to build with

Method Details

getBatchifier

public Batchifier getBatchifier()

Returns 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<T>

Parameters:

ctx - the toolkit for creating the input NDArray

input - the input object

Returns:

the NDList after pre-processing