问题描述
我在 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