ai.djl.modality.cv.translator

Class ImageClassificationTranslator.Builder

java.lang.Object

ai.djl.modality.cv.translator.ImageTranslator.BaseBuilder<ImageClassificationTranslator.Builder>

ai.djl.modality.cv.translator.ImageClassificationTranslator.Builder

Enclosing class:

ImageClassificationTranslator

public static class ImageClassificationTranslator.Builder
extends ImageTranslator.BaseBuilder<ImageClassificationTranslator.Builder>

A Builder to construct a ImageClassificationTranslator.

Field Summary

Fields inherited from class ai.djl.modality.cv.translator.ImageTranslator.BaseBuilder

flag, pipeline

Method Summary

Modifier and Type	Method and Description
ImageClassificationTranslator	build() Builds the ImageClassificationTranslator with the provided data.
ImageClassificationTranslator.Builder	optApplySoftmax(boolean applySoftmax) Sets whether to apply softmax when processing output.
protected ImageClassificationTranslator.Builder	self()
ImageClassificationTranslator.Builder	setSynsetArtifactName(java.lang.String synsetArtifactName) Sets the name of the synset file listing the potential classes for an image.

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

optFlag, setPipeline

Methods inherited from class java.lang.Object

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

Method Detail

setSynsetArtifactName

public ImageClassificationTranslator.Builder setSynsetArtifactName(java.lang.String synsetArtifactName)

Sets the name of the synset file listing the potential classes for an image.

Parameters:

synsetArtifactName - a file listing the potential classes for an image

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

build

public ImageClassificationTranslator build()

Builds the ImageClassificationTranslator with the provided data.

Returns:

an ImageClassificationTranslator