无法在JAVA中导入onnx

Question

我用 JAVA 的工作不多，但我需要加载一个用 python 训练的模型，并检查我是否可以用 JAVA 进行推理。我正在尝试在 JAVA 中加载一个 onnx 文件。为此，我在 JAVA 中导入了 onnx，但它抛出了一个包不存在的错误。

test_first % javac src/main/java/org/example/Main.java
src/main/java/org/example/Main.java:3: error: package ai.onnxruntime.OrtSession.Result does not exist
import ai.onnxruntime.OrtSession.Result.*;
^
1 error

我已将 onnx 添加为我的代码的依赖项：

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>org.example</groupId>
    <artifactId>test_first</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <maven.compiler.source>11</maven.compiler.source>
        <maven.compiler.target>11</maven.compiler.target>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    </properties>

    <dependencies>
        <!-- https://mvnrepository.com/artifact/com.microsoft.onnxruntime/onnxruntime -->
        <dependency>
            <groupId>com.microsoft.onnxruntime</groupId>
            <artifactId>onnxruntime</artifactId>
            <version>1.13.1</version>
        </dependency>
    </dependencies>

</project>

在此之后运行 pom.xml 以获取依赖项。我可以在我的外部库中看到它： directory structure image

但是，当我在添加 import 语句后尝试编译 java 代码时，出现了上述错误。

我也试过跑步：

mvn install:install-file -Dfile=/Users/XXXX/.m2/repository/com/microsoft/onnxruntime/onnxruntime/1.13.1/onnxruntime-1.13.1.jar -DgroupId=com.microsoft.onnxruntime -DartifactId=onnxruntime -Dversion=1.13.1 -Dpackaging=jar -DgeneratePom=true

但它说工件已经存在于本地存储库中。

我正在运行以下代码来检查导入是否成功。

package org.example;


import ai.onnxruntime.OrtEnvironment;

public class Main {
    public static void main(String[] args) {
        var env = OrtEnvironment.getEnvironment();
    }
}

Answer 1

我不完全确定这是否有帮助，但是将导入从 import ai.onnxruntime.OrtSession.Result.*; 更改为 import ai.onnxruntime.OrtSession.Result; 对我有用。