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

Question

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

Answer 1

如果这是一个独立文件，请使用numpy.load加载文件，然后使用this之类的内容转换内容：

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）。