当我试图将我的Keras模型转换为CoreML模型时,我得到错误'InputLayer object has no attribute'inbound'u nodes'。在
这是我的代码:
loaded_model = load_model("diffinception.h5")
coreml_model = coremltools.converters.keras.convert(loaded_model,
input_names="imageSculp", output_names="category")
coreml_model.save("transfertestinception.mlmodel")
“diffinception.h5”是从Keras导入的一个Inception模型,我为迁移学习培训了额外的层。在
以下是生成该模型的代码:
^{pr2}$我是Keras的最新版本。使用Python2.7
我更新了我机器上的_topology2.py代码以匹配以下版本(2018年1月17日更新):
https://github.com/apple/coremltools/blob/master/coremltools/converters/keras/_topology2.py
这就解决了这个问题。在
相关问题 更多 >
编程相关推荐