我一直在构建自定义卷积网络,该网络已保存到.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."
我正在冻结原始卷积神经网络的各个层。
我使用的版本是:
对于转化:
coreml_model = coremltools.converters.keras.convert(
pathToh5File,
class_labels=['0','1','2','3','4','5','6','7','8','9']
)
我尝试添加输入名称等等。仍然得到相同的结果。
对于任何建议,我将不胜感激。 预先谢谢你!