尝试将模型从keras导出到CoreML

Question

这是我第一个使用keras的项目。在

这个想法是创建一个模型，可以识别木材的颜色。如果一种颜色是我要识别的颜色，我得到一个1如果不是，我得到一个0。在

我有一个这样的csv文件：

red,green,blue,isWood
63,110,255,0
58,104,255,0
63,112,255,0
96,141,198,1
95,140,197,1
95,138,195,1

我根据我在网上找到的几个教程的组合创建了这个模型。在

^{2}$

这个应用在网上崩溃了

coreml_model.input_description['green'] = 'green value'

出现以下错误消息：

Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/site-packages/coremltools/models/model.py in __setitem__(self, key, value)
     77                 f.shortDescription = value
     78                 return
---> 79         raise AttributeError("No feature with name %s." % key)
     80 
     81     def __iter__(self):

AttributeError: No feature with name green.

我试图将LINE 1修改为

coreml_model = coremltools.converters.keras.convert(model, input_names=['red','green','blue'], output_names='isWood')

或者

coreml_model = coremltools.converters.keras.convert(model, input_names=['red','green','blue'], output_names=['isWood'])

没有区别。在

有什么想法吗？在

Answer 1

{keri>你认为只有一个输入是你的输入。在

尝试将LINE 1替换为以下内容：

coreml_model = coremltools.converters.keras.convert(model, 'rgb-color', 'isWood')