如何对彩色图像进行RBM训练？

Question

我很难理解输入颜色的策略。大多数关于RBM的教程只训练灰度图像。

如果图像是灰度的，则输入单元可以是二进制的，我可以将灰度值归一化为0,1，然后将它们视为输入层中的概率。或白化数据集，并在输入层中使用高斯单元。

如何处理彩色图像？显然，输入单元不能是二进制的--除非我为三个颜色通道中的每一个复制这些单元？或者流行的策略是什么？

Answer 1

我正在做一个类似的项目。这个想法还在试验中，所以我不能保证结果。您可以将图像转换为具有8个级别的灰度图像，然后可以使用这8个级别作为连接到具有(宽度*高度)图像的隐藏层的可见层，然后使用该隐藏层作为一个可见层，以您的类数作为隐藏层。