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

2022-05-07 14:06:32 浏览数 (1)

译者:冯宝宝

本教程将向您展示如何使用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,但其他版本也应该可以正常工作。

代码语言:javascript复制
python3.6 -m venv venv
source ./venv/bin/activate

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

代码语言:javascript复制
pip install torchvision onnx-coreml

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

阅读全文/改进本文

0 人点赞