Конвертируйте данные изображения в оттенки серого из файла 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).

Другие вопросы по тегам