Package ai.djl.modality.cv.translator

Class BaseImagePreProcessor

java.lang.Object

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

ai.djl.modality.cv.translator.BaseImagePreProcessor

All Implemented Interfaces:

PostProcessor<Void>, PreProcessor<Image>, Translator<Image,Void>

public class BaseImagePreProcessor
extends BaseImageTranslator<Void>

A BaseImageTranslator that only handles pre-processing inputs.

Nested Class Summary

Nested classes/interfaces inherited from class ai.djl.modality.cv.translator.BaseImageTranslator

BaseImageTranslator.BaseBuilder<T extends BaseImageTranslator.BaseBuilder>, BaseImageTranslator.ClassificationBuilder<T extends BaseImageTranslator.BaseBuilder>, BaseImageTranslator.SynsetLoader

Field Summary

Fields inherited from class ai.djl.modality.cv.translator.BaseImageTranslator

height, pipeline, width

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	Description
NDList	processInput(TranslatorContext ctx, Image input)	Processes the input and converts it to NDList.
Void	processOutput(TranslatorContext ctx, NDList list)	Processes the output NDList to the corresponding output object.

Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator

getBatchifier

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

batchProcessInput, batchProcessOutput, getExpansions, prepare

Constructor Details

BaseImagePreProcessor

public BaseImagePreProcessor(BaseImageTranslator.BaseBuilder<?> builder)

Constructs an ImageTranslator with the provided builder.

Parameters:

builder - the data to build with

Method Details

processOutput

public Void processOutput(TranslatorContext ctx,
 NDList list)

Processes the output NDList to the corresponding output object.

Parameters:

ctx - the toolkit used for post-processing

list - the output NDList after inference, usually immutable in engines like PyTorch. @see Issue 1774

Returns:

the output object of expected type

processInput

public NDList processInput(TranslatorContext ctx,
 Image input)

Processes the input and converts it to NDList.

Specified by:

processInput in interface PreProcessor<Image>

Overrides:

processInput in class BaseImageTranslator<Void>

Parameters:

ctx - the toolkit for creating the input NDArray

input - the input object

Returns:

the NDList after pre-processing