将下载的图片存储到变量时出错

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

您可以尝试：

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)

Answer 2

您的文件将只包含一个图像文件。您必须在循环中运行它才能获取所有图像。

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)