将经过培训的AWS SageMaker MXNet模型部署/转换为iOS设备

Question

我使用SageMaker对象检测算法训练了MXnet SSD resnet-50模型，并希望在iOS设备上使用它。因此，我需要将其转换为Apple CoreML格式。我尝试使用mxnet来实现coreml。

也许有更好的方法？有人知道一种方便的方法来完成这项任务吗？

我的模型由两个文件组成：

• resnet50\U ssd\U型号符号。json

在转换模型之前，我使用MXnet提供的“deploy.py”将其设置为部署状态。

mxnet到coreml转换器失败，出现以下错误：

引发TypeError（"不支持类型为%s的MXNet层。"%层）TypeError：不支持类型为_copy的MXNet层。

我这样调用转换器脚本：

python mxnet_coreml_converter.py --model-prefix='ssd_resnet50_512' -- 
epoch=0 --input-shape='{"data":"3, 512, 512"}' --mode=classifier --pre- 
processing-arguments='{"image_input_names":"data"}' --output- 
file="resnet50.mlmodel"

Answer 1

如本文所述，从MXNet到CoreML的转换器尚不支持该层（“_copy”）（即MXNet操作符）。对此有一个功能请求。