有没有办法从图像列表创建Numpy数组？

Question

我正试图从包含1768张图片的列表中创建numpy数组。这是我的代码：

w = []
directory = os.listdir(PATH)
directory = sorted(directory, key=len)
for item in directory:
    img = Image.open(os.path.join(PATH, item))
    w.append(img)
    count+=1
print('w_shape: ', np.array(w, dtype='float32').shape)

当运行它时，我面临这个错误：

TypeError: float() argument must be a string or a number, not 'JpegImageFile'

有人能帮我解决吗？

Answer 1

您会收到这个错误，因为列表中的每个元素都是JpegImageFile，它与浮动（）不兼容。要避免这个错误，请直接将图像作为数组添加到列表中。
您可以使用np.asarray（）将图像读取为Numpy数组。请查看下面的代码片段：

w = []
directory = os.listdir(PATH)
directory = sorted(directory, key=len)
for item in directory:
    img = Image.open(os.path.join(PATH, item))
    w.append(np.asarray(img)) # convert it to an array of floats
    count+=1 
print('w_shape: ', np.array(w, dtype='float32').shape)

注意：您也可以使用np.array（img）来代替。看看这里有什么不同