Package ai.djl.modality.cv.translator
Class SimplePoseTranslator
- java.lang.Object
-
- ai.djl.modality.cv.translator.BaseImageTranslator<Joints>
-
- ai.djl.modality.cv.translator.SimplePoseTranslator
-
- All Implemented Interfaces:
PostProcessor<Joints>
,PreProcessor<Image>
,Translator<Image,Joints>
public class SimplePoseTranslator extends BaseImageTranslator<Joints>
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
SimplePoseTranslator.Builder
The builder for Pose Estimation translator.-
Nested classes/interfaces inherited from class ai.djl.modality.cv.translator.BaseImageTranslator
BaseImageTranslator.BaseBuilder<T extends BaseImageTranslator.BaseBuilder>, BaseImageTranslator.ClassificationBuilder<T extends BaseImageTranslator.BaseBuilder>, BaseImageTranslator.SynsetLoader
-
-
Field Summary
-
Fields inherited from class ai.djl.modality.cv.translator.BaseImageTranslator
pipeline
-
-
Constructor Summary
Constructors Constructor Description SimplePoseTranslator(SimplePoseTranslator.Builder builder)
Creates the Pose Estimation translator from the given builder.
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static SimplePoseTranslator.Builder
builder()
Creates a builder to build aSimplePoseTranslator
.static SimplePoseTranslator.Builder
builder(java.util.Map<java.lang.String,?> arguments)
Creates a builder to build aSimplePoseTranslator
with specified arguments.Joints
processOutput(TranslatorContext ctx, NDList list)
Processes the output NDList to the corresponding output object.-
Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator
getBatchifier, processInput
-
Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
-
Methods inherited from interface ai.djl.translate.Translator
prepare
-
-
-
-
Constructor Detail
-
SimplePoseTranslator
public SimplePoseTranslator(SimplePoseTranslator.Builder builder)
Creates the Pose Estimation translator from the given builder.- Parameters:
builder
- the builder for the translator
-
-
Method Detail
-
processOutput
public Joints processOutput(TranslatorContext ctx, NDList list)
Processes the output NDList to the corresponding output object.- Parameters:
ctx
- the toolkit used for post-processinglist
- the output NDList after inference- Returns:
- the output object of expected type
-
builder
public static SimplePoseTranslator.Builder builder()
Creates a builder to build aSimplePoseTranslator
.- Returns:
- a new builder
-
builder
public static SimplePoseTranslator.Builder builder(java.util.Map<java.lang.String,?> arguments)
Creates a builder to build aSimplePoseTranslator
with specified arguments.- Parameters:
arguments
- arguments to specify builder options- Returns:
- a new builder
-
-