模型是MobileNet v3 small + 微调分类器，pyTorch训练后导出onnx模型 pom： <dependencies> <dependency> <groupId>com.microsoft.onnxruntime</groupId> <artifactId>onnxruntime</artifactId> <version>1.17.3</version> </dependency> </dependencies> 代码： package org.example; import ai.onnxruntime.OnnxTensor; import ai.onnxruntime.OrtEnvironment; import ai.onnxruntime.OrtSession; import javax.imageio.ImageIO; import java.awt.*; import java.awt.image.BufferedImage; import java.io.File; import java.io.IOException; import java.nio.file.Files; import java.nio.file.Paths; import java.util.HashMap; public class OnnxModelInference { private OrtSession session; public OnnxModelInference(String modelPath) throws Exception { OrtEnvironment env = OrtEnvironment.getEnvironment(); […]