ai.djl.modality.cv.translator.wrapper

Class InputStreamTranslator<T>

java.lang.Object

ai.djl.modality.cv.translator.wrapper.InputStreamTranslator<T>

Type Parameters:

T - the output object type

All Implemented Interfaces:

PostProcessor<T>, PreProcessor<java.io.InputStream>, Translator<java.io.InputStream,T>

public class InputStreamTranslator<T>
extends java.lang.Object
implements Translator<java.io.InputStream,T>

Built-in Translator that provides image pre-processing from InputStream.

Constructor Summary

Constructors

Constructor and Description
InputStreamTranslator(Translator<java.awt.image.BufferedImage,T> translator) Creates a UrlTranslator instance.

Method Summary

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

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.PreProcessor

getPipeline

Constructor Detail

InputStreamTranslator

public InputStreamTranslator(Translator<java.awt.image.BufferedImage,T> translator)

Creates a UrlTranslator instance.

Parameters:

translator - a Translator that can process image

Method Detail

processInput

public NDList processInput(TranslatorContext ctx,
                           java.io.InputStream input)
                    throws java.lang.Exception

Processes the input and converts it to NDList.

Specified by:

processInput in interface PreProcessor<java.io.InputStream>

Parameters:

ctx - the toolkit for creating the input NDArray

input - the input object

Returns:

the NDList after pre-processing

Throws:

java.lang.Exception - if an error occurs during processing input

processOutput

public T processOutput(TranslatorContext ctx,
                       NDList list)
                throws java.lang.Exception

Processes the output NDList to the corresponding output object.

Specified by:

processOutput in interface PostProcessor<T>

Parameters:

ctx - the toolkit used for post-processing

list - the output NDList after inference

Returns:

the output object of expected type

Throws:

java.lang.Exception - if an error occurs during processing output