译者:冯宝宝
本教程将向您展示如何使用ONNX将已从PyTorch导出的神经模型传输模型转换为Apple CoreML格式。这将允许您在Apple设备上轻松运行深度学习模型,在这种情况下,可以从摄像机直播演示。
什么是ONNX
ONNX(开放式神经网络交换)是一种表示深度学习模型的开放格式。借助ONNX,AI开发人员可以更轻松地在最先进的工具之间移动模型,并选择最适合它们的组合。ONNX由合作伙伴社区开发和支持。 您可以访问 onnx.ai,了解有关ONNX的更多信息以及支持的工具。
教程预览
本教程将带你走过如下主要4步:
- 下载(或训练)Pytorch风格装换模型
- 将PyTorch模型转换至ONNX模型
- 将ONNX模型转换至CoreML模型
- 在支持风格转换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。
阅读全文/改进本文