Skip navigation links
A B F G I L M P Q R S T 

A

ACTION_RECOGNITION - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 
ActionRecognitionModelLoader - Class in ai.djl.mxnet.zoo.cv.actionrecognition
Model loader for Action Recognition models.
ActionRecognitionModelLoader(Repository) - Constructor for class ai.djl.mxnet.zoo.cv.actionrecognition.ActionRecognitionModelLoader
Creates the Model loader from the given repository.
ai.djl.mxnet.zoo - package ai.djl.mxnet.zoo
Contains the built-in MxModelZoo.
ai.djl.mxnet.zoo.cv.actionrecognition - package ai.djl.mxnet.zoo.cv.actionrecognition
Contains classes for the MRL.Model.CV.ACTION_RECOGNITION models in the MxModelZoo.
ai.djl.mxnet.zoo.cv.classification - package ai.djl.mxnet.zoo.cv.classification
Contains classes for the MRL.Model.CV.IMAGE_CLASSIFICATION models in the MxModelZoo.
ai.djl.mxnet.zoo.cv.objectdetection - package ai.djl.mxnet.zoo.cv.objectdetection
Contains classes for the MRL.Model.CV.OBJECT_DETECTION models in the MxModelZoo.
ai.djl.mxnet.zoo.cv.poseestimation - package ai.djl.mxnet.zoo.cv.poseestimation
Contains classes for the MRL.Model.CV.POSE_ESTIMATION models in the MxModelZoo.
ai.djl.mxnet.zoo.cv.segmentation - package ai.djl.mxnet.zoo.cv.segmentation
Contains classes for the MRL.Model.CV.INSTANCE_SEGMENTATION models in the MxModelZoo.
ai.djl.mxnet.zoo.nlp.qa - package ai.djl.mxnet.zoo.nlp.qa
Contains classes for the MRL.Model.NLP.QUESTION_ANSWER models in the MxModelZoo.

B

BERT_QA - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 
BertDataParser - Class in ai.djl.mxnet.zoo.nlp.qa
This is the Utility for pre-processing data for the Bert Model.
BertDataParser() - Constructor for class ai.djl.mxnet.zoo.nlp.qa.BertDataParser
 
BertQAModelLoader - Class in ai.djl.mxnet.zoo.nlp.qa
Model loader for BERT QA models.
BertQAModelLoader(Repository) - Constructor for class ai.djl.mxnet.zoo.nlp.qa.BertQAModelLoader
Creates the Model loader from the given repository.
BertQATranslator - Class in ai.djl.mxnet.zoo.nlp.qa
The translator for BertQAModelLoader.

F

formTokens(List<String>, List<String>, int) - Static method in class ai.djl.mxnet.zoo.nlp.qa.BertDataParser
Forms tokens with separation that can be used for BERT.

G

getBatchifier() - Method in class ai.djl.mxnet.zoo.nlp.qa.BertQATranslator
getModelLoader(String) - Static method in interface ai.djl.mxnet.zoo.MxModelZoo
Gets the ModelLoader based on the model name.
getParagraph() - Method in class ai.djl.mxnet.zoo.nlp.qa.QAInput
Gets the resource document that contains the answer.
getQuestion() - Method in class ai.djl.mxnet.zoo.nlp.qa.QAInput
Gets the question for the model.
getSeqLength() - Method in class ai.djl.mxnet.zoo.nlp.qa.QAInput
Gets the sequence length.
getTokenTypes(List<String>, List<String>, int) - Static method in class ai.djl.mxnet.zoo.nlp.qa.BertDataParser
Forms the token types List [0000...1111...000] where all questions are 0 and answers are 1.
getTranslator(Artifact) - Method in class ai.djl.mxnet.zoo.cv.actionrecognition.ActionRecognitionModelLoader
getTranslator(Artifact) - Method in class ai.djl.mxnet.zoo.cv.classification.ImageClassificationModelLoader
getTranslator(Artifact) - Method in class ai.djl.mxnet.zoo.cv.objectdetection.SingleShotDetectionModelLoader
getTranslator(Artifact) - Method in class ai.djl.mxnet.zoo.cv.poseestimation.SimplePoseModelLoader
getTranslator(Artifact) - Method in class ai.djl.mxnet.zoo.cv.segmentation.InstanceSegmentationModelLoader
getTranslator(Artifact) - Method in class ai.djl.mxnet.zoo.nlp.qa.BertQAModelLoader
GROUP_ID - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 

I

