A B C G O P S T
All Classes All Packages

A

ai.djl.tensorflow.zoo - package ai.djl.tensorflow.zoo

Contains the built-in TfModelZoo.

ai.djl.tensorflow.zoo.cv.objectdetction - package ai.djl.tensorflow.zoo.cv.objectdetction

Contains classes for the Application.CV.OBJECT_DETECTION models in the TfModelZoo.

B

build() - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator.Builder

Builds the translator.

buildBaseTranslator(Model, Map<String, ?>) - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslatorFactory

builder() - Static method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator

Creates a builder to build a TfSSDTranslator.

builder(Map<String, ?>) - Static method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator

Creates a builder to build a TfSSDTranslator with specified arguments.

Builder() - Constructor for class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator.Builder

C

configPostProcess(Map<String, ?>) - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator.Builder

configPreProcess(Map<String, ?>) - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator.Builder

G

getGroupId() - Method in class ai.djl.tensorflow.zoo.TfModelZoo

getModelZoo() - Method in class ai.djl.tensorflow.zoo.TfZooProvider

getSupportedEngines() - Method in class ai.djl.tensorflow.zoo.TfModelZoo

GROUP_ID - Static variable in class ai.djl.tensorflow.zoo.TfModelZoo

O

optBoundingBoxOutputName(String) - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator.Builder

Set the output name used for bounding boxes.

optClassLabelOutputName(String) - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator.Builder

Set the output name used for class label.

optMaxBoxes(int) - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator.Builder

Set the maximum number of bounding boxes to display.

optNumDetectionsOutputName(String) - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator.Builder

Set the output name used for number of detections.

optScoresOutputName(String) - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator.Builder

Set the output name used for detection scores.

P

processInput(TranslatorContext, Image) - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator

processOutput(TranslatorContext, NDList) - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator

S

self() - Method in class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator.Builder

T

TfModelZoo - Class in ai.djl.tensorflow.zoo

TfModelZoo is a repository that contains the TensorFlow models for DJL.

TfSsdTranslator - Class in ai.djl.tensorflow.zoo.cv.objectdetction

A TfSsdTranslator that post-process the NDArray into DetectedObjects with boundaries.

TfSsdTranslator(TfSsdTranslator.Builder) - Constructor for class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator

Creates the SSD translator from the given builder.

TfSsdTranslator.Builder - Class in ai.djl.tensorflow.zoo.cv.objectdetction

The builder for TensorFlow SSD translator.

TfSsdTranslatorFactory - Class in ai.djl.tensorflow.zoo.cv.objectdetction

An TranslatorFactory that creates a TfSsdTranslator instance.

TfSsdTranslatorFactory() - Constructor for class ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslatorFactory

TfZooProvider - Class in ai.djl.tensorflow.zoo

An TensorFlow model zoo provider implements the ZooProvider interface.

TfZooProvider() - Constructor for class ai.djl.tensorflow.zoo.TfZooProvider

A B C G O P S T
All Classes All Packages