时间戳

首页 > 解决方案 > onnx_to_keras：“NotImplementedError：无法修改此类数据”

python - onnx_to_keras：“NotImplementedError：无法修改此类数据”

问题描述

我正在尝试导入一个 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：无法修改此类数据

以下屏幕截图描述了整个错误：

点击查看截图

我已经检查了 input_names 字段是否正确。

标签： pythondeep-learningonnx

解决方案

推荐阅读