ai.djl.modality.cv.translator

Class ObjectDetectionTranslator.BaseBuilder<T extends ObjectDetectionTranslator.BaseBuilder>

java.lang.Object

ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<T>

ai.djl.modality.cv.translator.ObjectDetectionTranslator.BaseBuilder<T>

Direct Known Subclasses:

SingleShotDetectionTranslator.Builder, YoloTranslator.Builder

Enclosing class:

ObjectDetectionTranslator

public abstract static class ObjectDetectionTranslator.BaseBuilder<T extends ObjectDetectionTranslator.BaseBuilder>
extends BaseImageTranslator.BaseBuilder<T>

The base builder for the object detection translator.

Field Summary

Fields

Modifier and Type	Field and Description
protected java.util.List<java.lang.String>	classes
protected double	imageHeight
protected double	imageWidth
protected java.lang.String	synsetArtifactName
protected float	threshold

Fields inherited from class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder

flag, pipeline

Constructor Summary

Constructors

Constructor and Description
BaseBuilder()

Method Summary

Modifier and Type	Method and Description
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.
T	optRescaleSize(double imageWidth, double imageHeight) Sets the optional rescale size.
T	optThreshold(float threshold) Sets the threshold for prediction accuracy.
protected abstract T	self()
T	setClasses(java.util.List<java.lang.String> classes) Sets the class list.
T	setSynsetArtifactName(java.lang.String synsetArtifactName) Sets the name for the synset.
protected void	validate()

Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder

optFlag, setPipeline

Methods inherited from class java.lang.Object

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

Field Detail

threshold

protected float threshold

synsetArtifactName

protected java.lang.String synsetArtifactName

classes

protected java.util.List<java.lang.String> classes

imageWidth

protected double imageWidth

imageHeight

protected double imageHeight

Constructor Detail

BaseBuilder

public BaseBuilder()

Method Detail

self

protected abstract T self()

Specified by:

self in class BaseImageTranslator.BaseBuilder<T extends ObjectDetectionTranslator.BaseBuilder>

validate

protected void validate()

optThreshold

public T 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 T 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 T 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 T 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

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