gpt4 book ai didi

swift - 快速从 .npy 转换为 MLMultiArray 以进行 CoreML 预测

转载 作者:行者123 更新时间:2023-12-03 08:44:57 25 4
gpt4 key购买 nike

我已将 PyTorch 模型导出到 CoreML,并希望快速进行推理。我将输入数据作为 2D float32 numpy ndarray .npy 存储在磁盘上,并需要快速加载到 MLMultiArray 中。有没有方便的方法来做到这一点?

最佳答案

不要保存为 .npy(已腌制),而是保存 NumPy 中的原始数据:

array.astype(np.float32).tofile(filename)

现在您可以简单地将其加载到 Swift 中的 Data 对象中,然后将其复制到 MLMultiArray 中。

关于swift - 快速从 .npy 转换为 MLMultiArray 以进行 CoreML 预测,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/61854333/

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