Index (Deep Java Library 0.25.0 - pytorch-model-zoo API)

A B C G N P S
All Classes All Packages

A

ai.djl.pytorch.zoo - package ai.djl.pytorch.zoo: Contains the built-in PtModelZoo.
ai.djl.pytorch.zoo.cv.objectdetection - package ai.djl.pytorch.zoo.cv.objectdetection: Contains classes for the Application.CV.OBJECT_DETECTION models in the PtModelZoo.
ai.djl.pytorch.zoo.nlp.qa - package ai.djl.pytorch.zoo.nlp.qa: Contains classes for the Application.NLP.QUESTION_ANSWER models in the PtModelZoo.
ai.djl.pytorch.zoo.nlp.sentimentanalysis - package ai.djl.pytorch.zoo.nlp.sentimentanalysis: Contains classes for the Application.NLP.SENTIMENT_ANALYSIS models in the PtModelZoo.
ai.djl.pytorch.zoo.nlp.textgeneration - package ai.djl.pytorch.zoo.nlp.textgeneration: Contains classes for the Application.NLP.TEXT_GENERATION models.

B

build() - Method in class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator.Builder: Builds the translator.
build() - Method in class ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslator.Builder: Builds the translator.
buildBaseTranslator(Model, Map<String, ?>) - Method in class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslatorFactory
builder() - Static method in class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator: Creates a builder to build a PtSSDTranslatorBuilder.
builder() - Static method in class ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslator: Creates a builder to build a PtBertQATranslator.
builder(Map<String, ?>) - Static method in class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator: Creates a builder to build a PtSSDTranslatorBuilder with specified arguments.
builder(Map<String, ?>) - Static method in class ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslator: Creates a builder to build a PtSSDTranslatorBuilder with specified arguments.
Builder() - Constructor for class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator.Builder
Builder() - Constructor for class ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslator.Builder

C

configPostProcess(Map<String, ?>) - Method in class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator.Builder
configPreProcess(Map<String, ?>) - Method in class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator.Builder

G

getGroupId() - Method in class ai.djl.pytorch.zoo.PtModelZoo
getModelZoo() - Method in class ai.djl.pytorch.zoo.PtZooProvider
getSupportedEngines() - Method in class ai.djl.pytorch.zoo.PtModelZoo
getSupportedTypes() - Method in class ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslatorFactory
getSupportedTypes() - Method in class ai.djl.pytorch.zoo.nlp.sentimentanalysis.PtDistilBertTranslatorFactory
getSupportedTypes() - Method in class ai.djl.pytorch.zoo.nlp.textgeneration.PtGptTranslatorFactory
GROUP_ID - Static variable in class ai.djl.pytorch.zoo.PtModelZoo

N

newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslatorFactory
newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.pytorch.zoo.nlp.sentimentanalysis.PtDistilBertTranslatorFactory
newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.pytorch.zoo.nlp.textgeneration.PtGptTranslatorFactory

P

prepare(TranslatorContext) - Method in class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator
prepare(TranslatorContext) - Method in class ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslator
prepare(TranslatorContext) - Method in class ai.djl.pytorch.zoo.nlp.sentimentanalysis.PtDistilBertTranslator
processInput(TranslatorContext, QAInput) - Method in class ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslator
processInput(TranslatorContext, NDList) - Method in class ai.djl.pytorch.zoo.nlp.textgeneration.PtGptTranslator
processInput(TranslatorContext, String) - Method in class ai.djl.pytorch.zoo.nlp.sentimentanalysis.PtDistilBertTranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.pytorch.zoo.nlp.sentimentanalysis.PtDistilBertTranslator
processOutput(TranslatorContext, NDList) - Method in class ai.djl.pytorch.zoo.nlp.textgeneration.PtGptTranslator
PtBertQATranslator - Class in ai.djl.pytorch.zoo.nlp.qa: The Translator for PyTorch Question Answering model.
PtBertQATranslator.Builder - Class in ai.djl.pytorch.zoo.nlp.qa: The builder for Bert QA translator.
PtBertQATranslatorFactory - Class in ai.djl.pytorch.zoo.nlp.qa: An TranslatorFactory that creates a PtBertQATranslator instance.
PtBertQATranslatorFactory() - Constructor for class ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslatorFactory
PtDistilBertTranslator - Class in ai.djl.pytorch.zoo.nlp.sentimentanalysis: The Translator for PyTorch Sentiment Analysis model.
PtDistilBertTranslator() - Constructor for class ai.djl.pytorch.zoo.nlp.sentimentanalysis.PtDistilBertTranslator
PtDistilBertTranslatorFactory - Class in ai.djl.pytorch.zoo.nlp.sentimentanalysis: An TranslatorFactory that creates a PtDistilBertTranslator instance.
PtDistilBertTranslatorFactory() - Constructor for class ai.djl.pytorch.zoo.nlp.sentimentanalysis.PtDistilBertTranslatorFactory
PtGptTranslator - Class in ai.djl.pytorch.zoo.nlp.textgeneration: The Translator for PyTorch GPT2 model.
PtGptTranslator(long, int, int) - Constructor for class ai.djl.pytorch.zoo.nlp.textgeneration.PtGptTranslator: Constructs a new instance of PtGptTranslator.
PtGptTranslatorFactory - Class in ai.djl.pytorch.zoo.nlp.textgeneration: An TranslatorFactory that creates a PtGptTranslator instance.
PtGptTranslatorFactory() - Constructor for class ai.djl.pytorch.zoo.nlp.textgeneration.PtGptTranslatorFactory
PtModelZoo - Class in ai.djl.pytorch.zoo: PtModelZoo is a repository that contains all PyTorch models in PtSymbolBlock for DJL.
PtSsdTranslator - Class in ai.djl.pytorch.zoo.cv.objectdetection: A PtSsdTranslator that post-process the NDArray into DetectedObjects with boundaries.
PtSsdTranslator(PtSsdTranslator.Builder) - Constructor for class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator: Creates the SSD translator from the given builder.
PtSsdTranslator.Builder - Class in ai.djl.pytorch.zoo.cv.objectdetection: The builder for SSD translator.
PtSsdTranslatorFactory - Class in ai.djl.pytorch.zoo.cv.objectdetection: An TranslatorFactory that creates a PtSsdTranslator instance.
PtSsdTranslatorFactory() - Constructor for class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslatorFactory
PtZooProvider - Class in ai.djl.pytorch.zoo: An PyTorch model zoo provider implements the ZooProvider interface.
PtZooProvider() - Constructor for class ai.djl.pytorch.zoo.PtZooProvider

S

self() - Method in class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator.Builder
self() - Method in class ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslator.Builder: Returns the builder.
setBoxes(int, int[], int[], int[], int[][]) - Method in class ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator.Builder: Set the box parameter to reconstruct the anchor box.

A B C G N P S
All Classes All Packages