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