ios - 使用 coremltools 3 将 TF 2.* keras 模型转换为 iOS 12 的 mlmodel

Question

我有一个keras模型 ( TF 2.*)，我想将其转换为mlmodeliOS 版本 12。

根据coremltools 4 发行说明：“要将 Core ML 模型部署到 iOS12、macOS 10.13、watchOS 5、tvOS 12 或更早版本的目标，请使用 coremltools 3 和 tfcoreml 1。”

tfcoreml仅适用于TF 1.*，所以我知道我需要使用coremltools version 3.*.

我正在尝试coremltools version 3.2，我想使用这个功能[coremltools.converters.keras.convert][2]。

我从一个简单的尝试开始，使用带有coremltools==3.2, tensorflow==2.4.1,的新鲜 venv keras==2.4.3：

base_model = load_model(keras_model_p)
coreml_model = coremltools.converters.keras.convert(model)

但我收到以下错误：

TypeError: Keras layer of type <class 'tensorflow.python.keras.engine.functional.Functional'> is not supported.

我试图改变功能部分，从：

base_model2 = load_model(keras_model_p)
top_model2 = Sequential()
top_model2.add(Permute((3, 1, 2)))
model2 = Model(inputs=base_model2.input, outputs=top_model2(base_model2.output))

至

base_model = load_model(keras_model_p)
model = Sequential()
model.add(base_model)
model.add(Permute((3, 1, 2)))

并收到错误： TypeError: 'InputLayer' object is not iterable

第二次尝试是使用第一个 TF 2.* 版本：tensorflow==2.0.0和keras==2.3.0.

我执行相同但收到：

ValueError: Keras layer '<class 'tensorflow.python.keras.engine.input_layer.InputLayer'>' not supported. 

我该如何解决？

谢谢

标签： iostensorflowkerascoremltoolsmlmodel