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 a ImageClassificationTranslator.

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

Modifier and Type	Method and Description
ImageClassificationTranslator	build() Builds the ImageClassificationTranslator 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.
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

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

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 class BaseImageTranslator.BaseBuilder<ImageClassificationTranslator.Builder>

configPostProcess

protected void configPostProcess(java.util.Map<java.lang.String,?> arguments)

Overrides:

configPostProcess in class BaseImageTranslator.ClassificationBuilder<ImageClassificationTranslator.Builder>

build

public ImageClassificationTranslator build()

Builds the ImageClassificationTranslator with the provided data.

Returns:

an ImageClassificationTranslator