public interface MxModelZoo
MxSymbolBlock for DJL.| Modifier and Type | Field and Description |
|---|---|
static ActionRecognitionModelLoader |
ACTION_RECOGNITION |
static BertQAModelLoader |
BERT_QA |
static java.lang.String |
GROUP_ID |
static InstanceSegmentationModelLoader |
MASK_RCNN |
static Mlp |
MLP |
static java.lang.String |
MXNET_REPO_URL |
static ai.djl.repository.Repository |
REPOSITORY |
static Resnet |
RESNET |
static Resnext |
RESNEXT |
static SeResnext |
SE_RESNEXT |
static Senet |
SENET |
static SimplePoseModelLoader |
SIMPLE_POSE |
static Squeezenet |
SQUEEZENET |
static SingleShotDetectionModelLoader |
SSD |
| Modifier and Type | Method and Description |
|---|---|
static <I,O> ai.djl.repository.zoo.ModelLoader<I,O> |
getModelLoader(java.lang.String name)
Gets the
ModelLoader based on the model name. |
static java.util.List<ai.djl.repository.zoo.ModelLoader<?,?>> |
listModels()
Lists the available models in the ModelZoo.
|
static final java.lang.String MXNET_REPO_URL
static final ai.djl.repository.Repository REPOSITORY
static final java.lang.String GROUP_ID
static final Mlp MLP
static final SingleShotDetectionModelLoader SSD
static final Resnet RESNET
static final Resnext RESNEXT
static final Senet SENET
static final SeResnext SE_RESNEXT
static final Squeezenet SQUEEZENET
static final SimplePoseModelLoader SIMPLE_POSE
static final InstanceSegmentationModelLoader MASK_RCNN
static final ActionRecognitionModelLoader ACTION_RECOGNITION
static final BertQAModelLoader BERT_QA
static <I,O> ai.djl.repository.zoo.ModelLoader<I,O> getModelLoader(java.lang.String name)
throws ai.djl.repository.zoo.ModelNotFoundException
ModelLoader based on the model name.I - the input data type for preprocessingO - the output data type after postprocessingname - the name of the modelModelLoader of the modelai.djl.repository.zoo.ModelNotFoundException - when the model cannot be foundstatic java.util.List<ai.djl.repository.zoo.ModelLoader<?,?>> listModels()