我们可以将在RGB图像上训练的模型的权重用于灰度图像吗? [英] Can we use the weights of a model trained on RGB images for Grayscale images?

问题描述

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

actually I am trying to use transfer learning from pre trained inception model on Image Net to my dataset which is in gray scale , so i just want to know can i use the same weights or some changes should be made first before using the weights.

谢谢.

推荐答案

有一些可用选项:

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

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

Each option has different trade-offs respectively:

1. 这可能是行得通的，但您要付出额外计算的代价，在这种情况下，可能不算什么.
2. 您可能实际上是在学习尝试映射到RGB的滤镜，因此这可能与1类似.
3. 这是一个hack，可能无法像您期望的那样工作.
4. 与3相同.

这篇关于我们可以将在RGB图像上训练的模型的权重用于灰度图像吗?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！