public class SimplePoseModelLoader extends BaseModelLoader
The model was trained on Gluon and loaded in DJL in MXNet Symbol Block. See Simple Pose.
factories, modelZoo, resource
Constructor and Description |
---|
SimplePoseModelLoader(Repository repository,
java.lang.String groupId,
java.lang.String artifactId,
java.lang.String version,
ModelZoo modelZoo)
Creates the Model loader from the given repository.
|
Modifier and Type | Method and Description |
---|---|
ZooModel<Image,Joints> |
loadModel()
Loads the model.
|
ZooModel<Image,Joints> |
loadModel(java.util.Map<java.lang.String,java.lang.String> filters,
Device device,
ai.djl.util.Progress progress)
Loads the model with the given search filters.
|
ZooModel<Image,Joints> |
loadModel(ai.djl.util.Progress progress)
Loads the model.
|
createModel, getApplication, getArtifactId, listModels, loadModel, toString
public SimplePoseModelLoader(Repository repository, java.lang.String groupId, java.lang.String artifactId, java.lang.String version, ModelZoo modelZoo)
repository
- the repository to load the model fromgroupId
- the group id of the modelartifactId
- the artifact id of the modelversion
- the version number of the modelmodelZoo
- the modelZoo type that is being used to get supported engine typespublic ZooModel<Image,Joints> loadModel() throws MalformedModelException, ModelNotFoundException, java.io.IOException
java.io.IOException
- for various exceptions loading data from the repositoryModelNotFoundException
- if no model with the specified criteria is foundMalformedModelException
- if the model data is malformedpublic ZooModel<Image,Joints> loadModel(ai.djl.util.Progress progress) throws MalformedModelException, ModelNotFoundException, java.io.IOException
progress
- the progress tracker to update while loading the modeljava.io.IOException
- for various exceptions loading data from the repositoryModelNotFoundException
- if no model with the specified criteria is foundMalformedModelException
- if the model data is malformedpublic ZooModel<Image,Joints> loadModel(java.util.Map<java.lang.String,java.lang.String> filters, Device device, ai.djl.util.Progress progress) throws java.io.IOException, ModelNotFoundException, MalformedModelException
filters
- the search filters to match against the loaded modeldevice
- the device the loaded model should useprogress
- the progress tracker to update while loading the modeljava.io.IOException
- for various exceptions loading data from the repositoryModelNotFoundException
- if no model with the specified criteria is foundMalformedModelException
- if the model data is malformed