PIL如何批量给图片添加文字水印？

在图像处理领域，为图片添加文字水印是保护版权、标注来源或添加品牌标识的常见需求。Python的PIL（Python Imaging Library，现以Pillow库形式维护）因其简洁的API和强大的功能，成为开发者处理图片的首选工具之一。本文将深入探讨如何使用PIL库批量为图片添加文字水印，从基础操作到高级技巧，逐步展开。

一、环境准备与基础概念

1.1 安装Pillow库

在开始之前，需确保已安装Pillow库。可通过pip命令快速安装：

pip install Pillow

1.2 PIL基础概念

Pillow库提供了Image类用于处理图像，ImageDraw模块用于在图像上绘制图形和文本，ImageFont模块用于加载字体文件。这些是添加文字水印的基础组件。

二、单张图片添加文字水印

2.1 加载图片与创建绘图对象

首先，加载一张图片并创建一个可以在其上绘制的对象：

from PIL import Image, ImageDraw, ImageFont
# 加载图片
image_path = 'example.jpg'
image = Image.open(image_path)
# 创建一个可以在图片上绘图的对象
draw = ImageDraw.Draw(image)

2.2 加载字体与设置文本属性

选择字体并设置文本的大小、颜色等属性：

# 加载字体文件，确保路径正确
font_path = 'arial.ttf'  # 或使用系统默认字体，如：ImageFont.load_default()
font_size = 40
try:
    font = ImageFont.truetype(font_path, font_size)
except IOError:
    print("字体文件未找到，使用默认字体")
    font = ImageFont.load_default()
# 文本内容与颜色
text = "Watermark"
text_color = (255, 255, 255, 128)  # RGBA格式，A为透明度

2.3 添加文字水印

确定文本位置并添加到图片上：

# 文本位置（左下角为基准点）
text_position = (10, image.height - font_size - 10)
# 添加文本
draw.text(text_position, text, font=font, fill=text_color)
# 保存图片
output_path = 'example_with_watermark.jpg'
image.save(output_path)

三、批量处理图片

3.1 遍历文件夹中的图片

使用os模块遍历指定文件夹中的所有图片文件：

import os
# 图片文件夹路径
folder_path = 'images/'
# 遍历文件夹中的所有文件
for filename in os.listdir(folder_path):
    if filename.lower().endswith(('.png', '.jpg', '.jpeg')):
        # 构建完整文件路径
        file_path = os.path.join(folder_path, filename)
        # 加载图片、添加水印、保存的逻辑与单张图片处理相同
        # ...

3.2 完整批量处理示例

将上述步骤整合，实现批量添加水印：

def add_watermark(input_folder, output_folder, text, font_path=None, font_size=40, text_color=(255, 255, 255, 128)):
    # 确保输出文件夹存在
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)
    # 加载字体
    try:
        font = ImageFont.truetype(font_path, font_size) if font_path else ImageFont.load_default()
    except IOError:
        print("字体文件未找到，使用默认字体")
        font = ImageFont.load_default()
    # 遍历输入文件夹
    for filename in os.listdir(input_folder):
        if filename.lower().endswith(('.png', '.jpg', '.jpeg')):
            input_path = os.path.join(input_folder, filename)
            output_path = os.path.join(output_folder, filename)
            # 加载图片
            image = Image.open(input_path)
            draw = ImageDraw.Draw(image)
            # 添加水印
            text_position = (10, image.height - font_size - 10)
            draw.text(text_position, text, font=font, fill=text_color)
            # 保存图片
            image.save(output_path)
            print(f"Processed {filename}")
# 使用示例
input_folder = 'images/'
output_folder = 'watermarked_images/'
add_watermark(input_folder, output_folder, "My Watermark", font_path='arial.ttf')

四、高级技巧与优化

4.1 调整水印透明度

通过RGBA颜色值中的A（Alpha通道）控制水印透明度，如前文示例中的text_color = (255, 255, 255, 128)，其中128表示半透明。

4.2 多行文本与居中显示

对于多行文本或需要居中显示的情况，可以计算文本宽度和高度，调整位置：

# 假设有多行文本
lines = ["Line 1", "Line 2"]
line_height = font_size + 5  # 行间距
total_height = len(lines) * line_height
start_y = (image.height - total_height) // 2  # 垂直居中
for i, line in enumerate(lines):
    text_width, text_height = draw.textsize(line, font=font)
    start_x = (image.width - text_width) // 2  # 水平居中
    draw.text((start_x, start_y + i * line_height), line, font=font, fill=text_color)

4.3 性能优化

对于大量图片处理，考虑使用多线程或多进程加速：

from concurrent.futures import ThreadPoolExecutor
def process_image(args):
    input_path, output_path, text, font, text_color = args
    image = Image.open(input_path)
    draw = ImageDraw.Draw(image)
    # 添加水印逻辑...
    image.save(output_path)
# 准备参数列表
args_list = [(os.path.join(input_folder, f), os.path.join(output_folder, f), "My Watermark", font, text_color) 
              for f in os.listdir(input_folder) if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
# 使用多线程处理
with ThreadPoolExecutor(max_workers=4) as executor:
    executor.map(process_image, args_list)

五、实际应用场景与注意事项

• 版权保护：在分享或发布图片前添加水印，防止未经授权的使用。
• 品牌宣传：在产品图片或宣传材料中添加品牌标识，增强品牌认知度。
• 注意事项：确保水印不影响图片主要内容，避免过度遮挡；选择合适的字体和颜色，确保水印清晰可见但不过于突兀。

通过上述方法，开发者可以高效地使用PIL库批量为图片添加文字水印，满足各种场景下的需求。