批量读取文件夹下图片 python

原创

mob64ca12f3bbc7 2023-10-07 03:43:22 ©著作权

文章标签 缩放 Image 读取文件 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12f3bbc7的原创作品，请联系作者获取转载授权，否则将追究法律责任

批量读取文件夹下图片的方法

引言

在实际的开发中，我们经常需要处理大量的图片数据。而这些图片数据通常存储在文件夹中，我们需要一种方法来批量读取这些图片并进行相应的处理。在Python中，有很多库可以完成这个任务，本文将介绍一种常用的方法。

准备工作

在开始之前，我们需要安装Python的图像处理库Pillow。可以通过以下命令来安装Pillow：

pip install pillow

读取文件夹下的图片

首先，我们需要确定要读取的图片存储在哪个文件夹下。假设我们的图片存储在名为"images"的文件夹中。下面是一种常用的方法来读取文件夹下的图片：

import os
from PIL import Image

folder_path = "images"  # 文件夹路径

image_list = []  # 存储图片的列表

# 遍历文件夹下的所有文件
for filename in os.listdir(folder_path):
    # 判断文件是否为图片
    if filename.endswith(".jpg") or filename.endswith(".png"):
        # 图片的完整路径
        image_path = os.path.join(folder_path, filename)
        
        # 读取图片并添加到列表中
        image = Image.open(image_path)
        image_list.append(image)

上述代码首先使用os库的listdir()函数列出文件夹下的所有文件名。然后，使用endswith()函数判断文件是否为图片（这里假设图片的扩展名为".jpg"或".png"）。如果文件是图片，则使用join()函数获取图片的完整路径。最后，使用Pillow库的open()函数读取图片，并将图片添加到列表中。

流程图

flowchart TD
    A[开始] --> B{遍历文件夹下的所有文件}
    B --> C{文件是否为图片}
    C --> D[获取图片完整路径]
    D --> E[读取图片并添加到列表中]
    C --> B
    B --> F[结束]

类图

classDiagram
    class os.listdir
    class os.path.join
    class Image.open
    os.listdir --|> os.path.join
    Image.open --> Image

进一步处理图片

一旦我们读取了文件夹下的所有图片，我们可以对这些图片进行一些进一步的处理。比如，可以对每张图片进行缩放、裁剪、旋转等操作。以图片缩放为例，下面是一个简单的示例：

# 缩放图片
def resize_image(image, width, height):
    resized_image = image.resize((width, height))
    return resized_image

# 对图片列表中的每张图片进行缩放
for image in image_list:
    resized_image = resize_image(image, 800, 600)
    # 进一步处理缩放后的图片...

在上述代码中，我们定义了一个resize_image()函数，用于缩放图片。resize()函数接受目标宽度和高度作为参数，并返回缩放后的图片对象。然后，我们可以对图片列表中的每张图片调用resize_image()函数来进行缩放。根据实际需求，还可以添加其他操作。

总结

本文介绍了一种常用的方法来批量读取文件夹下的图片。首先，我们使用os库的listdir()函数列出文件夹下的所有文件名。然后，通过判断文件是否为图片，并使用Pillow库的open()函数读取图片。最后，我们可以对读取的图片进行进一步的处理。这种方法可以方便地批量读取和处理文件夹下的图片，提高开发效率。