Python处理图像时如何添加水印？Pillow图层叠加方法

from PIL import Image

def add_watermark(image_path, watermark_path, output_path, position=(0, 0), opacity=128):
    """
    给图像添加水印。

    Args:
        image_path: 目标图像路径。
        watermark_path: 水印图像路径。
        output_path: 输出图像路径。
        position: 水印位置 (x, y)。
        opacity: 水印透明度 (0-255)。
    """
    try:
        base_image = Image.open(image_path).convert("RGBA")
        watermark = Image.open(watermark_path).convert("RGBA")

        # 调整水印大小 (可选)
        # watermark = watermark.resize((base_image.width // 4, base_image.height // 4))

        # 创建透明度蒙版
        alpha = watermark.split()[3]
        alpha = alpha.point(lambda x: min(x, opacity))
        watermark.putalpha(alpha)

        # 叠加水印
        base_image.paste(watermark, position, watermark)

        # 保存结果
        base_image.save(output_path)
        print(f"水印已成功添加到 {output_path}")

    except FileNotFoundError:
        print("文件未找到，请检查图像路径。")
    except Exception as e:
        print(f"发生错误: {e}")

# 示例用法
add_watermark("image.jpg", "watermark.png", "output.png", position=(10, 10), opacity=100)

import os
import glob
from PIL import Image

def batch_add_watermark(image_dir, watermark_dir, output_dir):
    """
    批量给图像添加水印。

    Args:
        image_dir: 包含目标图像的目录。
        watermark_dir: 包含水印图像的目录。
        output_dir: 输出图像的目录。
    """
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)

    image_files = glob.glob(os.path.join(image_dir, "*.jpg")) + glob.glob(os.path.join(image_dir, "*.png"))
    watermark_files = glob.glob(os.path.join(watermark_dir, "*.png"))

    for image_file in image_files:
        image_name = os.path.basename(image_file)
        output_file = os.path.join(output_dir, image_name)

        # 选择水印 (例如，根据图像文件名选择)
        watermark_file = watermark_files[0]  # 简单示例，使用第一个水印

        add_watermark(image_file, watermark_file, output_file)

# 示例用法
batch_add_watermark("images", "watermarks", "output")