Skip navigation links
A B C E F G I L M O P R S U V 

A

ai.djl.basicmodelzoo - package ai.djl.basicmodelzoo
Contains the built-in BasicModelZoo.
ai.djl.basicmodelzoo.basic - package ai.djl.basicmodelzoo.basic
Contains the basic built-in models.
ai.djl.basicmodelzoo.cv.classification - package ai.djl.basicmodelzoo.cv.classification
Contains classes for the Application.CV.IMAGE_CLASSIFICATION models in the BasicModelZoo.
ai.djl.basicmodelzoo.cv.object_detection.ssd - package ai.djl.basicmodelzoo.cv.object_detection.ssd
Contains classes for the Application.CV.OBJECT_DETECTION models in the BasicModelZoo.
ai.djl.basicmodelzoo.nlp.embedding - package ai.djl.basicmodelzoo.nlp.embedding
Contains classes for the natural language processing section (Application.NLP) of the BasicModelZoo.

B

BasicModelZoo - Class in ai.djl.basicmodelzoo
Model Zoo is a repository that contains all models for DJL.
BasicModelZoo() - Constructor for class ai.djl.basicmodelzoo.BasicModelZoo
 
BasicZooProvider - Class in ai.djl.basicmodelzoo
An imperative model zoo provider implements the ZooProvider interface.
BasicZooProvider() - Constructor for class ai.djl.basicmodelzoo.BasicZooProvider
 
build() - Method in class ai.djl.basicmodelzoo.cv.classification.ResNetV1.Builder
Builds a ResNetV1 block.
build() - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection.Builder
Builds a SingleShotDetection block.
builder() - Static method in class ai.djl.basicmodelzoo.cv.classification.ResNetV1
Creates a builder to build a ResNetV1.
builder() - Static method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
Creates a builder to build a SingleShotDetection.

C

close() - Method in class ai.djl.basicmodelzoo.nlp.embedding.ModelZooWordEmbedding
 
createModel(Device, Artifact, Map<String, Object>) - Method in class ai.djl.basicmodelzoo.cv.classification.MlpModelLoader
 
createModel(Device, Artifact, Map<String, Object>) - Method in class ai.djl.basicmodelzoo.cv.classification.ResNetModelLoader
 
createModel(Device, Artifact, Map<String, Object>) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetectionModelLoader
 

E

embedWord(NDArray) - Method in class ai.djl.basicmodelzoo.nlp.embedding.ModelZooWordEmbedding
 

F

forward(ParameterStore, NDList, PairList<String, Object>) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
 

G

getAnchorPredictionBlock(int) - Static method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
Creates a anchor prediction block used in an SSD.
getApplication() - Method in class ai.djl.basicmodelzoo.cv.classification.MlpModelLoader
getApplication() - Method in class ai.djl.basicmodelzoo.cv.classification.ResNetModelLoader
getApplication() - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetectionModelLoader
getChildren() - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
 
getClassPredictionBlock(int, int) - Static method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
Creates a class prediction block used in an SSD.
getDirectParameters() - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
 
getDownSamplingBlock(int) - Static method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
Creates a Block that reduces the size of a convolutional block by half.
getGroupId() - Method in class ai.djl.basicmodelzoo.BasicModelZoo
getModelZoo() - Method in class ai.djl.basicmodelzoo.BasicZooProvider
getOutputShapes(NDManager, Shape[]) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
 
getParameterShape(String, Shape[]) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
 
getSupportedEngines() - Method in class ai.djl.basicmodelzoo.BasicModelZoo
GROUP_ID - Static variable in class ai.djl.basicmodelzoo.BasicModelZoo
 

I

initialize(NDManager, DataType, Shape...) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
 

L

loadModel(Map<String, String>, Device, Progress) - Method in class ai.djl.basicmodelzoo.cv.classification.MlpModelLoader
Loads the model with the given search filters.
loadModel(Map<String, String>, Device, Progress) - Method in class ai.djl.basicmodelzoo.cv.classification.ResNetModelLoader
Loads the model with the given search filters.
loadModel(Map<String, String>, Device, Progress) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetectionModelLoader
Loads the model with the given search filters.
loadParameters(NDManager, DataInputStream) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
 

M

Mlp - Class in ai.djl.basicmodelzoo.basic
Multilayer Perceptron (MLP) NeuralNetworks.
Mlp(int, int, int[]) - Constructor for class ai.djl.basicmodelzoo.basic.Mlp
Create an MLP NeuralNetwork using RELU.
Mlp(int, int, int[], Function<NDList, NDList>) - Constructor for class ai.djl.basicmodelzoo.basic.Mlp
Create an MLP NeuralNetwork.
MLP - Static variable in class ai.djl.basicmodelzoo.BasicModelZoo
 
