При загрузке набора данных изображения в массив numpy, как я могу преобразовать 3d-формат изображения в 2d?

Question

Чтобы загрузить мой набор данных изображений, я выполнил следующую кодировку

X=[]
for i in range(1,682):
    image=Image.open(str(i)+'.jpg')
    image=image.resize((100,100))
    temp=asarray(image)
    X.append(temp)

Форма X равна (681 100 100 300), но я хочу, чтобы форма X была (681 100 100). Как я могу это сделать?

Answer 1

Вы можете использовать opencv для чтения изображений, которые он читает как numpy массивы

import cv2
X=[]
for i in range(1,682):
    temp = cv2.imread(str(i)+'.jpg', cv2.IMREAD_GRAYSCALE)
    temp = cv2.resize(temp, (100,100))
    X.append(temp)