如何将 onnx 格式转换为 ptl 格式？

【问题标题】：How can I convert onnx format to ptl format?如何将 onnx 格式转换为 ptl 格式？
【发布时间】：2021-10-27 22:20:14
【问题描述】：

这里我已经得到了我的.onnx格式文件：retinaface.onnx，我想将它转换为PyTorch移动支持的格式：.ptl或.pt，然后我可以在Android平台上进行推理。但我没有转换或找到相关问题。

// I want to load directly but failed
mRetinafaceDector =  LiteModuleLoader.load(mModelPath + "/retinaface.onnx");

似乎LiteModuleLoader.load() 只能加载.ptl 或.pt，但我手里只有retinaface.onnx 格式..

【问题讨论】：

【解决方案1】：

您无法将 onnx 模型直接加载到 pytorch，因为它尚不支持。

幸运的是，你可以使用onnx2pytorch 工具将 onnx 转换为 pytorch

然后您可以使用以下代码轻松转换它：

onnx_model = onnx.load(path_to_onnx_model)
pytorch_model = ConvertModel(onnx_model)

尽管如此，在 pytorch 中直接导入 onnx 存在一个未解决的问题：

【讨论】：