ai.djl.modality.cv

Class ImageClassificationTranslator.Builder

java.lang.Object

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

ai.djl.modality.cv.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.ImageTranslator.BaseBuilder

flag, pipeline

Method Summary

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

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