如何使用mask-rcnn中的resnet50预览我的图像

时间:2017-08-27 07:17:47

标签: machine-learning resnet

我正在研究面具r-cnn。我想知道如何使用mask-rcnn中的resnet50来预览我的图像(刀,沙发,宝贝,......)。我很难在github中找到它,但我不能。请帮助我知道如何处理它的人。

2 个答案:

答案 0 :(得分:0)

在github here上尝试使用Mask RCNN的这个实现。

答案 1 :(得分:0)

您可以关注Mask_RCNN github repo。它有resnet50和resnet100(这里可能有问题)。我会说这是一个很棒的实现。基本模型来自FAIR(Facebook AI Research)。在开始工作之前,您可以查看演示文件。

如果效果很好,您可以看到我的answer,它会帮助您使用自定义数据训练模型。答案有点长,但它列出了所有步骤。

我个人喜欢这个实现的东西是:

  • 设置简单。不会打扰你的依赖关系。拥有python虚拟环境可以创造奇迹。
  • 它会自动从CPU版本回退到GPU,反之亦然。
  • 得到了开发者的大力支持。它经常提交。
  • 代码非常可定制。所以如果你想做一些改变,这很容易。一些布尔和数字上下变化,你就完成了...... !!!