给我写一段python批量提取图片的代码
时间: 2023-05-24 20:07:06 浏览: 127
分别提取图片YUV分量 python代码
以下是一个简单的Python脚本,可批量从文件夹中提取所有图片。
```python
import os
from PIL import Image
# 设置输入和输出文件夹路径
input_folder_path = "/input_folder"
output_folder_path = "/output_folder"
# 遍历输入文件夹中的所有文件
for filename in os.listdir(input_folder_path):
file_path = os.path.join(input_folder_path, filename)
# 判断文件是否为图片文件
if filename.endswith(".jpg") or filename.endswith(".jpeg") or filename.endswith(".png"):
# 打开文件
with Image.open(file_path) as img:
# 获取文件名(不带后缀)
file_name = os.path.splitext(filename)[0]
# 设置输出文件路径和文件名(使用JPEG格式)
output_file_path = os.path.join(output_folder_path, f"{file_name}.jpg")
# 将文件保存为JPEG格式
img.convert("RGB").save(output_file_path)
```
以上代码使用Python的os模块遍历输入文件夹中的所有文件,并使用PIL库(Python Imaging Library)打开和处理图像文件。通过判断文件扩展名来确定文件是否为图像文件。文件名和路径的操作使用了os.path模块。最后,将每个图像文件转换为JPEG格式并保存到输出文件夹中。
阅读全文