来自UIImage的MLMultiArray用于sklearn

时间:2018-04-06 14:12:29

标签: scikit-learn coreml coremltools

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

问题是,如何将我的UIImage转换为模型所需的MLMultiArray。 (或者甚至更好,我如何为模型启用图像)

1 个答案:

答案 0 :(得分:0)

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