onnx_to_keras: "NotImplementedError: невозможно изменить этот тип данных" - PullRequest
0 голосов
/ 25 мая 2020

Я пытаюсь импортировать модель onnx, а затем преобразовать ее в Keras.

Ниже приведен фрагмент кода, который я использую для сохранения модели в формате onnx:

import onnxmltools

onnx_model = onnxmltools.convert_keras(model, model.name)
onnxmltools.utils.save_model(onnx_model, 'gdrive/My Drive/Model/RNN_comfort_model.onnx')

Ниже приведен фрагмент кода, который я использую для импорта и преобразования:

import onnx
import onnxmltools
from onnx2keras import onnx_to_keras

onnx_model = onnx.load('gdrive/My Drive/Model/RNN_comfort_model.onnx')
k_model = onnx_to_keras(onnx_model, ['lstm_5_input'], name_policy='renumerate')

Когда я запускаю последний фрагмент кода, я получаю

NotImplementedError: не могу изменить этот тип данных

На следующем снимке экрана описана вся ошибка:

Click to see the screenshot

Я уже проверил правильность поля input_names.

...