Конвертируйте данные изображения в оттенки серого из файла npy в pylearn2
Я тренирую простую нейронную сеть свертки, используя pylearn2. У меня есть данные изображения RGB, хранящиеся в файле npy. Есть ли способ преобразовать эти данные непосредственно в данные в градациях серого непосредственно из файла npy?
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
).