ONNX系列五 --- 在C＃中使用可移植的ONNX AI模型

目录

安装和导入ONNX运行时

载入ONNX模型

使用ONNX运行时进行预测

摘要和后续步骤

参考文献

---

• 下载源547.1 KB

系列文章列表如下：

ONNX系列一 --- 带有ONNX的便携式神经网络

ONNX系列二 --- 使用ONNX使Keras模型可移植

ONNX系列三 --- 使用ONNX使PyTorch AI模型可移植

ONNX系列四 --- 使用ONNX使TensorFlow模型可移植

ONNX系列五 --- 在C＃中使用可移植的ONNX AI模型

ONNX系列六 --- 在Java中使用可移植的ONNX AI模型

ONNX系列七 --- 在Python中使用可移植的ONNX AI模型

在关于2020年使用便携式神经网络的系列文章中，您将学习如何在x64架构上安装ONNX并在C＃中使用它。

微软与Facebook和AWS共同开发了ONNX。ONNX格式和ONNX运行时都具有行业支持，以确保所有重要框架都能够将其图形导出到ONNX，并且这些模型可以在任何硬件配置上运行。

ONNX Runtime是用于运行已转换为ONNX格式的机器学习模型的引擎。传统机器学习模型和深度学习模型（神经网络）都可以导出为ONNX格式。运行时可以在Linux，Windows和Mac上运行，并且可以在各种芯片体系结构上运行。它还可以利用诸如GPU和TPU之类的硬件加速器。但是，没有针对操作系统，芯片体系结构和加速器的每种组合的安装包，因此，如果不使用任何一种常见组合，则可能需要从源代码构建运行时。检查ONNX运行时网站以获得所需组合的安装说明。本文将展示如何在具有默认CPU的x64体系结构和具有GPU的x64体系结构上安装ONNX Runtime。

除了可以在许多硬件配置上运行之外，还可以从大多数流行的编程语言中调用运行时。本文的目的是展示如何在C＃中使用ONNX Runtime。我将展示如何安装onnxruntime软件包。安装ONNX Runtime后，我会将先前导出的MNIST模型加载到ONNX Runtime中，并使用它进行预测。

安装和导入ONNX运行时

在使用ONNX Runtime之前，您需要安装Microsoft.ML.OnnxRuntime，它是一个NuGet软件包。如果尚未安装.NET CLI，则还需要安装它。以下命令将运行时安装在具有默认CPU的x64体系结构上：

dotnet add package microsoft.ml.onnxruntime

要将运行时安装在带有GPU的x64架构上，请使用以下命令：

dotnet add package microsoft.ml.onnxruntime.gpu

一旦安装了运行时，就可以使用以下using语句将其导入到C＃代码文件中：

using Microsoft.ML.OnnxRuntime;
using Microsoft.ML.OnnxRuntime.Tensors;

引入Tensor工具的using语句将帮助我们为ONNX模型创建输入，并解释ONNX模型的输出（预测）。

载入ONNX模型

下面的代码段显示了如何将ONNX模型加载到以C＃运行的ONNX运行时中。此代码创建可用于进行预测的会话对象。这里使用的模型是从PyTorch导出的ONNX模型。

这里有几件事值得注意。首先，您需要查询会话以获取其输入。这是使用会话的InputMetadata属性完成的。我们的MNIST模型只有一个输入参数：784个浮点数组，代表MNIST数据集中的一张图像。如果您的模型有多个输入参数，则InputMetadata的每个参数都有一个条目。

Utilities.LoadTensorData();
string modelPath = Directory.GetCurrentDirectory() + @"/pytorch_mnist.onnx";

using (var session = new InferenceSession(modelPath))
{
   float[] inputData = Utilities.ImageData[imageIndex];
   string label = Utilities.ImageLabels[imageIndex];
   Console.WriteLine("Selected image is the number: " + label);

   var inputMeta = session.InputMetadata;
   var container = new List();

   foreach (var name in inputMeta.Keys)
   {
      var tensor = new DenseTensor(inputData, inputMeta[name].Dimensions);
      container.Add(NamedOnnxValue.CreateFromTensor(name, tensor));
   }

   // Run code omitted for brevity.
}

上面的代码未显示用于读取原始MNIST图像并将每个图像转换为784个浮点数组的实用程序。还可以从MNIST数据集中读取每个图像的标签，以便可以确定预测的准确性。该代码是标准的.NET代码，但仍然鼓励您检出并使用它。如果您需要读入与MNIST数据集相似的图像，它将节省您的时间。

使用ONNX运行时进行预测

以下函数显示了如何使用在加载ONNX模型时创建的ONNX会话：

{
   // Load code not shown for brevity.

   // Run the inference
   using (var results = session.Run(container))
   {
      // Get the results
      foreach (var r in results)
      {
         Console.WriteLine("Output Name: {0}", r.Name);
         int prediction = MaxProbability(r.AsTensor());
         Console.WriteLine("Prediction: " + prediction.ToString());
      }
   }
}

大多数神经网络不会直接返回预测。它们返回每个输出类的概率列表。对于我们的MNIST模型，每个图像的返回值将是10个概率的列表。可能性最高的条目是预测。您可以做的一个有趣的测试是，将ONNX模型在创建模型的框架中运行时返回的概率与从原始模型返回的概率进行比较。理想情况下，模型格式和运行时的更改不应更改所产生的任何概率。这将使每当模型发生更改时都可以运行良好的单元测试。

摘要和后续步骤

在本文中，我简要介绍了ONNX运行时和ONNX格式。然后，我展示了如何在ONNX Runtime中使用C＃加载和运行ONNX模型。

本文的代码示例包含一个工作的控制台应用程序，该应用程序演示了此处显示的所有技术。此代码示例是GitHub存储库的一部分，该存储库探讨了使用神经网络预测MNIST数据集中发现的数字。具体来说，有一些示例显示了如何在Keras，PyTorch，TensorFlow 1.0和TensorFlow 2.0中创建神经网络。

如果您想了解有关导出为ONNX格式和使用ONNX Runtime的更多信息，请查阅本系列的其他文章。

参考文献

• https://docs.microsoft.com/zh-cn/dotnet/core/tutorials/with-visual-studio-code
• https://docs.microsoft.com/zh-cn/nuget/quickstart/install-and-use-a-package-using-the-dotnet-cli
• https://code.visualstudio.com/docs/setup/mac
• https://docs.microsoft.com/zh-cn/dotnet/core/tools/dotnet
• https://microsoft.github.io/onnxruntime/
• https://github.com/microsoft/onnxruntime/blob/master/docs/CSharp_API.md#getting-started
• https://github.com/keithpij/onnx-lab