我正在研究 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 回答