Могу ли я преобразовать модель / файл ".py" или ".ipynb", созданную в Tensorflow Google Colab, в модель / файл coreML?

Question

Я создал модель CNN в Tensorflow Google Colab и загрузил файл в виде файла «.py».Как я могу преобразовать этот файл в файл ".coreml"?

Любой совет очень ценится!

Спасибо!

Answer 1

Если вы скачали файл .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 2

Документация о том, как преобразовать в Core ML .mlmodel s: 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