Package 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
All Methods Instance Methods Concrete Methods Modifier and Type Method Description SimplePoseTranslatorbuild()Builds the translator.protected voidconfigPostProcess(java.util.Map<java.lang.String,?> arguments)SimplePoseTranslator.BuilderoptThreshold(float threshold)Sets the threshold for prediction accuracy.protected SimplePoseTranslator.Builderself()-
Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder
addTransform, configPreProcess, optBatchifier, optFlag, setPipeline, validate
-
-
-
-
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:
selfin classBaseImageTranslator.BaseBuilder<SimplePoseTranslator.Builder>
-
configPostProcess
protected void configPostProcess(java.util.Map<java.lang.String,?> arguments)
- Overrides:
configPostProcessin classBaseImageTranslator.BaseBuilder<SimplePoseTranslator.Builder>
-
build
public SimplePoseTranslator build()
Builds the translator.- Returns:
- the new translator
-
-