ai.djl.modality.cv

Class SimplePoseTranslator.Builder

java.lang.Object

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

ai.djl.modality.cv.SimplePoseTranslator.Builder

Enclosing class:

SimplePoseTranslator

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

The builder for Pose Estimation translator.

Field Summary

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

flag, pipeline

Method Summary

Modifier and Type	Method and Description
SimplePoseTranslator	build() Builds the translator.
SimplePoseTranslator.Builder	optThreshold(float threshold) Sets the threshold for prediction accuracy.
protected SimplePoseTranslator.Builder	self()

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

self

protected SimplePoseTranslator.Builder self()

Specified by:

self in class ImageTranslator.BaseBuilder<SimplePoseTranslator.Builder>

optThreshold

public SimplePoseTranslator.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

build

public SimplePoseTranslator build()

Builds the translator.

Returns:

the new translator