Ошибка при сохранении загруженных изображений в переменную

Question

Я хотел бы сохранить все изображения, которые я скачал на своем компьютере, в переменную с именем X_data.

import glob
import cv2
import numpy as np

X_data = []
for img in glob.glob ("C:\\Users\\User\*.png"):
    image = cv2.imread(img)
    X_data+=[image]
    print('X_data shape:', np.array(X_data).shape)
Output:
X_data shape: (1,)
X_data shape: (2,)
X_data shape: (3,)
X_data shape: (4,)
X_data shape: (5,)

Однако, когда я печатаю X_data:

[None,
 None,
 None,
 None,
 None,
 None,
....]

Answer 1

Ваши файлы будут занимать только один файл изображения. Вы должны запустить его в al oop, чтобы взять все изображения.

import glob
import cv2
import numpy as np

X_data = []
for img in glob.glob ("your path\*.png"):
    image = cv2.imread(img)
    X_data.append(image)
print('X_data shape:', np.array(X_data).shape)

Answer 2

Вы можете попробовать:

import os
X_data = []
path="c:\\your_path"
files = [x for x in os.listdir(path) if ".png" == x[-4:]]
for myFile in files:
    image = cv2.imread(myFile)
    X_data.append(image)
print('X_data shape:', np.array(X_data).shape)