PyTorch到CoreML,输入问题

时间:2018-03-09 14:02:18

标签: python ios pytorch coreml

我正在尝试在iOS设备上运行我的PyTorch网络。 但是,当我将网络放入Apple的演示(https://developer.apple.com/documentation/vision/classifying_images_with_vision_and_core_ml)时,我遇到了输入格式不正确的问题,我的输入是MultiArray(Double 3 x 224 x 224),而且是在演示是Image(Color 224 x 224)。

我没有找到一种方法来获得输入图像,我已经完成了所有的研究。 要将我的网络转换为.mlmodel,我将其保存为带有torch.onnx._export的.onnx,然后使用此库https://github.com/onnx/onnx-coreml来获取.mlmodel。

你们中是否有人遇到过同样的问题并有可能的解决方案?

谢谢!

0 个答案:

没有答案