idx2token(List<Integer>) - Method in class ai.djl.mxnet.zoo.nlp.qa.BertDataParser
Converts indexes to tokens.
ImageClassificationModelLoader - Class in ai.djl.mxnet.zoo.cv.classification
Model loader for Image Classification models.
ImageClassificationModelLoader(Repository, String, String) - Constructor for class ai.djl.mxnet.zoo.cv.classification.ImageClassificationModelLoader
Creates the Model loader from the given repository.
InstanceSegmentationModelLoader - Class in ai.djl.mxnet.zoo.cv.segmentation
Model loader for Instance Segmentation models.
InstanceSegmentationModelLoader(Repository) - Constructor for class ai.djl.mxnet.zoo.cv.segmentation.InstanceSegmentationModelLoader
Creates the Model loader from the given repository.

L

listModels() - Static method in interface ai.djl.mxnet.zoo.MxModelZoo
Lists the available models in the ModelZoo.

M

MASK_RCNN - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 
Mlp - Class in ai.djl.mxnet.zoo.cv.classification
Model loader for MLP Symbolic models.
Mlp(Repository) - Constructor for class ai.djl.mxnet.zoo.cv.classification.Mlp
Creates the Model loader from the given repository.
MLP - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 
MxModelZoo - Interface in ai.djl.mxnet.zoo
MxModelZoo is a repository that contains all MXNet models in MxSymbolBlock for DJL.
MXNET_REPO_URL - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 

P

pad(List<E>, E, int) - Static method in class ai.djl.mxnet.zoo.nlp.qa.BertDataParser
Pads the tokens to the required length.
parse(InputStream) - Static method in class ai.djl.mxnet.zoo.nlp.qa.BertDataParser
Parses the Vocabulary to JSON files.
processInput(TranslatorContext, QAInput) - Method in class ai.djl.mxnet.zoo.nlp.qa.BertQATranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.mxnet.zoo.nlp.qa.BertQATranslator

Q

QAInput - Class in ai.djl.mxnet.zoo.nlp.qa
The input container for BERT QA model.
QAInput(String, String, int) - Constructor for class ai.djl.mxnet.zoo.nlp.qa.QAInput
Creates the BERT QA model.

R

REPOSITORY - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 
Resnet - Class in ai.djl.mxnet.zoo.cv.classification
Model loader for Resnet Symbolic models.
Resnet(Repository) - Constructor for class ai.djl.mxnet.zoo.cv.classification.Resnet
Creates the Model loader from the given repository.
RESNET - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 
Resnext - Class in ai.djl.mxnet.zoo.cv.classification
Model loader for Resnext Symbolic models.
Resnext(Repository) - Constructor for class ai.djl.mxnet.zoo.cv.classification.Resnext
Creates the Model loader from the given repository.
RESNEXT - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 

S

SE_RESNEXT - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 
Senet - Class in ai.djl.mxnet.zoo.cv.classification
Model loader for Senet Symbolic models.
Senet(Repository) - Constructor for class ai.djl.mxnet.zoo.cv.classification.Senet
Creates the Model loader from the given repository.
SENET - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 
SeResnext - Class in ai.djl.mxnet.zoo.cv.classification
Model loader for SeResnext Symbolic models.
SeResnext(Repository) - Constructor for class ai.djl.mxnet.zoo.cv.classification.SeResnext
Creates the Model loader from the given repository.
SIMPLE_POSE - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 
SimplePoseModelLoader - Class in ai.djl.mxnet.zoo.cv.poseestimation
The translator for Simple Pose models.
SimplePoseModelLoader(Repository) - Constructor for class ai.djl.mxnet.zoo.cv.poseestimation.SimplePoseModelLoader
Creates the Model loader from the given repository.
SingleShotDetectionModelLoader - Class in ai.djl.mxnet.zoo.cv.objectdetection
Model loader for Single Shot Detection (SSD) models.
SingleShotDetectionModelLoader(Repository) - Constructor for class ai.djl.mxnet.zoo.cv.objectdetection.SingleShotDetectionModelLoader
Creates the Model loader from the given repository.
Squeezenet - Class in ai.djl.mxnet.zoo.cv.classification
Model loader for Squeezenet Symbolic models.
Squeezenet(Repository) - Constructor for class ai.djl.mxnet.zoo.cv.classification.Squeezenet
Creates a new instance of SqueezeNet.
SQUEEZENET - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 
SSD - Static variable in interface ai.djl.mxnet.zoo.MxModelZoo
 

T

token2idx(List<String>) - Method in class ai.djl.mxnet.zoo.nlp.qa.BertDataParser
Converts tokens to indexes.
tokenizer(String) - Static method in class ai.djl.mxnet.zoo.nlp.qa.BertDataParser
Tokenizes the input, splits all kinds of whitespace, and separates the end of sentence symbol.
A B F G I L M P Q R S T 
Skip navigation links