我正在尝试创建一个使用核心ml模型的手写数字识别器。 我正在从另一个类似的项目中获取代码: https://github.com/r4ghu/iOS-CoreML-MNIST
但是我需要将我的ml模型合并到该项目中。 这是我的模型:(输入图像为299x299) https://github.com/LOLIPOP-INTELLIGENCE/createml_handwritten
我的问题是,在类似的项目中要进行哪些更改,以使其融入我的coreml模型
我尝试将形状更改为299x299,但这给了我一个错误
答案 0 :(得分:0)
在viewDidLoad
中,您应该将对CVPixelBufferCreate()
的呼叫中的数字28更改为299。在原始应用中,mlmodel预期图像为28x28,但您的模型使用299x299图像。
但是,您还需要更改其他内容:将kCVPixelFormatType_OneComponent8
替换为kCVPixelFormatType_32BGRA
或kCVPixelFormatType_32RGBA
。原始模型使用灰度图像,但您希望使用彩色图像。
P.S。下次在您的问题中包括实际的错误消息。对于试图回答的人来说,这是重要的信息。 :-)