tensorflow - ONNX 模型转换中“未注册架构”

Question

我正在使用 Kaggle 笔记本。我正在尝试将我的 pytorch 模型转换为tensorflow模型以与tensorflowJS一起运行。我使用下面的代码将onnx模型转换为tensorflow模型-

import onnx
from onnx_tf.backend import prepare
 
onnx_model = onnx.load("../input/onnx-model/model.onnx")
tf_rep = prepare(onnx_model)
tf_rep.export_graph("output/model.pb")

我得到了

SchemaError: No schema registered for 'BitShift'!

我尝试使用 onnx 版本 1.8.1 、 1.8.0 ，然后进一步降级到 1.6.0 。

此外，我尝试直接使用 onnx.js 运行 onnx 模型，但面临图像标准化和调整大小的问题。因此，我决定改用tfjs。

Answer 1

我也遇到了同样的问题。卸载onnx-tf并运行 pip install git+https://github.com/onnx/onnx-tensorflow.git。问题似乎与某些异常类型有关。