Question

我想在字典中存储图像元数据（高度，宽度，颜色通道等）。有多个图像，所以我想使用Python os迭代图像文件夹，并为每个图像创建一个字典，包括图像元数据。这是一些不好的代码，可以说明我的意思。

PATH_TO_IMAGES = 'images/data'
print(os.listdir(PATH_TO_IMAGES))

list_of_images = [os.listdir(PATH_TO_IMAGES)]
for im in list_of_images:
    imdict = {}
    imdict['height'] = ???
    imdict['width'] = ???
    imdict['channels'] = ???
    imdict['name'] = ???

我也很高兴知道哪些包（PIL，matplotlib，？）对此有好处。感谢。

Answer 1

我不确定'频道'是什么，但这应该可以获得您需要的其他信息。看看PIL。看起来你也可以使用Pillow。我从未使用它，但它声称易于使用。

import os
from PIL import Image

PATH_TO_IMAGES = 'images/'
image_paths = os.listdir(PATH_TO_IMAGES)
print(image_paths)

list_of_image_data = []
for im in image_paths:
    image = Image.open(PATH_TO_IMAGES + im)
    width, height = image.size
    imdict = {
        'name': im,
        'height': height,
        'width': width,
    }
    list_of_image_data.append(imdict)

print(list_of_image_data)

如何将图像数据存储在字典中

1 个答案: