ios – 为Keras模型将UIImage转换为MLMultiArray

概述在Python中,我使用keras训练了一个图像分类模型,以接收作为[224,224,3]数组的输入并输出预测(1或0).当我加载保存模型并将其加载到xcode时,它声明输入必须是MLMultiArray格式. 有没有办法让我将UIImage转换为MLMultiArray格式？或者有没有办法让我改变我的keras模型接受CVPixelBuffer类型对象作为输入. 在Core ML转换脚本中,您可 在Python中,我使用keras训练了一个图像分类模型,以接收作为[224,224,3]数组的输入并输出预测(1或0).当我加载保存模型并将其加载到xcode时,它声明输入必须是MLMultiArray格式.

有没有办法让我将UIImage转换为MLMultiArray格式？或者有没有办法让我改变我的keras模型接受CVPixelBuffer类型对象作为输入.

解决方法 在Core ML转换脚本中,您可以提供参数image_input_names =’data’,其中data是输入的名称.

现在Core ML将此输入视为图像(CVPixelBuffer)而不是多数组.

总结

以上是内存溢出为你收集整理的ios – 为Keras模型将UIImage转换为MLMultiArray全部内容，希望文章能够帮你解决ios – 为Keras模型将UIImage转换为MLMultiArray所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。