MlpModelLoader - Class in ai.djl.basicmodelzoo.cv.classification
Model loader for MLP models.
MlpModelLoader(Repository) - Constructor for class ai.djl.basicmodelzoo.cv.classification.MlpModelLoader
Creates the Model loader from the given repository.
ModelZooWordEmbedding - Class in ai.djl.basicmodelzoo.nlp.embedding
A WordEmbedding using a ZooModel.
ModelZooWordEmbedding(ZooModel<NDList, NDList>) - Constructor for class ai.djl.basicmodelzoo.nlp.embedding.ModelZooWordEmbedding
Constructs a ModelZooWordEmbedding.

O

optBatchNormMomemtum(float) - Method in class ai.djl.basicmodelzoo.cv.classification.ResNetV1.Builder
Sets the size of the output.
optFeatures(List<Block>) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection.Builder
Sets the Conv2D blocks to be appended to the network to get multi-output network.
optGlobalPool(boolean) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection.Builder
Sets the boolean whether to attach a global average pooling layer as the last output layer.

P

preprocessWordToEmbed(NDManager, String) - Method in class ai.djl.basicmodelzoo.nlp.embedding.ModelZooWordEmbedding
 

R

residualUnit(int, Shape, boolean, boolean, float) - Static method in class ai.djl.basicmodelzoo.cv.classification.ResNetV1
Builds a Block that represents a residual unit used in the implementation of the Resnet model.
RESNET - Static variable in class ai.djl.basicmodelzoo.BasicModelZoo
 
resnet(ResNetV1.Builder) - Static method in class ai.djl.basicmodelzoo.cv.classification.ResNetV1
Creates a new Block of ResNetV1 with the arguments from the given ResNetV1.Builder.
ResNetModelLoader - Class in ai.djl.basicmodelzoo.cv.classification
Model loader for ResNet_V1.
ResNetModelLoader(Repository) - Constructor for class ai.djl.basicmodelzoo.cv.classification.ResNetModelLoader
Creates the Model loader from the given repository.
ResNetV1 - Class in ai.djl.basicmodelzoo.cv.classification
ResNetV1 contains a generic implementation of ResNet adapted from https://github.com/tornadomeet/ResNet/blob/master/symbol_resnet.py (Original author Wei Wu) by Antti-Pekka Hynninen.
ResNetV1.Builder - Class in ai.djl.basicmodelzoo.cv.classification
The Builder to construct a ResNetV1 object.

S

saveParameters(DataOutputStream) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection
 
setBaseNetwork(Block) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection.Builder
Sets the base network for the SSD framework.
setImageShape(Shape) - Method in class ai.djl.basicmodelzoo.cv.classification.ResNetV1.Builder
Sets the shape of the image.
setNumClasses(int) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection.Builder
Sets the number of classes of objects to be detected.
setNumFeatures(int) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection.Builder
Sets the number of down sampling blocks to be applied.
setNumLayers(int) - Method in class ai.djl.basicmodelzoo.cv.classification.ResNetV1.Builder
Sets the number of layers in the network.
setOutSize(long) - Method in class ai.djl.basicmodelzoo.cv.classification.ResNetV1.Builder
Sets the size of the output.
setRatios(List<List<Float>>) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection.Builder
Sets the list of aspect ratios of generated anchor boxes.
setSizes(List<List<Float>>) - Method in class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection.Builder
Sets the list of sizes of generated anchor boxes.
SingleShotDetection - Class in ai.djl.basicmodelzoo.cv.object_detection.ssd
SingleShotDetection is an implementation of Block that implements a Single Shot Detection (SSD) model for object detection.
SingleShotDetection.Builder - Class in ai.djl.basicmodelzoo.cv.object_detection.ssd
The Builder to construct a SingleShotDetection.
SingleShotDetectionModelLoader - Class in ai.djl.basicmodelzoo.cv.object_detection.ssd
Model loader for SingleShotDetection(SSD).
SingleShotDetectionModelLoader(Repository) - Constructor for class ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetectionModelLoader
Creates the Model loader from the given repository.
SSD - Static variable in class ai.djl.basicmodelzoo.BasicModelZoo
 

U

unembedWord(NDArray) - Method in class ai.djl.basicmodelzoo.nlp.embedding.ModelZooWordEmbedding
 

V

vocabularyContains(String) - Method in class ai.djl.basicmodelzoo.nlp.embedding.ModelZooWordEmbedding
 
A B C E F G I L M O P R S U V 
Skip navigation links