ai.djl.mxnet.zoo

Class MxModelZoo

java.lang.Object

ai.djl.mxnet.zoo.MxModelZoo

All Implemented Interfaces:

ai.djl.repository.zoo.ModelZoo

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

MxModelZoo is a repository that contains all MXNet models in MxSymbolBlock for DJL.

Field Summary

Fields

Modifier and Type	Field and Description
static ai.djl.modality.cv.zoo.ActionRecognitionModelLoader	ACTION_RECOGNITION
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	ALEXNET
static BertQAModelLoader	BERT_QA
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	DARKNET
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	DENSENET
static GloveWordEmbeddingModelLoader	GLOVE
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	GOOGLENET
static java.lang.String	GROUP_ID
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	INCEPTIONV3
static ai.djl.modality.cv.zoo.InstanceSegmentationModelLoader	MASK_RCNN
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	MLP
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	MOBILENET
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	RESNEST
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	RESNET
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	RESNEXT
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	SE_RESNEXT
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	SENET
static ai.djl.modality.cv.zoo.SimplePoseModelLoader	SIMPLE_POSE
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	SQUEEZENET
static ai.djl.modality.cv.zoo.ObjectDetectionModelLoader	SSD
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	VGG
static ai.djl.modality.cv.zoo.ImageClassificationModelLoader	XCEPTION
static ai.djl.modality.cv.zoo.YoloModelLoader	YOLO

Constructor Summary

Constructors

Constructor and Description
MxModelZoo()

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, hasModelZoo, listModels, listModels, loadModel

Field Detail

GROUP_ID

public static final java.lang.String GROUP_ID

See Also:

Constant Field Values

SSD

public static final ai.djl.modality.cv.zoo.ObjectDetectionModelLoader SSD

YOLO

public static final ai.djl.modality.cv.zoo.YoloModelLoader YOLO

ALEXNET

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader ALEXNET

DARKNET

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader DARKNET

DENSENET

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader DENSENET

GOOGLENET

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader GOOGLENET

INCEPTIONV3

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader INCEPTIONV3

MLP

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader MLP

MOBILENET

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader MOBILENET

RESNEST

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader RESNEST

RESNET

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader RESNET

RESNEXT

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader RESNEXT

SENET

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader SENET

SE_RESNEXT

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader SE_RESNEXT

SQUEEZENET

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader SQUEEZENET

VGG

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader VGG

XCEPTION

public static final ai.djl.modality.cv.zoo.ImageClassificationModelLoader XCEPTION

SIMPLE_POSE

public static final ai.djl.modality.cv.zoo.SimplePoseModelLoader SIMPLE_POSE

MASK_RCNN

public static final ai.djl.modality.cv.zoo.InstanceSegmentationModelLoader MASK_RCNN

ACTION_RECOGNITION

public static final ai.djl.modality.cv.zoo.ActionRecognitionModelLoader ACTION_RECOGNITION

BERT_QA

public static final BertQAModelLoader BERT_QA

GLOVE

public static final GloveWordEmbeddingModelLoader GLOVE

Constructor Detail

MxModelZoo

public MxModelZoo()

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