ios - 如何在这个项目中实现新的 .mlmodel

Question

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

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

我的问题是要在类似项目中进行哪些更改，以便它包含我的 coreml 模型

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

score 0 · Accepted Answer

在viewDidLoad，您应该将呼叫中的数字 28 更改为 299 CVPixelBufferCreate()。在原始应用程序中，mlmodel 需要 28x28 图像，但您的模型使用 299x299 图像。

但是，您还需要更改其他内容：替换kCVPixelFormatType_OneComponent8为kCVPixelFormatType_32BGRAor kCVPixelFormatType_32RGBA。原始模型使用灰度图像，但您需要彩色图像。

PS下次在您的问题中包含实际的错误消息。对于试图回答的人来说，这是一条重要的信息。:-)

1 回答 1