如何将使用 Mask Rcnn 在自定义对象检测上创建蒙版图像的 Keras 模型转换为 CoreML 模型以在 iOS 应用程序中使用？

Posted 2023-03-27

【中文标题】如何将使用 Mask Rcnn 在自定义对象检测上创建蒙版图像的 Keras 模型转换为 CoreML 模型以在 iOS 应用程序中使用？

【英文标题】：How to convert Keras model which uses Mask Rcnn to create masked image on custom object detection to CoreML model for use in iOS app?

【发布时间】：2020-07-19 14:14:18

【问题描述】：

我开发了一个 Keras 模型，该模型在自定义图像数据集上进行了训练，并且在发送输入图像时它可以很好地进行对象检测和检测到的对象的掩蔽。我想将此模型导出到 CoreML (.mlmodel)，以便我可以将其直接嵌入到我的 ios 应用程序中。

我也有使用 Tensorflow 和 Mask RCNN 训练的模型。因此，将 Keras 或 Tensorflow 模型转换为 CoreML 的任何帮助。我已经尝试并成功地将 Tensorflow 模型转换为 CoreML，但该模型不支持 Mask RCNN。

我正在寻求帮助，将支持 Mask RCNN 的模型转换为 CoreML。

【参考方案1】：

您查看过https://github.com/edouardlp/Mask-RCNN-CoreML 吗？这家伙基本上是在用matterport实现的带有core ml的mask rcnn。