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

发布时间：2020-12-15 01:42:02 所属栏目：百科来源：网络整理

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

在Python中,我使用keras训练了一个图像分类模型,以接收作为[224,224,3]数组的输入并输出预测(1或0).当我加载保存模型并将其加载到xcode时,它声明输入必须是MLMultiArray格式.

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

解决方法

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

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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!