public static class ImageClassificationTranslator.Builder extends BaseImageTranslator.ClassificationBuilder<ImageClassificationTranslator.Builder>
ImageClassificationTranslator
.synsetLoader
batchifier, flag, height, pipeline, width
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() |
optSynset, optSynsetArtifactName, optSynsetUrl, validate
addTransform, configPreProcess, optBatchifier, optFlag, setPipeline
public ImageClassificationTranslator.Builder optApplySoftmax(boolean applySoftmax)
applySoftmax
- boolean whether to apply softmaxprotected ImageClassificationTranslator.Builder self()
self
in class BaseImageTranslator.BaseBuilder<ImageClassificationTranslator.Builder>
protected void configPostProcess(java.util.Map<java.lang.String,?> arguments)
public ImageClassificationTranslator build()
ImageClassificationTranslator
with the provided data.ImageClassificationTranslator