У меня есть папка с многомерными изображениями RGB, принадлежащая трем классам. Мне нужно создать файл pickle с двумя массивами, которые будут иметь изображения в качестве первого массива и соответствующие выходные данные для каждого класса в качестве второго массива, 1-й класс будет 0, 2-й класс будет 1, а третий будет 2. Pickle Файл должен иметь массив изображений в качестве входных данных и соответствующий массив в качестве выходных данных в соответствии с классом, таким как X и y для классификации.
Я достиг первой части, поместив все изображения в один массив, но для второй части мне нужно направление.
import glob
filelist = glob.glob('/folder_name/*.jpg')
import numpy as np
from PIL import Image
import pickle
x = np.array([np.array(Image.open(name)) for name in filelist])
filehandle = open('filename.pkl', 'wb')
pickle.dump( x, filehandle, protocol=2 )
filehandle.close()
Ожидаемый результат - файл pickle с массивом всех изображений вместе с соответствующим классом.