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

How to convert matterport mask_rcnn keras(.h5) model to coreml model(.mlmodel) using coremltools

我使用 matterport maskrcnn 训练了一个模型。我已经有了 .h5 模型文件,但无法将其转换为 .mlmodel。由于涉及许多自定义层。我已经尝试了我在 google 上能找到的关于相同内容的所有内容。我还尝试了 https://github.com/edouardlp/Mask-RCNN-CoreML 进行转换。到目前为止没有成功。

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

我可以使用问题中提到的相同 github 存储库进行转换。但是您不能调试 Xcode 中的代码,因为 maskrcnn 占用大量内存。最好用deeplab之类的其他架构。

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

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