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