С выпуском новых функций, таких как CoreML, которые упростили преобразование модели машинного обучения в модель coreML.

Доступная модель машинного обучения, которую можно преобразовать в модель coreML.

Примечание. В качестве примера использовался набор данных Cat and Dog.

Шаги, которые необходимо предпринять

1 . Создайте модель (например, model.h5)

2. Преобразуйте model.h5 в coreML (например, model.h5 в MyModel.mlmodel)

3 . Добавьте преобразованный файл .mlmodel в xcode, т.е. MyModel.mlmodel в xcode.

Шаг 1 (Создание модели)

Здесь использовалась предварительно обученная модель (VGG16).

Шаг 2 (преобразование model.h5 в MyModel.mlmodel)

pip установите coremtools, если у вас его нет

Примечание: важными строками здесь являются первая и последняя строки, которые

coreml_model = coremltools.converters.keras.convert('model.h5', input_names=['image'], output_names=['output'],image_input_names='image')

и

coreml_model.save('MyModel.mlmodel')

Шаг 3

Чтобы проверить успешность добавления, вы можете нажать на модель MyModel.ml, подробности модели будут отображаться справа в вашем xcode.

Чтобы получить полный доступ к быстрому коду, нажмите на ссылку ниже.

https://github.com/Abdullah8888/CatAndDogKerasModelToMlModel