Package ai.djl.modality.cv.translator
Class ImageClassificationTranslator.Builder
- java.lang.Object
-
- ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<T>
-
- ai.djl.modality.cv.translator.BaseImageTranslator.ClassificationBuilder<ImageClassificationTranslator.Builder>
-
- ai.djl.modality.cv.translator.ImageClassificationTranslator.Builder
-
- Enclosing class:
- ImageClassificationTranslator
public static class ImageClassificationTranslator.Builder extends BaseImageTranslator.ClassificationBuilder<ImageClassificationTranslator.Builder>
A Builder to construct aImageClassificationTranslator
.
-
-
Field Summary
-
Fields inherited from class ai.djl.modality.cv.translator.BaseImageTranslator.ClassificationBuilder
synsetLoader
-
Fields inherited from class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
batchifier, flag, height, pipeline, width
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ImageClassificationTranslator
build()
Builds theImageClassificationTranslator
with the provided data.protected void
configPostProcess(java.util.Map<java.lang.String,?> arguments)
ImageClassificationTranslator.Builder
optApplySoftmax(boolean applySoftmax)
Sets whether to apply softmax when processing output.ImageClassificationTranslator.Builder
optTopK(int topK)
Set the topK number of classes to be displayed.protected ImageClassificationTranslator.Builder
self()
-
Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator.ClassificationBuilder
optSynset, optSynsetArtifactName, optSynsetUrl, validate
-
Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
addTransform, configPreProcess, optBatchifier, optFlag, setPipeline
-
-
-
-
Method Detail
-
optTopK
public ImageClassificationTranslator.Builder optTopK(int topK)
Set the topK number of classes to be displayed.- Parameters:
topK
- the number of top classes to return- Returns:
- the builder
-
optApplySoftmax
public ImageClassificationTranslator.Builder optApplySoftmax(boolean applySoftmax)
Sets whether to apply softmax when processing output. Some models already include softmax in the last layer, so don't apply softmax when processing model output.- Parameters:
applySoftmax
- boolean whether to apply softmax- Returns:
- the builder
-
self
protected ImageClassificationTranslator.Builder self()
- Specified by:
self
in classBaseImageTranslator.BaseBuilder<ImageClassificationTranslator.Builder>
-
configPostProcess
protected void configPostProcess(java.util.Map<java.lang.String,?> arguments)
-
build
public ImageClassificationTranslator build()
Builds theImageClassificationTranslator
with the provided data.- Returns:
- an
ImageClassificationTranslator
-
-