遮罩R-CNN用于目标检测和分割
这是基于Python 3,Keras和TensorFlow的的实现。该模型为图像中对象的每个实例生成边界框和分割蒙版。它基于功能金字塔网络(FPN)和ResNet101主干网。
该存储库包括:
基于FPN和ResNet101构建的Mask R-CNN的源代码。
MS COCO的培训代码
MS COCO的预训练砝码
Jupyter笔记本电脑可在每个步骤可视化检测管线
用于Multi-GPU训练的ParallelModel类
评估MS COCO指标(AP)
在自己的数据集上进行训练的示例
该代码已记录并设计为易于扩展。如果您在研究中使用它,请考虑引用该存储库(下面的bibtex)。如果您从事3D视觉工作,您可能会发现我们最近发布的数据集也很有用。此数据集是由我们的客户捕获的3D重构空间创建的,这些客户同意将其公开提供给学术用途。您可以看到更多示例。
入门
是
2021-12-21 20:06:02
73.68MB
Python
1