CoreMLTools转换导致“读取protobuf规范验证器错误时出错”

时间:2018-07-01 21:33:46

标签: tensorflow keras coremltools transfer-learning

我一直在构建自定义卷积网络,该网络已保存到.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)
  • Tensorflow(1.8.0)

对于转化:

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

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

对于任何建议,我将不胜感激。 预先谢谢你!

0 个答案:

没有答案