ai.djl.modality.cv

Class SingleShotDetectionTranslator.Builder

java.lang.Object

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

ai.djl.modality.cv.SingleShotDetectionTranslator.Builder

Enclosing class:

SingleShotDetectionTranslator

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

The builder for SSD translator.

Field Summary

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

flag, pipeline

Method Summary

Modifier and Type	Method and Description
SingleShotDetectionTranslator	build() Builds the translator.
SingleShotDetectionTranslator.Builder	optRescaleSize(double imageWidth, double imageHeight) Sets the optional rescale size.
SingleShotDetectionTranslator.Builder	optThreshold(float threshold) Sets the threshold for prediction accuracy.
protected SingleShotDetectionTranslator.Builder	self()
SingleShotDetectionTranslator.Builder	setClasses(java.util.List<java.lang.String> classes) Sets the class list.
SingleShotDetectionTranslator.Builder	setSynsetArtifactName(java.lang.String synsetArtifactName) Sets the name for the synset.

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

optThreshold

public SingleShotDetectionTranslator.Builder optThreshold(float threshold)

Sets the threshold for prediction accuracy.

Predictions below the threshold will be dropped.

Parameters:

threshold - the threshold for the prediction accuracy

Returns:

this builder

setSynsetArtifactName

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

Sets the name for the synset.

Synset is used to convert the prediction classes to their actual names.

Set either the synset or the classes.

Parameters:

synsetArtifactName - the name of synset

Returns:

this builder

setClasses

public SingleShotDetectionTranslator.Builder setClasses(java.util.List<java.lang.String> classes)

Sets the class list.

Set either the synset or the classes.

Parameters:

classes - the list of classes

Returns:

this builder

optRescaleSize

public SingleShotDetectionTranslator.Builder optRescaleSize(double imageWidth,
                                                            double imageHeight)

Sets the optional rescale size.

Parameters:

imageWidth - the width to rescale images to

imageHeight - the height to rescale images to

Returns:

this builder

self

protected SingleShotDetectionTranslator.Builder self()

Specified by:

self in class ImageTranslator.BaseBuilder<SingleShotDetectionTranslator.Builder>

build

public SingleShotDetectionTranslator build()

Builds the translator.

Returns:

the new translator