ai.djl.basicmodelzoo

Class BasicModelZoo

java.lang.Object

ai.djl.basicmodelzoo.BasicModelZoo

All Implemented Interfaces:

ai.djl.repository.zoo.ModelZoo

public class BasicModelZoo
extends java.lang.Object
implements ai.djl.repository.zoo.ModelZoo

Model Zoo is a repository that contains all models for DJL.

Field Summary

Fields

Modifier and Type	Field and Description
static java.lang.String	GROUP_ID
static MlpModelLoader	MLP
static ResNetModelLoader	RESNET
static SingleShotDetectionModelLoader	SSD

Constructor Summary

Constructors

Constructor and Description
BasicModelZoo()

Method Summary

Modifier and Type	Method and Description
java.lang.String	getGroupId()
java.util.Set<java.lang.String>	getSupportedEngines()

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.repository.zoo.ModelZoo

getModelLoader, getModelLoaders, listModels, loadModel

Field Detail

GROUP_ID

public static final java.lang.String GROUP_ID

See Also:

Constant Field Values

RESNET

public static final ResNetModelLoader RESNET

MLP

public static final MlpModelLoader MLP

SSD

public static final SingleShotDetectionModelLoader SSD

Constructor Detail

BasicModelZoo

public BasicModelZoo()

Method Detail

getGroupId

public java.lang.String getGroupId()

Specified by:

getGroupId in interface ai.djl.repository.zoo.ModelZoo

getSupportedEngines

public java.util.Set<java.lang.String> getSupportedEngines()

Specified by:

getSupportedEngines in interface ai.djl.repository.zoo.ModelZoo