以下是一个示例代码,用于遍历文件夹中的图像文件:
import os
from PIL import Image
# 指定要遍历的文件夹路径
folder_path = 'path/to/folder'
# 遍历文件夹中的所有文件
for filename in os.listdir(folder_path):
file_path = os.path.join(folder_path, filename)
# 检查文件是否为图像文件
if os.path.isfile(file_path) and file_path.lower().endswith(('.png', '.jpg', '.jpeg')):
# 打开图像文件
image = Image.open(file_path)
# 在这里进行图像处理操作,例如缩放、裁剪等
# 保存处理后的图像文件
output_path = 'path/to/output/directory/' + filename
image.save(output_path)
在上述代码中,需要将folder_path
变量替换为要遍历的文件夹的路径。然后,代码将遍历文件夹中的所有文件,并检查每个文件是否为图像文件(根据文件扩展名进行判断)。如果是图像文件,则通过PIL库的Image.open()
函数打开图像文件。在打开图像文件后,可以对图像进行各种操作,例如缩放、裁剪等。最后,使用image.save()
函数保存处理后的图像文件到指定的输出目录中。
需要确保已安装PIL库(也称为Pillow库),可以使用以下命令进行安装:
pip install pillow
下一篇:遍历文件夹中的文件