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 file

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

Answer 2

Кажется, вы используете основную ветку. Последняя ветка master должна была исправить эту проблему. Поддержка PyTorch официально не поддерживается (будет выпущена в начале апреля). Вы можете попробовать стабильную версию с помощью checkout v0.3.0.