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

Question

我已将PyTorch模型导出到CoreML，并希望快速进行推断。我将输入数据作为2D float32 numpy ndarray <div class="ex-table"> <div class="title">Within the UK</div> <div class="title">Outside the UK</div> <div>Within 1</div> <div>Outside 1</div> <div>Within 2</div> <div>Outside 2</div> <div>Within 3</div> <div>Outside 3</div> <div>Within 4</div> <div>Outside 4</div> </div>存储在磁盘上，需要快速加载到.npy中。有方便的方法吗？

Answer 1

从NumPy中保存原始数据，而不是另存为.npy（已腌制）：

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

现在，您只需将其加载到Swift中的Data对象中，然后将其复制到MLMultiArray中即可。