java - 找不到 ai.djl.pytorch.zoo.PtZooProvider

Question

我正在测试框架 djl https://djl.ai/

当我尝试加载模型时:ModelZoo.loadModel(criteria)

我收到此错误:

java.util.ServiceConfigurationError: ai.djl.repository.zoo.ZooProvider: Provider ai.djl.pytorch.zoo.PtZooProvider not found

Criteria<BufferedImage, DetectedObjects> criteria =
                Criteria.builder()
                        .optApplication(Application.CV.OBJECT_DETECTION)
                        .setTypes(BufferedImage.class, DetectedObjects.class)
                        .optFilter("size", "512")
                        .optFilter("backbone", "resnet50")
                        .optFilter("flavor", "v1")
                        .optFilter("dataset", "voc")
                        .optProgress(new ProgressBar())
                        .build();

Answer 1

终于找到解决办法了。我只是将其添加到我的 Maven pom.xml 文件中

    <dependency>
        <groupId>ai.djl.pytorch</groupId>
        <artifactId>pytorch-native-auto</artifactId>
        <version>1.4.0-SNAPSHOT</version>
    </dependency>