使用 ONNX 将 Pytorch 模型转换为 Keras

Question

我想根据这篇中等文章使用 onnx 将 pytorch 模型转换为 keras：

https://medium.com/analytics-vidhya/pytorch-to-keras-using-onnx-71d98258ad76

我复制了与本文相同的代码，但是当我想将 onnx 模型转换为 keras 时，我遇到了以下错误：

ValueError: 'onnx::Add_6_reshape/' 不是有效的根范围名称。根范围名称必须匹配以下模式：^[A-Za-z0-9.][A-Za-z0-9_.\/>-]*$

有人知道我该如何修复它吗？

Answer 1

取决于您要尝试转换的模型，但您可以尝试以下操作：

1. 安装pt2keras：https://github.com/JWLee89/pt2keras

pip install -U pt2keras

2. 使用下面的脚本转换模型（已测试）

pip install -U pt2keras

我是该软件包的作者，这主要是在我的空闲时间快速完成的，但希望这对您有用。

我在本地环境中测试了上面的代码，它对我有用。