public class SingleShotDetectionModelLoader
extends ai.djl.repository.zoo.BaseModelLoader<java.awt.image.BufferedImage,ai.djl.modality.cv.output.DetectedObjects>
| Constructor and Description |
|---|
SingleShotDetectionModelLoader(ai.djl.repository.Repository repository)
Creates the Model loader from the given repository.
|
| Modifier and Type | Method and Description |
|---|---|
protected ai.djl.Model |
createModel(ai.djl.Device device,
ai.djl.repository.Artifact artifact,
java.util.Map<java.lang.String,java.lang.Object> arguments) |
ai.djl.Application |
getApplication() |
ai.djl.repository.zoo.ZooModel<java.awt.image.BufferedImage,ai.djl.modality.cv.output.DetectedObjects> |
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.
|
getArtifactId, listModels, loadModel, match, toStringpublic SingleShotDetectionModelLoader(ai.djl.repository.Repository repository)
repository - the repository to load the model frompublic ai.djl.Application getApplication()
public ai.djl.repository.zoo.ZooModel<java.awt.image.BufferedImage,ai.djl.modality.cv.output.DetectedObjects> 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 malformedprotected ai.djl.Model createModel(ai.djl.Device device,
ai.djl.repository.Artifact artifact,
java.util.Map<java.lang.String,java.lang.Object> arguments)
createModel in class ai.djl.repository.zoo.BaseModelLoader<java.awt.image.BufferedImage,ai.djl.modality.cv.output.DetectedObjects>