如何將圖像數據存儲在字典中

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)