【问题标题】:Convert onnx model to keras将 onnx 模型转换为 keras
【发布时间】:2020-02-12 04:36:32
【问题描述】:

我尝试将 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'

【问题讨论】:

    标签: python machine-learning keras onnx


    【解决方案1】:

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

    【讨论】:

      猜你喜欢
      • 2021-11-11
      • 2020-03-31
      • 2022-10-13
      • 2023-01-31
      • 1970-01-01
      • 2018-10-05
      • 2018-11-25
      • 2019-11-10
      • 2022-09-27
      相关资源
      最近更新 更多