Deep Java Library (DJL) API 介绍

Deep Java Library (DJL) 提供了一套丰富的 API，使得 Java 开发者可以轻松地进行深度学习任务，包括模型加载、训练、推理等。以下是 DJL 的主要模块和 API 的详细介绍。
1. 核心模块
1.1 Model
Model 类是 DJL 中的核心类之一，用于表示深度学习模型。它可以加载预训练模型或自定义模型，并提供训练和推理的方法。

import ai.djl.Model;

// 创建一个空模型
Model model = Model.newInstance("myModel");

// 加载预训练模型
ZooModel<Image, Classifications> loadedModel = Criteria.builder()
    .setTypes(Image.class, Classifications.class)
    .optModelName("resnet18_v1")
    .optEngine("PyTorch")
    .optProgress(new ProgressBar())
    .build()
    .loadModel();

1.2 Predictor
Predictor 类用于执行模型的推理操作。通过 Predictor，可以将输入数据传递给模型并获取预测结果。

import ai.djl.inference.Predictor;

try (Predictor<Image, Classifications> predictor = loadedModel.newPredictor()) {
    Image img = ImageFactory.getInstance().fromFile(Paths.get("path/to/your/image.jpg"));
    Classifications result = predictor.predict(img);
    System.out.println("预测结果: " + result);
}