Package ai.djl.modality.cv.translator
Class ImageClassificationTranslator
- java.lang.Object
-
- ai.djl.modality.cv.translator.BaseImageTranslator<Classifications>
-
- ai.djl.modality.cv.translator.ImageClassificationTranslator
-
- All Implemented Interfaces:
PostProcessor<Classifications>
,PreProcessor<Image>
,Translator<Image,Classifications>
public class ImageClassificationTranslator extends BaseImageTranslator<Classifications>
A genericTranslator
for Image Classification tasks.
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
ImageClassificationTranslator.Builder
A Builder to construct aImageClassificationTranslator
.-
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
pipeline
-
-
Constructor Summary
Constructors Constructor Description ImageClassificationTranslator(ImageClassificationTranslator.Builder builder)
Constructs an Image Classification usingImageClassificationTranslator.Builder
.
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static ImageClassificationTranslator.Builder
builder()
Creates a builder to build aImageClassificationTranslator
.static ImageClassificationTranslator.Builder
builder(java.util.Map<java.lang.String,?> arguments)
Creates a builder to build aImageClassificationTranslator
with specified arguments.void
prepare(TranslatorContext ctx)
Prepares the translator with the manager and model to use.Classifications
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, processInput
-
-
-
-
Constructor Detail
-
ImageClassificationTranslator
public ImageClassificationTranslator(ImageClassificationTranslator.Builder builder)
Constructs an Image Classification usingImageClassificationTranslator.Builder
.- Parameters:
builder
- the data to build with
-
-
Method Detail
-
prepare
public void prepare(TranslatorContext ctx) throws java.io.IOException
Prepares the translator with the manager and model to use.- Parameters:
ctx
- the context for thePredictor
.- Throws:
java.io.IOException
-
processOutput
public Classifications processOutput(TranslatorContext ctx, NDList list)
Processes the output NDList to the corresponding output object.- Parameters:
ctx
- the toolkit used for post-processinglist
- the output NDList after inference- Returns:
- the output object of expected type
-
builder
public static ImageClassificationTranslator.Builder builder()
Creates a builder to build aImageClassificationTranslator
.- Returns:
- a new builder
-
builder
public static ImageClassificationTranslator.Builder builder(java.util.Map<java.lang.String,?> arguments)
Creates a builder to build aImageClassificationTranslator
with specified arguments.- Parameters:
arguments
- arguments to specify builder options- Returns:
- a new builder
-
-