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
Modifier and TypeMethodDescriptionbuild()
Builds the translator.protected void
configPostProcess
(Map<String, ?> arguments) 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
-
Method Details
-
optThreshold
Sets the threshold for prediction accuracy.Predictions below the threshold will be dropped.
- Parameters:
threshold
- the threshold for prediction accuracy- Returns:
- the builder
-
self
- Specified by:
self
in classBaseImageTranslator.BaseBuilder<SimplePoseTranslator.Builder>
-
configPostProcess
- Overrides:
configPostProcess
in classBaseImageTranslator.BaseBuilder<SimplePoseTranslator.Builder>
-
build
Builds the translator.- Returns:
- the new translator
-