ai.djl.modality.cv.translator

Class SimplePoseTranslator.Builder

java.lang.Object

ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<SimplePoseTranslator.Builder>

ai.djl.modality.cv.translator.SimplePoseTranslator.Builder

Enclosing class:

SimplePoseTranslator

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

The builder for Pose Estimation translator.

Field Summary

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

batchifier, flag, height, pipeline, width

Method Summary

Modifier and Type	Method and Description
SimplePoseTranslator	build() Builds the translator.
protected void	configPostProcess(java.util.Map<java.lang.String,?> arguments)
SimplePoseTranslator.Builder	optThreshold(float threshold) Sets the threshold for prediction accuracy.
protected SimplePoseTranslator.Builder	self()

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

addTransform, configPreProcess, optBatchifier, optFlag, setPipeline, validate

Methods inherited from class java.lang.Object

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

Method Detail

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

self

protected SimplePoseTranslator.Builder self()

Specified by:

self in class BaseImageTranslator.BaseBuilder<SimplePoseTranslator.Builder>

configPostProcess

protected void configPostProcess(java.util.Map<java.lang.String,?> arguments)

Overrides:

configPostProcess in class BaseImageTranslator.BaseBuilder<SimplePoseTranslator.Builder>

build

public SimplePoseTranslator build()

Builds the translator.

Returns:

the new translator