public class SingleShotDetectionModelLoader
extends ai.djl.repository.zoo.BaseModelLoader<ai.djl.modality.cv.Image,ai.djl.modality.cv.output.DetectedObjects>
ModelLoader for Single Shot Detection (SSD) models.
These models were built as part of the Gluon CV library and imported into DJL.
SSD is a model to solve Application.CV#OBJECT_DETECTION. Prior models before SSD
typically used non-deep learning strategies to find object proposals and then would execute
classification at those proposed locations. SSD integrated the proposal process into the model
and then classifies at all of those locations in parallel. This both simplifies the model and
training process while better leveraging the power of model DL engines. [paper]
Today, SSD is not typically used in favor of newer models which outperform it.
| Constructor and Description |
|---|
SingleShotDetectionModelLoader(ai.djl.repository.Repository repository)
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.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.
|
createModel, 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<ai.djl.modality.cv.Image,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 malformed