PyTorch 1.0 中文官方教程：ONNX 现场演示教程

译者：冯宝宝

本教程将向您展示如何使用ONNX将已从PyTorch导出的神经模型传输模型转换为Apple CoreML格式。这将允许您在Apple设备上轻松运行深度学习模型，在这种情况下，可以从摄像机直播演示。

什么是ONNX

ONNX（开放式神经网络交换）是一种表示深度学习模型的开放格式。借助ONNX，AI开发人员可以更轻松地在最先进的工具之间移动模型，并选择最适合它们的组合。ONNX由合作伙伴社区开发和支持。 您可以访问 onnx.ai，了解有关ONNX的更多信息以及支持的工具。

教程预览

本教程将带你走过如下主要4步：

1. 下载（或训练）Pytorch风格装换模型
2. 将PyTorch模型转换至ONNX模型
3. 将ONNX模型转换至CoreML模型
4. 在支持风格转换iOS App中运行CoreML模型

环境准备

我们将在虚拟环境工作，以避免与您的本地环境冲突。在本教程中使用Python 3.6，但其他版本也应该可以正常工作。

python3.6 -m venv venv
source ./venv/bin/activate

我们需要安装Pytorch和 onnx->coreml 转换器：

pip install torchvision onnx-coreml

如果要在iPhone上运行iOS样式传输应用程序，还需要安装XCode。您也可以在Linux中转换模型，但要运行iOS应用程序本身，您将需要一台Mac。

阅读全文／改进本文