如何在 Linux 上的 C++ 代码中使用 ONNX 模型?

问题描述

我在 Pytorch 中训练了一些基于 Unet 的模型。它将图像作为输入,并返回一个掩码。 训练后,我将它保存为 ONNX 格式,使用 onnxruntime python 模块运行它,它就像一个魅力:)

现在,我想在 Linux 的 C++ 代码中使用这个模型。

是否有简单的教程(Hello world)解释:

  • 如何在 Ubuntu 中将 onnxruntime 模块合并到 C++ 程序中 (安装共享库等)?
  • 如何正确加载图像并将其传递给模型?

P.S.我只找到了这个:https://www.onnxruntime.ai/docs/tutorials/samples_catalog.html#cc 但是没有关于加载图像并将其转换为 ONNX - C++ 代码中的兼容格式的信息。

解决方法

Linux 上的安装请参考https://www.onnxruntime.ai/。 您可以参考以下链接获取有关如何加载和运行 ONNX 模型的帮助。 https://github.com/microsoft/onnxruntime/tree/master/csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests.Capi

相关问答

Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其...
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。...
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbc...