public static class ImageClassificationTranslator.Builder extends ImageTranslator.BaseBuilder<ImageClassificationTranslator.Builder>
ImageClassificationTranslator
.flag, pipeline
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.
|
optFlag, setPipeline
public ImageClassificationTranslator.Builder setSynsetArtifactName(java.lang.String synsetArtifactName)
synsetArtifactName
- a file listing the potential classes for an imagepublic ImageClassificationTranslator.Builder optApplySoftmax(boolean applySoftmax)
applySoftmax
- boolean whether to apply softmaxprotected ImageClassificationTranslator.Builder self()
self
in class ImageTranslator.BaseBuilder<ImageClassificationTranslator.Builder>
public ImageClassificationTranslator build()
ImageClassificationTranslator
with the provided data.ImageClassificationTranslator