Python 从目录中导入图片到列表或字典

在本文中，我们将介绍如何使用Python从指定目录中导入图片，并将其保存到列表或字典中。导入图片可以帮助我们在处理图像数据、图像识别、计算机视觉等方面的项目中更加便捷地管理和处理图片。

导入必要的库

在开始之前，我们需要导入一些必要的库来处理图片和目录操作。我们将使用Pillow库来处理图片，使用os库来操作文件和目录。

from PIL import Image
import os

从目录中导入图片到列表

首先，我们将演示如何将目录中的所有图片导入到一个列表中。这对于简单的图片处理操作非常方便。以下是一个示例代码：

# 设置目录路径
directory = '/path/to/directory'

# 遍历目录中的所有文件
for filename in os.listdir(directory):
    # 检查文件是否是图片文件
    if filename.endswith(".jpg") or filename.endswith(".png"):
        # 使用Pillow库打开并处理图片
        image = Image.open(os.path.join(directory, filename))
        # 添加图片到列表中
        image_list.append(image)

在上述代码中，我们首先设置了目录的路径。然后，我们使用os.listdir()函数遍历目录中的所有文件。通过检查文件的扩展名，我们确定文件是一个图片文件。然后，我们使用Pillow库的Image.open()函数打开图片，并将其添加到图片列表中。

从目录中导入图片到字典

除了将图片导入到列表中，我们还可以将图片导入到字典中，以便更好地对图片进行管理和访问。以下是一个示例代码：

# 设置目录路径
directory = '/path/to/directory'

# 创建空字典
image_dict = {}

# 遍历目录中的所有文件
for filename in os.listdir(directory):
    # 检查文件是否是图片文件
    if filename.endswith(".jpg") or filename.endswith(".png"):
        # 使用Pillow库打开并处理图片
        image = Image.open(os.path.join(directory, filename))
        # 添加图片到字典中，以文件名作为键
        image_dict[filename] = image