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 BertQAModelLoader	BERT_QA
static java.lang.String	GROUP_ID
static InstanceSegmentationModelLoader	MASK_RCNN
static MlpModelLoader	MLP
static java.lang.String	NAME
static Resnet	RESNET
static Resnext	RESNEXT
static SeResnext	SE_RESNEXT
static Senet	SENET
static SimplePoseModelLoader	SIMPLE_POSE
static Squeezenet	SQUEEZENET
static SingleShotDetectionModelLoader	SSD

Constructor Summary

Constructors

Constructor and Description
MxModelZoo()

Method Summary

Modifier and Type	Method and Description
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, getModelZoo, loadModel

Field Detail

NAME

public static final java.lang.String NAME

See Also:

Constant Field Values

GROUP_ID

public static final java.lang.String GROUP_ID

See Also:

Constant Field Values

MLP

public static final MlpModelLoader MLP

SSD

public static final SingleShotDetectionModelLoader SSD

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

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

Constructor Detail

MxModelZoo

public MxModelZoo()

Method Detail

getSupportedEngines

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

Specified by:

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