ai.djl.modality.cv

Class ImageClassificationTranslator

java.lang.Object

ai.djl.modality.cv.ImageTranslator<Classifications>

ai.djl.modality.cv.ImageClassificationTranslator

All Implemented Interfaces:

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

public class ImageClassificationTranslator
extends ImageTranslator<Classifications>

A generic Translator for Image Classification tasks.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ImageClassificationTranslator.Builder A Builder to construct a ImageClassificationTranslator.

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

ImageTranslator.BaseBuilder<T extends ImageTranslator.BaseBuilder>

Constructor Summary

Constructors

Constructor and Description
ImageClassificationTranslator(ImageClassificationTranslator.Builder builder) Constructs an Image Classification using ImageClassificationTranslator.Builder.

Method Summary

Modifier and Type	Method and Description
static ImageClassificationTranslator.Builder	builder() Creates a builder to build a ImageClassificationTranslator.
Classifications	processOutput(TranslatorContext ctx, NDList list) Processes the output NDList to the corresponding output object.

Methods inherited from class ai.djl.modality.cv.ImageTranslator

getPipeline, processInput

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

Constructor Detail

ImageClassificationTranslator

public ImageClassificationTranslator(ImageClassificationTranslator.Builder builder)

Constructs an Image Classification using ImageClassificationTranslator.Builder.

Parameters:

builder - the data to build with

Method Detail

processOutput

public Classifications processOutput(TranslatorContext ctx,
                                     NDList list)
                              throws java.io.IOException

Processes the output NDList to the corresponding output object.

Parameters:

ctx - the toolkit used for post-processing

list - the output NDList after inference

Returns:

the output object of expected type

Throws:

java.io.IOException

builder

public static ImageClassificationTranslator.Builder builder()

Creates a builder to build a ImageClassificationTranslator.

Returns:

a new builder