我可以將在Google Colab的Tensorflow上創建的".py"或".ipynb"模型/檔案轉換為coreML模型/檔案嗎？

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模型作为转换函数的输入。以下是一些示例： 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'](