gpt4 book ai didi

ios - 将 UIImage 转换为 Keras 模型的 MLMultiArray

转载 作者:可可西里 更新时间:2023-11-01 06:18:24 25 4
gpt4 key购买 nike

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

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

最佳答案

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

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

关于ios - 将 UIImage 转换为 Keras 模型的 MLMultiArray,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44529869/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com