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 classSimplePoseTranslator.BuilderThe 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.Builderbuilder()Creates a builder to build aSimplePoseTranslator.static SimplePoseTranslator.Builderbuilder(java.util.Map<java.lang.String,?> arguments)Creates a builder to build aSimplePoseTranslatorwith specified arguments.JointsprocessOutput(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
getExpansions, prepare, toBatchTranslator, toBatchTranslator
-
-
-
-
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, usually immutable in engines like PyTorch. @see Issue 1774- 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 aSimplePoseTranslatorwith specified arguments.- Parameters:
arguments- arguments to specify builder options- Returns:
- a new builder
-
-