ai.djl.basicmodelzoo.cv.object_detection.ssd

Class SsdModelLoader

java.lang.Object

ai.djl.repository.zoo.BaseModelLoader

ai.djl.modality.cv.zoo.ObjectDetectionModelLoader

ai.djl.basicmodelzoo.cv.object_detection.ssd.SsdModelLoader

All Implemented Interfaces:

ai.djl.repository.zoo.ModelLoader

public class SsdModelLoader
extends ai.djl.modality.cv.zoo.ObjectDetectionModelLoader

Model loader for SingleShotDetection(SSD).

Field Summary

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

factories, modelZoo, resource

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
protected ai.djl.Model	createModel(java.lang.String name, ai.djl.Device device, ai.djl.repository.Artifact artifact, java.util.Map<java.lang.String,java.lang.Object> arguments, java.lang.String engine)

Methods inherited from class ai.djl.modality.cv.zoo.ObjectDetectionModelLoader

loadModel, loadModel, loadModel

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

getApplication, getArtifactId, listModels, loadModel, toString

Methods inherited from class java.lang.Object

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

Constructor Detail

SsdModelLoader

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

Creates the Model loader from the given repository.

Parameters:

repository - the repository to load the model from

Method Detail

createModel

protected ai.djl.Model createModel(java.lang.String name,
                                   ai.djl.Device device,
                                   ai.djl.repository.Artifact artifact,
                                   java.util.Map<java.lang.String,java.lang.Object> arguments,
                                   java.lang.String engine)

Overrides:

createModel in class ai.djl.repository.zoo.BaseModelLoader