如何训练Mask-RCNN获取没有边界框的训练数据?

时间:2018-04-21 02:12:15

标签: tensorflow computer-vision deep-learning

我想在NYU RGBD数据集上训练Mask-RCNN,它不包含边界框。我可以为训练数据创建自己的边界框但是任务太忙了我也只想测试输入数据上的实例分段。那么如何训练没有边界框的训练数据?

2 个答案:

答案 0 :(得分:1)

简单地说,你不能。对象检测算法需要用于训练的边界框。

答案 1 :(得分:-1)

您可以从实例蒙版中提取边界框。在官方MaskRCNN代码存储库中,他们为此here提供了一个代码。