import cv2
import numpy as np
import onnxruntime as ort
def load_model(model_path):
"""加载ONNX模型"""
session = ort.InferenceSession(model_path)
return session
def preprocess_image(image_path):
image = cv2.imread(image_path)[......]
#include <opencv2/opencv.hpp>
#include <onnxruntime/onnxruntime_cxx_api.h>
#include <vector>
#include <iostream>
int main() {
// load onnx model
Ort::Env env(OrtLoggingLevel::ORT_LOGGING_LEVEL_WARNING, "test"[......]
模型是MobileNet v3 small + 微调分类器，pyTorch训练后导出onnx模型
pom：
<dependencies>
<dependency>
<groupId>com.microsoft.onnxruntime</groupId>
<artifactId>onnxruntime</artifactId>
<version>[......]