如何在此项目中实现新的.mlmodel

时间:2018-12-30 09:16:23

标签: ios cgcontext coreml createml

我正在尝试创建一个使用核心ml模型的手写数字识别器。 我正在从另一个类似的项目中获取代码: https://github.com/r4ghu/iOS-CoreML-MNIST

但是我需要将我的ml模型合并到该项目中。 这是我的模型:(输入图像为299x299) https://github.com/LOLIPOP-INTELLIGENCE/createml_handwritten

我的问题是,在类似的项目中要进行哪些更改,以使其融入我的coreml模型

我尝试将形状更改为299x299,但这给了我一个错误

1 个答案:

答案 0 :(得分:0)

viewDidLoad中,您应该将对CVPixelBufferCreate()的呼叫中的数字28更改为299。在原始应用中,mlmodel预期图像为28x28,但您的模型使用299x299图像。

但是,您还需要更改其他内容:将kCVPixelFormatType_OneComponent8替换为kCVPixelFormatType_32BGRAkCVPixelFormatType_32RGBA。原始模型使用灰度图像,但您希望使用彩色图像。

P.S。下次在您的问题中包括实际的错误消息。对于试图回答的人来说,这是重要的信息。 :-)