【发布时间】:2015-04-08 10:42:41
【问题描述】:
我正在使用 pylearn2 训练一个简单的卷积神经网络。我将我的 RGB 图像数据存储在一个 npy 文件中。有没有办法直接从 npy 文件中直接将该数据转换为灰度数据?
【问题讨论】:
标签: machine-learning neural-network deep-learning conv-neural-network pylearn
我正在使用 pylearn2 训练一个简单的卷积神经网络。我将我的 RGB 图像数据存储在一个 npy 文件中。有没有办法直接从 npy 文件中直接将该数据转换为灰度数据?
【问题讨论】:
标签: machine-learning neural-network deep-learning conv-neural-network pylearn
如果这是一个独立文件,则使用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)。
【讨论】: