ai.djl.modality.cv.translator

Class InstanceSegmentationTranslator.Builder

java.lang.Object

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

ai.djl.modality.cv.translator.InstanceSegmentationTranslator.Builder

Enclosing class:

InstanceSegmentationTranslator

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

The builder for Instance Segmentation translator.

Field Summary

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

flag, pipeline

Method Summary

Modifier and Type	Method and Description
InstanceSegmentationTranslator	build() Builds the translator.
InstanceSegmentationTranslator.Builder	optMaxEdge(int maxEdge) Sets the maximum edge length of the rescaled image.
InstanceSegmentationTranslator.Builder	optShortEdge(int shortEdge) Sets the shorter edge length of the rescaled image.
InstanceSegmentationTranslator.Builder	optThreshold(float threshold) Sets the threshold for prediction accuracy.
protected InstanceSegmentationTranslator.Builder	self()
InstanceSegmentationTranslator.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

Method Detail

setSynsetArtifactName

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

Sets the name for the synset.

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

Parameters:

synsetArtifactName - the name of synset

Returns:

the builder

optThreshold

public InstanceSegmentationTranslator.Builder optThreshold(float threshold)

Sets the threshold for prediction accuracy.

Predictions below the threshold will be dropped.

Parameters:

threshold - the threshold for prediction accuracy

Returns:

the builder

optShortEdge

public InstanceSegmentationTranslator.Builder optShortEdge(int shortEdge)

Sets the shorter edge length of the rescaled image.

Parameters:

shortEdge - the length of the short edge

Returns:

the builder

optMaxEdge

public InstanceSegmentationTranslator.Builder optMaxEdge(int maxEdge)

Sets the maximum edge length of the rescaled image.

Parameters:

maxEdge - the length of the longest edge

Returns:

the builder

self

protected InstanceSegmentationTranslator.Builder self()

Specified by:

self in class ImageTranslator.BaseBuilder<InstanceSegmentationTranslator.Builder>

build

public InstanceSegmentationTranslator build()

Builds the translator.

Returns:

the new translator