来自 UIImage 的 MLMultiArray 用于 sklearn

Posted 2023-03-28

【中文标题】来自 UIImage 的 MLMultiArray 用于 sklearn

【英文标题】：MLMultiArray from UIImage for sklearn

【发布时间】：2018-09-16 14:25:49

【问题描述】：

我刚刚训练了一个 sklearn 模型并将其转换为 CoreML。问题是，我使用图像作为该模型的输入。但是，模型要求的是MultiArray（即MLMultiArray）。从this question 可以看到coremltools 有一个参数可以让你指定输入是图片。但该参数是用于转换 Keras 模型的，在sklearn 转换工具中，此选项不存在。

问题是，如何将我的 UIImage 转换为模型所需的 MLMultiArray。 （或者更好的是，如何为模型启用图像）

【参考方案1】：

您可以将 UIImage 转换为字节值，然后将这些字节加载到 MLMultiArray 中。问题是：MLMultiArray 是什么形状，它的值应该在什么范围内？

【讨论】：

它是 32x32x3，它在 RGB 层中，每个层都有一层。