我正在研究 mask r-cnn。我想知道如何在 mask-rcnn 中使用 resnet50 预训练我的图像（刀、沙发、婴儿、.....）。我很难在 github 中找到它，但我不能。请任何知道如何处理它的人帮助我。
问问题
1180 次
2 回答
-1
你可以关注Mask_RCNN github repo。它同时具有 resnet50 和 resnet100 （这里可能是错误的）。我会说这是一个漂亮的实现。基本模型来自 FAIR（Facebook AI Research）。有一个演示文件，您可以在开始工作之前检查它。
如果效果很好，您可以查看我的答案，它将帮助您使用自定义数据训练模型。答案有点长，但它列出了所有步骤。
我个人喜欢这个实现的一点是：
- 它很容易设置。不会因为依赖关系而困扰您。拥有一个 python 虚拟环境可以创造奇迹。
- 它会自动从 CPU 版本回退到 GPU，反之亦然。
- 它得到了开发人员的良好支持。它经常提交。
- 代码非常可定制。所以如果你想做一些改变，这很容易。一些布尔值和数字上下变化，你就完成了......！！！
于 2018-05-09T15:06:27.547 回答
-1
在 github 上尝试这个 Mask RCNN 的实现。
于 2017-12-12T16:40:42.903 回答