训练后如何保存 MASK RCNN 模型？

Question

我正在使用 Matterport 存储库在自定义数据集上训练 MASK RCNN。我在训练中取得了成功。现在我想保存经过训练的模型并在 Web 应用程序中使用它来检测对象。训练后如何保存mask rcnn模型？请指导我。

存储库链接：

https://github.com/matterport/Mask_RCNN

Answer 1

在“加载模型”主题下的Inspect_model.ipynb笔记本中，您可以在以推理模式加载模型后保存它。