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>

Answer 2

看来你用的是主支行。最近的总分行应该已经解决了这个问题。PyTorch支持不受官方支持(将在4月初发布)。您可以通过签出v0.3.0来尝试稳定的版本。