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>
-
Nested Class Summary
Nested ClassesModifier and TypeClassDescriptionstatic class
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
ConstructorsConstructorDescriptionCreates the Pose Estimation translator from the given builder. -
Method Summary
Modifier and TypeMethodDescriptionstatic SimplePoseTranslator.Builder
builder()
Creates a builder to build aSimplePoseTranslator
.static SimplePoseTranslator.Builder
Creates a builder to build aSimplePoseTranslator
with specified arguments.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
getExpansions, prepare, toBatchTranslator, toBatchTranslator
-
Constructor Details
-
SimplePoseTranslator
Creates the Pose Estimation translator from the given builder.- Parameters:
builder
- the builder for the translator
-
-
Method Details
-
processOutput
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
Creates a builder to build aSimplePoseTranslator
.- Returns:
- a new builder
-
builder
Creates a builder to build aSimplePoseTranslator
with specified arguments.- Parameters:
arguments
- arguments to specify builder options- Returns:
- a new builder
-