我用 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 以获取依赖项。我可以在我的外部库中看到它:
但是当我在添加 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();
}
}
调用java编译器时需要在类路径中添加依赖(-cp参数),但是从Java11开始可以直接用单源文件调用java解释器(JEP 330):
[主路径] % java -cp "Users\XXXX\.m2 储存器