【问题标题】:how to pretrain my image using resnet50 in mask-rcnn如何在 mask-rcnn 中使用 resnet50 预训练我的图像
【发布时间】:2018-02-04 18:06:01
【问题描述】:

我正在研究 mask r-cnn。我想知道如何在 mask-rcnn 中使用 resnet50 预训练我的图像(刀、沙发、婴儿、.....)。我很难在 github 中找到它,但我不能。请任何知道如何处理它的人帮助我。

【问题讨论】:

  • 预训练图像是什么意思?这不是一个常用术语。
  • Resn​​et 50 为我提供预训练图像。我认为它提供了掩码 r cnn 中的对象检测。可可数据集图像(完整图像)在 mask r cnn 中用于检测对象。所以我想在 resnet50 中预训练我的图像(例如刀、沙发、婴儿……)而不是预训练图像。 Resnet50 的 pretrining 输出是 cpkg 文件。那么如何通过我的图像在 resnet 中获得预训练的图像???

标签: machine-learning resnet


【解决方案1】:

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

【讨论】:

    【解决方案2】:

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

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

    我个人喜欢这个实现的一点是:

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

    【讨论】:

      猜你喜欢
      • 2020-07-25
      • 2020-03-31
      • 2019-11-17
      • 1970-01-01
      • 1970-01-01
      • 2015-09-09
      • 1970-01-01
      • 1970-01-01
      • 2020-01-11
      相关资源
      最近更新 更多