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

Question

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

提前致谢.

Answer 1

有几个选项可供选择:

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

每个选项分别有不同的权衡:

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