С выпуском новых функций, таких как 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