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

Question

我在Pytorch上构建了一个深度学习神经网络，我正寻求将其部署到IOS上。

Answer 1

我认为原生支持还不存在，但有些人所做的是导出ONNX模型，然后在支持IOS设备(也支持Androids)的Caffe2中打开它。

所以使用ONNX export tutorial和这个mobile integration helper。

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

还有一个选项是将ONNX移植到Swift或Objective C应用程序的TensorFlow Lite和integrate上。