在将下载的图像存储到一个变量中时出错。

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

你的files只需要一个图片文件. 你必须在循环中运行它以获取所有的图像。

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)