在Python中进行图像批处理可以使用多种库,如 Pillow
、OpenCV
和 imageio
。这些库可以用来执行各种图像处理任务,如调整大小、裁剪、旋转、滤镜应用等。以下是使用这些库进行图像批处理的示例。
使用 Pillow 进行图像批处理
Pillow
是一个功能强大的图像处理库。下面是如何使用 Pillow
进行图像批处理的基本示例:
-
安装 Pillow
pip install pillow
-
批处理图像
from PIL import Image import os def process_image(file_path, output_path): with Image.open(file_path) as img: # 示例处理:调整大小 img = img.resize((800, 800)) # 示例处理:应用滤镜 img = img.convert('L') # 转为灰度图 img.save(output_path) def batch_process_images(input_folder, output_folder): if not os.path.exists(output_folder): os.makedirs(output_folder) for filename in os.listdir(input_folder): if filename.lower().endswith(('.png', '.jpg', '.jpeg')): file_path = os.path.join(input_folder, filename) output_path = os.path.join(output_folder, filename) process_image(file_path, output_path) # 示例使用 batch_process_images('input_images', 'output_images')
使用 OpenCV 进行图像批处理
OpenCV
是一个功能强大的计算机视觉库,适用于更复杂的图像处理任务。下面是如何使用 OpenCV
进行图像批处理的示例:
-
安装 OpenCV
pip install opencv-python
-
批处理图像
import cv2 import os def process_image(file_path, output_path): # 读取图像 img = cv2.imread(file_path) # 示例处理:调整大小 img = cv2.resize(img, (800, 800)) # 示例处理:将图像转换为灰度 img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) cv2.imwrite(output_path, img) def batch_process_images(input_folder, output_folder): if not os.path.exists(output_folder): os.makedirs(output_folder) for filename in os.listdir(input_folder): if filename.lower().endswith(('.png', '.jpg', '.jpeg')): file_path = os.path.join(input_folder, filename) output_path = os.path.join(output_folder, filename) process_image(file_path, output_path) # 示例使用 batch_process_images('input_images', 'output_images')
使用 imageio 进行图像批处理
imageio
是另一个图像处理库,支持多种图像格式和文件类型。
-
安装 imageio
pip install imageio
-
批处理图像
import imageio import os def process_image(file_path, output_path): img = imageio.imread(file_path) # 示例处理:调整大小 img_resized = imageio.imresize(img, (800, 800)) # 示例处理:转换为灰度 img_gray = imageio.rgb2gray(img_resized) imageio.imwrite(output_path, img_gray) def batch_process_images(input_folder, output_folder): if not os.path.exists(output_folder): os.makedirs(output_folder) for filename in os.listdir(input_folder): if filename.lower().endswith(('.png', '.jpg', '.jpeg')): file_path = os.path.join(input_folder, filename) output_path = os.path.join(output_folder, filename) process_image(file_path, output_path) # 示例使用 batch_process_images('input_images', 'output_images')
总结
- Pillow 适合基本的图像处理任务,如调整大小、裁剪、滤镜应用等。
- OpenCV 适合更复杂的计算机视觉任务,如图像变换、特征检测等。
- imageio 提供了简洁的接口,适用于读取和保存各种图像格式。
你可以根据具体的需求选择合适的库来进行图像批处理。以上示例展示了如何处理图像的基本任务,你可以在这些基础上扩展和自定义更多的图像处理功能。