ai.djl.modality.cv.translator

Class SingleShotDetectionTranslator.Builder

java.lang.Object

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

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

flag, pipeline

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
SingleShotDetectionTranslator	build() Builds the translator.
java.util.List<java.lang.String>	getClasses() Get classes.
double	getImageHeight() Get resized image height.
double	getImageWidth() Get resized image width.
java.lang.String	getSynsetArtifactName() Get symset artifact name.
float	getThreshold() Get threshold.
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.translator.ImageTranslator.BaseBuilder

optFlag, setPipeline

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

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

getThreshold

public float getThreshold()

Get threshold.

Returns:

threshold

getSynsetArtifactName

public java.lang.String getSynsetArtifactName()

Get symset artifact name.

Returns:

name

getClasses

public java.util.List<java.lang.String> getClasses()

Get classes.

Returns:

classes

getImageWidth

public double getImageWidth()

Get resized image width.

Returns:

image width

getImageHeight

public double getImageHeight()

Get resized image height.

Returns:

image height