我们可以使用RGB图像训练的模型的灰度权重来进行灰度图像吗?
问题描述:
实际上我试图使用从图像网上的预训练初始模型的转移学习到我的灰度数据集,所以我只想知道我可以使用相同的权重还是应该先使用权重进行一些更改。我们可以使用RGB图像训练的模型的灰度权重来进行灰度图像吗?
在此先感谢。
答
有可用的几个选项:
- 使用标准OpenCV函数转换灰度图像彩色图像。
- 在前面添加一个卷积层,其深度为3作为输出。按照正常进行。
- 从现有网络的第一层中选择一组权重。
- 平均来自现有网络第一层的权重。
每个选项分别有不同的取舍:
- 这大概是什么工作好,但你付出额外的计算的价格,这可能不是很多在这种情况下。
- 您可能本质上是试图映射到RGB的学习过滤器,所以这可能类似于1.
- 这是一种黑客攻击,可能无法像您期望的那样工作。
- 同3.
你可以简单地使红色,绿色和蓝色通道等于灰度图像为灰度的RGB意味着红,绿,蓝值相同。也就是说,将灰度图像复制三次并将它们堆叠在第三维上以人为创建RGB中的灰度图像。然而,颜色的使用是一个强有力的暗示,所以只用灰度图像就可能得到相当弱的预测。 – rayryeng
可以请你提供我的代码,我怎么能做到这一点。 –