ai.djl.mxnet.zoo.cv.poseestimation

Class SimplePoseModelLoader

java.lang.Object

ai.djl.repository.zoo.BaseModelLoader<java.awt.image.BufferedImage,ai.djl.modality.cv.Joints>

ai.djl.mxnet.zoo.cv.poseestimation.SimplePoseModelLoader

All Implemented Interfaces:

ai.djl.repository.zoo.ModelLoader<java.awt.image.BufferedImage,ai.djl.modality.cv.Joints>

public class SimplePoseModelLoader
extends ai.djl.repository.zoo.BaseModelLoader<java.awt.image.BufferedImage,ai.djl.modality.cv.Joints>

The translator for Simple Pose models.

The model was trained on Gluon and loaded in DJL in MXNet Symbol Block. See Simple Pose.

See Also:

MxSymbolBlock

Field Summary

Fields inherited from class ai.djl.repository.zoo.BaseModelLoader

mrl, repository, version

Constructor Summary

Constructors

Constructor and Description
SimplePoseModelLoader(ai.djl.repository.Repository repository) Creates the Model loader from the given repository.

Method Summary

Modifier and Type	Method and Description
ai.djl.translate.Translator<java.awt.image.BufferedImage,ai.djl.modality.cv.Joints>	getTranslator(ai.djl.repository.Artifact artifact)

Methods inherited from class ai.djl.repository.zoo.BaseModelLoader

listModels, loadModel, loadModel, match, search, toString

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface ai.djl.repository.zoo.ModelLoader

loadModel, loadModel, loadModel, loadModel, loadModel

Constructor Detail

SimplePoseModelLoader

public SimplePoseModelLoader(ai.djl.repository.Repository repository)

Creates the Model loader from the given repository.

Parameters:

repository - the repository to load the model from

Method Detail

getTranslator

public ai.djl.translate.Translator<java.awt.image.BufferedImage,ai.djl.modality.cv.Joints> getTranslator(ai.djl.repository.Artifact artifact)

Specified by:

getTranslator in class ai.djl.repository.zoo.BaseModelLoader<java.awt.image.BufferedImage,ai.djl.modality.cv.Joints>