machine-learning - 我们可以使用在 RGB 图像上训练的模型的权重来处理灰度图像吗？

Question

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

提前致谢。

Answer 1

有几个可用选项:

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

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

1. 这可能会很有效，但您需要付出额外计算的代价，在这种情况下可能不会很多。
2. 您可能本质上正在学习尝试映射到 RGB 的滤镜，因此这可能类似于 1。
3. 这是一种黑客行为，可能不会像您期望的那样工作。
4. 与 3 相同。