Package ai.djl.modality.cv.translator

Class YoloPoseTranslator.Builder

java.lang.Object

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

ai.djl.modality.cv.translator.YoloPoseTranslator.Builder

Enclosing class:

YoloPoseTranslator

public static class YoloPoseTranslator.Builder
extends BaseImageTranslator.BaseBuilder<YoloPoseTranslator.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	Description
YoloPoseTranslator	build()	Builds the translator.
protected void	configPostProcess(Map<String,?> arguments)
YoloPoseTranslator.Builder	optNmsThreshold(float nmsThreshold)	Sets the NMS threshold.
YoloPoseTranslator.Builder	optThreshold(float threshold)	Sets the threshold for prediction accuracy.
protected YoloPoseTranslator.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 Details

optThreshold

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

optNmsThreshold

public YoloPoseTranslator.Builder optNmsThreshold(float nmsThreshold)

Sets the NMS threshold.

Parameters:

nmsThreshold - the NMS threshold

Returns:

this builder

self

protected YoloPoseTranslator.Builder self()

Specified by:

self in class BaseImageTranslator.BaseBuilder<YoloPoseTranslator.Builder>

configPostProcess

protected void configPostProcess(Map<String,?> arguments)

Overrides:

configPostProcess in class BaseImageTranslator.BaseBuilder<YoloPoseTranslator.Builder>

build

public YoloPoseTranslator build()

Builds the translator.

Returns:

the new translator