matterport/Mask_RCNN官方教程
这是一个基于python3,keras和tensorflow的mask rcnn模型。这个模型对图像中的每一个目标实例产生候选框和分割掩膜。这个模型基于特征金字塔网络(Feature Pyramid Network, FPN)和 一个ResNet101骨架。
这个版本包括:
- Mask R-CNN基于FPN和ResNet101创建的源代码.
- MS COCO的训练代码
- MS COCO的预训练模型
- jupyter notebook,里面包含了检测步骤的每一步的可视化
- 并行模型类用于多GPU训练
- 在MS COCO上的评价函数
- 用于训练自己的数据的例子
代码以文档形式呈现,容易拓展。如果你使用它们在你的研究当中,请考虑引用这个版本。如果你的研究方向是三维视觉,你可能发现我们最近发布的Matterport3D数据集也很好用。这个数据集收集于我们的愿意开源的客户,用于三维重建。
开始:
- demo.ipynb 最简单的开始方式。展示了使用一个在MS COCO上预训练的模型在你的图片上实现图像分割。包括了在任意图片上实现图像检测和实例分割的代码。