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

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)