python - 将onnx模型转换为keras

标签 python machine-learning keras onnx

我尝试将 ONNX 模型转换为 Keras,但当我调用转换函数时,我收到以下错误消息“TypeError: unhashable type: 'google.protobuf.pyext._message.RepeatedScalarContainer'”

ONNX 模型输入:input_1

您可以在此处查看 ONNX 模型:https://ibb.co/sKnbxWY

import onnx2keras
from onnx2keras import onnx_to_keras
import keras
import onnx

onnx_model = onnx.load('onnxModel.onnx')
k_model = onnx_to_keras(onnx_model, ['input_1'])

keras.models.save_model(k_model,'kerasModel.h5',overwrite=True,include_optimizer=True)

  File "C:/../onnx2Keras.py", line 7, in <module>
    k_model = onnx_to_keras(onnx_model, ['input_1'])
  File "..\site-packages\onnx2keras\converter.py", line 80, in onnx_to_keras
    weights[onnx_extracted_weights_name] = numpy_helper.to_array(onnx_w)
TypeError: unhashable type: 'google.protobuf.pyext._message.RepeatedScalarContainer'

最佳答案

该问题已在新版本的onnx2keras库中得到解决 您可以在GitHub上查看该问题https://github.com/nerox8664/onnx2keras/issues/23

关于python - 将onnx模型转换为keras,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/58395644/

相关文章:

python - NeuralNetwork 类无法访问其属性

python - 我的 socket 和 pickle 有问题。值​不保存到 txt 中

python - 为什么 igraph 的 add_vertices() 不能处理一组字符串?

machine-learning - sklearn中的层次分类

Python:线性回归, reshape numpy 数组以在模型中使用

python - 使用 Keras Python 创建 RNN

python - 如何使用 matplotlib/numpy 将数组保存为灰度图像?

machine-learning - RNN/LSTM 的非常规使用

python - 如何在 tensorflow keras中重复矩阵?

python - keras LSTM 应该在没有 input_shape 参数的情况下工作吗?