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>
A generic
Translator
for Image Classification tasks.-
Nested Class Summary
Modifier and TypeClassDescriptionstatic class
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
ConstructorDescriptionConstructs an Image Classification usingImageClassificationTranslator.Builder
. -
Method Summary
Modifier and TypeMethodDescriptionbuilder()
Creates a builder to build aImageClassificationTranslator
.Creates a builder to build aImageClassificationTranslator
with specified arguments.Returns possibleTranslatorOptions
that can be built using thisTranslator
.void
prepare
(TranslatorContext ctx) Prepares the translator with the manager and model to use.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
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
toBatchTranslator, toBatchTranslator
-
Constructor Details
-
ImageClassificationTranslator
Constructs an Image Classification usingImageClassificationTranslator.Builder
.- Parameters:
builder
- the data to build with
-
-
Method Details
-
prepare
Prepares the translator with the manager and model to use.- Parameters:
ctx
- the context for thePredictor
.- Throws:
IOException
-
processOutput
Processes the output NDList to the corresponding output object.- Parameters:
ctx
- the toolkit used for post-processinglist
- the output NDList after inference, usually immutable in engines like PyTorch. @see Issue 1774- Returns:
- the output object of expected type
-
getExpansions
Returns possibleTranslatorOptions
that can be built using thisTranslator
.- Returns:
- possible options or null if not defined
-
builder
Creates a builder to build aImageClassificationTranslator
.- Returns:
- a new builder
-
builder
Creates a builder to build aImageClassificationTranslator
with specified arguments.- Parameters:
arguments
- arguments to specify builder options- Returns:
- a new builder
-