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 ActionRecognitionModelLoader	ACTION_RECOGNITION
static Alexnet	ALEXNET
static BertQAModelLoader	BERT_QA
static Darknet	DARKNET
static Densenet	DENSENET
static GloveWordEmbeddingModelLoader	GLOVE
static Googlenet	GOOGLENET
static java.lang.String	GROUP_ID
static InceptionV3	INCEPTIONV3
static InstanceSegmentationModelLoader	MASK_RCNN
static MlpModelLoader	MLP
static Mobilenet	MOBILENET
static Resnest	RESNEST
static Resnet	RESNET
static Resnext	RESNEXT
static SeResnext	SE_RESNEXT
static Senet	SENET
static SimplePoseModelLoader	SIMPLE_POSE
static Squeezenet	SQUEEZENET
static SingleShotDetectionModelLoader	SSD
static Vgg	VGG
static Xception	XCEPTION
static 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, listModels, loadModel

Field Detail

GROUP_ID

public static final java.lang.String GROUP_ID

See Also:

Constant Field Values

SSD

public static final SingleShotDetectionModelLoader SSD

YOLO

public static final YoloModelLoader YOLO

ALEXNET

public static final Alexnet ALEXNET

DARKNET

public static final Darknet DARKNET

DENSENET

public static final Densenet DENSENET

GOOGLENET

public static final Googlenet GOOGLENET

INCEPTIONV3

public static final InceptionV3 INCEPTIONV3

MLP

public static final MlpModelLoader MLP

MOBILENET

public static final Mobilenet MOBILENET

RESNEST

public static final Resnest RESNEST

RESNET

public static final Resnet RESNET

RESNEXT

public static final Resnext RESNEXT

SENET

public static final Senet SENET

SE_RESNEXT

public static final SeResnext SE_RESNEXT

SQUEEZENET

public static final Squeezenet SQUEEZENET

VGG

public static final Vgg VGG

XCEPTION

public static final Xception XCEPTION

SIMPLE_POSE

public static final SimplePoseModelLoader SIMPLE_POSE

MASK_RCNN

public static final InstanceSegmentationModelLoader MASK_RCNN

ACTION_RECOGNITION

public static final 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