ai.djl.modality.cv.translator

Class ImageTranslator<T>

java.lang.Object

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

Type Parameters:

T - the output object type

All Implemented Interfaces:

PostProcessor<T>, PreProcessor<java.awt.image.BufferedImage>, Translator<java.awt.image.BufferedImage,T>

Direct Known Subclasses:

ImageClassificationTranslator, InstanceSegmentationTranslator, SimplePoseTranslator, SingleShotDetectionTranslator

public abstract class ImageTranslator<T>
extends java.lang.Object
implements Translator<java.awt.image.BufferedImage,T>

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

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ImageTranslator.BaseBuilder<T extends ImageTranslator.BaseBuilder> A builder to extend for all classes extending the ImageTranslator.

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
Pipeline	getPipeline() Gets the Pipeline applied to the input.
NDList	processInput(TranslatorContext ctx, java.awt.image.BufferedImage input) Processes the BufferedImage 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

getBatchifier, prepare

Methods inherited from interface ai.djl.translate.PostProcessor

processOutput

Constructor Detail

ImageTranslator

public ImageTranslator(ImageTranslator.BaseBuilder<?> builder)

Constructs an ImageTranslator with the provided builder.

Parameters:

builder - the data to build with

Method Detail

getPipeline

public Pipeline getPipeline()

Gets the Pipeline applied to the input.

Specified by:

getPipeline in interface PreProcessor<java.awt.image.BufferedImage>

Returns:

the Pipeline

processInput

public NDList processInput(TranslatorContext ctx,
                           java.awt.image.BufferedImage input)

Processes the BufferedImage input and converts it to NDList.

Specified by:

processInput in interface PreProcessor<java.awt.image.BufferedImage>

Parameters:

ctx - the toolkit that helps create the input NDArray

input - the BufferedImage input

Returns:

a NDList