Конвертировать модель onnx в керасы - PullRequest
Новая
       33

Конвертировать модель onnx в керасы

1 голос
/ 15 октября 2019

Я пытаюсь преобразовать модель 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'

1 Ответ

0 голосов
/ 16 октября 2019

Проблема была решена в новой версии библиотеки onnx2keras. Вы можете увидеть проблему на GitHub https://github.com/nerox8664/onnx2keras/issues/23

...