Question

我在Google Colab的Tensorflow上创建了CNN模型，并将文件下载为“ .py”文件。如何将该文件转换为“ .coreml”文件？

任何建议都非常感谢！

谢谢！

Answer 1

有关如何转换为Core ML的.mlmodel的文档：https://developer.apple.com/documentation/coreml/converting_trained_models_to_core_ml

对于TensorFlow模型，您需要将其转换为frozen.pb模型，作为convert函数的输入。以下是一些示例：https://github.com/tf-coreml/tf-coreml/tree/master/examples

Answer 2

如果您下载了.py文件，则没有下载模型，而是下载了整个Python脚本。

编译模型后，可以使用python软件包coremltools将TensorFlow / Keras模型转换为.mlmodel文件。

import coremltools

core_mlmodel = coremltools.converters.keras.convert(your_keras_model, respect_trainable=True)
core_mlmodel.save("YourModel.mlmodel")

然后，您将在Files的左侧找到您的模型。

Answer 3

@ Sebastian J. Vogt。这是我的转换代码。这是对的吗？谢谢！

import coremltools

category_output_labels = ['dress', 'shirt', 'trousers']
color_output_labels = ['black', 'blue', 'green', 'pink', 'red', 
'white']

coreml_model = coremltools.converters.keras.convert('./fashion_multi_output_v02.h5',

input_names ='图像'，image_input_names ='图像'，class_labels = ['category_output_labels'，'color_output_labels']）

我可以将在Google Colab的Tensorflow中创建的“ .py”或“ .ipynb”模型/文件转换为coreML模型/文件吗？

3 个答案: