在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成员(我假设DenseDesignMatrix(。