2

我一直致力于构建一个自定义卷积网络，该网络已保存到 .h5 文件中。另外一个我通过弹出最后一层（FC）来应用迁移学习，然后用新数据编译模型。再次以 .h5 格式保存模型。当我尝试将此模型转换为 mlModel 格式时会出现问题。我收到以下错误：

return _MLModelProxy(filename)
RuntimeError: Error compiling model: "Error reading protobuf spec. validator error: Layer 'conv2d_2__activation__' consumes a layer named 'conv2d_2__activation___output' which is not present in this network."

我正在冻结原始卷积神经网络的层。我正在使用的版本是：

  • 喀拉斯 (2.1.6)
  • Protobuf(3.6.0)
  • 张量流（1.8.0）

对于转换：

coreml_model = coremltools.converters.keras.convert(
    pathToh5File,
    class_labels=['0','1','2','3','4','5','6','7','8','9']
)

我试过添加输入名称等等。仍然得到相同的结果。

我将不胜感激任何建议。先感谢您！

4

0 回答 0