无法在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以获取依赖项。我可以看到它在我的外部库:目录结构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

在调用java编译器(-cp参数)时，您需要将依赖项添加到类路径中，但是自java 11以来，您可以直接使用单个源文件(JEP 330)调用java解释器。

对于Windows，你可以这样写

[path-to-main] % java -cp "C:\Users\XXXX\.m2\repository\com\microsoft\onnxruntime\onnxruntime\1.13.1\onnxruntime-1.13.1.jar" Main.java