如何使用 coremltools 将 matterport mask_rcnn keras(.h5) 模型转换为 coreml 模型(.mlmodel)答案

【问题标题】：How to convert matterport mask_rcnn keras(.h5) model to coreml model(.mlmodel) using coremltools如何使用 coremltools 将 matterport mask_rcnn keras(.h5) 模型转换为 coreml 模型(.mlmodel)
【发布时间】：2020-11-17 04:35:55
【问题描述】：

我使用 matterport maskrcnn 训练了一个模型。我已经有 .h5 模型文件，但我无法将其转换为 .mlmodel。因为涉及到许多自定义层。我已经尝试了我在谷歌上能找到的任何东西。我还尝试了https://github.com/edouardlp/Mask-RCNN-CoreML 进行转换。到目前为止没有成功。

到目前为止，是否有人能够成功完成转换，如果是，您可以分享代码库或教程。

【问题讨论】：

标签： python coreml coremltools

【解决方案1】：

我可以使用问题中提到的相同 github 存储库进行转换。但是你不能在 Xcode 中调试代码，因为 maskrcnn 内存很重。最好使用其他架构，例如 deeplab。

【讨论】：

【解决方案2】：

这是一个带有 MaskRCNN.ml 模型的 github 项目 https://github.com/edouardlp/Mask-RCNN-CoreML/releases/tag/0.2。

注意：您必须将模型复制到项目中才能编译。

【讨论】：