public abstract class ImageClassificationModelLoader
extends ai.djl.repository.zoo.BaseModelLoader<ai.djl.modality.cv.Image,ai.djl.modality.Classifications>
| Constructor and Description |
|---|
ImageClassificationModelLoader(ai.djl.repository.Repository repository,
java.lang.String artifactId,
java.lang.String version)
Creates the Model loader from the given repository.
|
| Modifier and Type | Method and Description |
|---|---|
ai.djl.Application |
getApplication() |
ai.djl.repository.zoo.ZooModel<ai.djl.modality.cv.Image,ai.djl.modality.Classifications> |
loadModel(java.util.Map<java.lang.String,java.lang.String> filters,
ai.djl.Device device,
ai.djl.util.Progress progress)
Loads the model with the given search filters.
|
createModel, getArtifactId, listModels, loadModel, match, toStringpublic ImageClassificationModelLoader(ai.djl.repository.Repository repository,
java.lang.String artifactId,
java.lang.String version)
repository - the repository to load the model fromartifactId - the artifact id of the modelversion - the version number of the modelpublic ai.djl.Application getApplication()
public ai.djl.repository.zoo.ZooModel<ai.djl.modality.cv.Image,ai.djl.modality.Classifications> loadModel(java.util.Map<java.lang.String,java.lang.String> filters,
ai.djl.Device device,
ai.djl.util.Progress progress)
throws java.io.IOException,
ai.djl.repository.zoo.ModelNotFoundException,
ai.djl.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 repositoryai.djl.repository.zoo.ModelNotFoundException - if no model with the specified criteria is foundai.djl.MalformedModelException - if the model data is malformed