Как реализовать новую .mlmodel в этом проекте

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

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

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

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


ios coreml createml cgcontext
person Manas Bam - Grade 8    schedule 30.12.2018    source источник

Ответы (1)


arrow_upward
0
arrow_downward

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

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

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

person Matthijs Hollemans    schedule 30.12.2018
comment
Комментарии не подлежат расширенному обсуждению; этот разговор был перешел в чат. - person Bhargav Rao; 02.01.2019