TypeError:"InputLayer"对象不可与CoreMLTools迭代 [英] TypeError: 'InputLayer' object is not iterable with CoreMLTools
本文介绍了TypeError:"InputLayer"对象不可与CoreMLTools迭代的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
我正在尝试将VGG模型转换为coremltools.当我运行以下代码以转换模型时:
I'm trying to convert a VGG model to coremltools. When I run the following code to convert the model:
with CustomObjectScope({'relu6': keras.layers.ReLU,'DepthwiseConv2D': keras.layers.DepthwiseConv2D}):
from keras.models import load_model
import coremltools
model_directory = 'KerasModels/VGG-7-3-20_13categories.h5'
keras_model = load_model(model_directory)
input_layer = InputLayer(input_shape=(224, 224, 3), name="input_1")
# Save and convert :
keras_model.layers[0] = input_layer
keras_model.save(model_directory)
print("Changed2")
your_model = coremltools.converters.keras.convert(model_directory, input_names=['image'], output_names=['output'], image_input_names='image')
your_model.save('RecycleNet.mlmodel')
我收到以下错误:
TypeError: 'InputLayer' object is not iterable
我应该如何将该模型转换为coremltools?谢谢
How should I go about converting this model to coremltools? Thanks
推荐答案
我通过从以下方式切换来解决了该错误:
I fixed this error by switching from using:
coremltools.converters.keras
收件人:
coremltools.converters.tensorflow
这篇关于TypeError:"InputLayer"对象不可与CoreMLTools迭代的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文