无法将.h5 model 转换为.onnx 文件

Question

我想将带有 weights.h5 文件的 model 转换为 .onnx 文件。

我使用 TensorFlow/Keras 创建并训练了 model，但我想在 c# 应用程序中使用它，所以我需要将其转换为可以在 c# 中使用的格式。 （如果您有其他建议，请与我分享）

这是我的代码：

import os
import onnx
import keras2onnx

from tensorflow.python.keras import backend as K
from tensorflow.python.keras.models import load_model


os.environ['TF_KERAS'] = '1'

onnx_model_name = "weights_d.onnx"
model = load_model("weightFinal12_d.h5")
onnx_model = keras2onnx.convert_keras(model, model.name)
onnx.save_model(onnx_model, onnx_model_name)

但我收到此错误：

文件“C:\ProgramData\Anaconda3\envs\keras2onnx\lib\site-packages\tensorflow\python\keras\utils\generic_utils.py”，第 181 行，在 deserialize_keras_object 配置、module_objects、custom_objects、printable_module_name）文件“C:\ ProgramData\Anaconda3\envs\keras2onnx\lib\site-packages\tensorflow\python\keras\utils\generic_utils.py"，第 166 行，在 class_and_config_for_serialized_keras_object 中引发 ValueError('Unknown' + printable_module_name + ': ' + class_name) ValueError: Unknown层：功能性