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>

A BaseImageTranslator that post-process the NDArray into human Joints.

Nested Class Summary

Nested Classes

Modifier and Type	Class and 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 and Description
SimplePoseTranslator(SimplePoseTranslator.Builder builder) Creates the Pose Estimation translator from the given builder.

Method Summary

Modifier and Type	Method and Description
static SimplePoseTranslator.Builder	builder() Creates a builder to build a SimplePoseTranslator.
static SimplePoseTranslator.Builder	builder(java.util.Map<java.lang.String,?> arguments) Creates a builder to build a SimplePoseTranslator 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-processing

list - the output NDList after inference

Returns:

the output object of expected type

builder

public static SimplePoseTranslator.Builder builder()

Creates a builder to build a SimplePoseTranslator.

Returns:

a new builder

builder

public static SimplePoseTranslator.Builder builder(java.util.Map<java.lang.String,?> arguments)

Creates a builder to build a SimplePoseTranslator with specified arguments.

Parameters:

arguments - arguments to specify builder options

Returns:

a new builder