我们可以使用RGB图像训练模型的权重来绘制灰度图像吗?

时间:2017-06-21 06:53:22

标签: machine-learning tensorflow deep-learning

实际上我正在尝试使用Image Net上预训练初始模型的转移学习到我的灰度数据集,所以我只想知道我可以使用相同的权重,或者在使用之前应该先进行一些更改权重。

提前致谢。

1 个答案:

答案 0 :(得分:6)

有几种选择:

  1. 使用标准OpenCV函数将灰度图像转换为彩色图像。
  2. 在前面添加一个深度为3的卷积图层作为输出。按照正常情况继续进行。
  3. 从现有网络的第一层中选择一组权重。
  4. 平均现有网络第一层的权重。
  5. 每个选项分别有不同的权衡:

    1. 这可能会运作得很好,但是您需要支付额外计算的费用,在这种情况下这可能不会很多。
    2. 您可能实际上是在学习尝试映射到RGB的过滤器,因此这可能类似于1.
    3. 这是一个黑客,可能不会像你期望的那样工作。
    4. 与3相同。