Skip to content

DJL - TensorFlow model zoo

The TensorFlow model zoo contains symbolic models that can be used for inference. All the models in this model zoo contain pre-trained parameters for their specific datasets.

Documentation

The latest javadocs can be found on the djl.ai website.

You can also build the latest javadocs locally using the following command:

# for Linux/macOS:
./gradlew javadoc

# for Windows:
..\..\gradlew javadoc

The javadocs output is built in the build/doc/javadoc folder.

Installation

You can pull the ai.djl.tensorflow:tensorFlow-model-zoo from the central Maven repository by including the following dependency:

<dependency>
    <groupId>ai.djl.tensorflow</groupId>
    <artifactId>tensorflow-model-zoo</artifactId>
    <version>0.6.0</version>
</dependency>

Pre-trained models

The TensorFlow model zoo contains Computer Vision (CV) models. We currently only support image classification models.

  • CV
  • Image Classification

How to find a pre-trained model in model zoo

Please see DJL Model Zoo