Как реализовать новый .mlmodel в этом проекте - PullRequest
0 голосов
/ 30 декабря 2018

Я пытаюсь создать распознаватель рукописных цифр, который использует модель core ml.Я беру код из другого подобного проекта: https://github.com/r4ghu/iOS-CoreML-MNIST

Но мне нужно включить мою модель ml в этот проект.Это моя модель: (Исходное изображение 299x299) https://github.com/LOLIPOP-INTELLIGENCE/createml_handwritten

Мой вопрос заключается в том, какие изменения необходимо внести в аналогичный проект, чтобы он включал мою модель coreml

Я попытался изменитьформирует 299x299, но это дает мне ошибку

1 Ответ

0 голосов
/ 30 декабря 2018

В viewDidLoad вы должны изменить номер 28 на 299 в вызове на CVPixelBufferCreate().В исходном приложении mlmodel ожидает изображение 28x28, но ваша модель использует изображения 299x299.

Однако есть еще кое-что, что вам также нужно изменить: замените kCVPixelFormatType_OneComponent8 на kCVPixelFormatType_32BGRA или kCVPixelFormatType_32RGBA.Оригинальная модель использует изображения в градациях серого, но ваша ожидает цветные изображения.

PS В следующий раз включите фактическое сообщение об ошибке в ваш вопрос.Это важная информация для людей, которые пытаются ответить.: -)

...