machine-learning - 将图像数据从pylearn2中的npy文件转换为灰度

Question

我正在使用 pylearn2 训练一个简单的卷积神经网络。我将 RGB 图像数据存储在 npy 文件中。无论如何直接从npy文件将该数据直接转换为灰度数据？

Answer 1

如果这是一个独立文件，则使用numpy.load加载文件，然后使用以下内容转换内容：

def rgb2gray(rgb):
    return np.dot(rgb[...,:3], [0.299, 0.587, 0.144])

如果文件是 pylearn2 数据集的一部分（来自use_design_loc()），则加载数据集

from pylearn2.utils import serial
serial.load("file.pkl")

并将rgb2gray()函数应用于X成员（我假设 a DenseDesignMatrix）。