如何将Pytorch模型部署到IOS中？

Question

我有一个基于Pytorch的深度学习神经网络，我正在尝试将其部署到IOS上。

Answer 1

我认为仍然不存在本地支持，但是有些工作是导出ONNX模型，然后在支持IOS设备（也包括Android）的Caffe2中打开它。

因此，请使用ONNX export tutorial和此mobile integration helper。

还有一条路径converting ONNX to CoreML，但是根据您的项目，它可能不会很快。

还有一个选项可以将ONNX移植到TensorFlow Lite并通过您的Swift或Objective C应用程序移植到integrate。