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

时间:2019-08-06 08:29:27

标签: object-detection amazon-sagemaker mxnet coreml coremltools

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

也许有更好的方法可以做到? 有人知道实现此任务的便捷方法吗?

我的模型包含两个文件:

  • resnet50_ssd_model-symbol.json
  • resnet50_ssd_model-0000.params

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

mxnet-to-coreml转换器失败,出现以下错误:

  

提高TypeError(“不支持%s类型的MXNet层。”%层)       TypeError:不支持_copy类型的MXNet层。

我这样调用转换器脚本:

=C5/B5 <= 1

1 个答案:

答案 0 :(得分:1)

here所述,从MXNet到CoreML的转换器尚不支持此层(“ _copy”)(它是MXNet运算符)。有一个功能要求。