当我使用 numpy.array() 处理 'list' 时如何处理？

Question

import cv2
import numpy

list_pixel=[]
list_label=[]
for i in range(0,10):
    for j in range(0,10):
        list_pixel.append(cv2.imread("C:\\Users\\kimcho\\Desktop\\testdata\\testdata_"+str(i)+"_0"+str(j)+".png",0))
        list_label.append(i)
    j=0
list_pixel.pop(0)
list_label.pop(0)
list_pixel=numpy.array(list_pixel)
print(list_pixel)
print(list_pixel.shape)
print(list_pixel[0].shape)

当我使用numpy.array()处理'list'时如何处理？我想通过模仿keras来制作数据集。但是，我制作的数据集并不满足我。我希望它像keras一样,返回这样的值：

它可以返回一个值（60000,28,28）

但是对于我的数据集，它只能这样返回：

只返回一个值(99,)——我有99张图片，我想把它们的像素加载到list_pixel中

这是我的代码：

希望有人能帮我解决这个问题。非常感谢！！！

Answer 1

使用 cv2.imread 时，您已经将图像加载为 numpy 数组。

一个简单的方法如下：

import cv2
import numpy as np

list_label = np.arange(0, 10)
path = "C:\\Users\\kimcho\\Desktop\\testdata\\testdata_{0}_0{1}.png"
list_pixel = np.array([
    cv2.imread(path.format(i, j), 0) for i in range(0, 10)
    for j in range(0, 10)
])

让我们在一个简单的场景中尝试一下：我们假设只有一个 28x28 大小的图像，比如这个：

假设它在路径my/path/image.png中。

import cv2
import numpy as np

list_label = np.arange(0, 10)
path = "my/path/image.png"
list_pixel = np.array([
    cv2.imread(path,0) for i in range(0, 10)
    for j in range(0, 10)
])

当运行list_pixel.shape 时，你会得到(100, 28, 28)。