在Python中，批量去除多个文件夹内图片水印的核心在于**遍历文件夹、识别水印像素并将其替换为背景色**。这里提供一个基于Pillow库的简易解决方案，它易于安装和使用，适合新手。 ### 核心思路与方案设计 水印通常是半透明且颜色与背景有差异的。一种常见方法是： 1. **识别水印特征**：假设水印为特定颜色（如浅灰色 `(200, 200, 200)`）或在一定RGB阈值范围内。 2. **像素替换**：遍历图片每个像素，若其RGB值接近目标水印颜色，则将其替换为估计的背景色（如白色 `(255, 255, 255)`）或相邻像素的平均值。 3. **批量处理**：使用 `os` 和 `pathlib` 模块递归遍历用户指定的多个根文件夹，对所有支持的图片文件进行处理。 下面将分步实现，并提供完整的、可直接运行的脚本。 ### 步骤一：环境安装 只需安装一个核心库：`Pillow` (PIL Fork)。在命令行中执行以下命令： ```bash pip install Pillow ``` ### 步骤二：完整Python代码 将以下代码保存为 `batch_remove_watermark.py`。 ```python #!/usr/bin/env python3 """ 批量去除多个文件夹中图片水印的简易工具 作者：基于CSDN技术社区实践[ref_1, ref_5, ref_6] 使用方法：修改脚本底部的 `root_dirs` 列表和 `watermark_color` 变量，然后运行。 """ import os from pathlib import Path from PIL import Image import sys # 支持处理的图片格式 SUPPORTED_EXTENSIONS = ('.png', '.jpg', '.jpeg', '.bmp', '.tiff', '.webp') def estimate_background_color(image, sample_region=10): """ 通过取样图片边缘（假设边缘无水印）来估计背景色。 返回(R, G, B)元组。 """ width, height = image.size pixels = [] # 从上、下、左、右边缘取样 for x in range(0, min(width, sample_region)): pixels.append(image.getpixel((x, 0))) # 上边缘 pixels.append(image.getpixel((x, height-1))) # 下边缘 for y in range(0, min(height, sample_region)): pixels.append(image.getpixel((0, y))) # 左边缘 pixels.append(image.getpixel((width-1, y))) # 右边缘 # 计算平均RGB值 avg_r = sum(p[0] for p in pixels) // len(pixels) avg_g = sum(p[1] for p in pixels) // len(pixels) avg_b = sum(p[2] for p in pixels) // len(pixels) return (avg_r, avg_g, avg_b) def remove_watermark_from_image(input_path, output_path, watermark_color_threshold, tolerance=30): """ 核心函数：去除单张图片的水印。 :param input_path: 输入图片路径 :param output_path: 输出图片路径 :param watermark_color_threshold: 假定水印颜色的RGB阈值 (R, G, B) :param tolerance: 颜色容差，值越大匹配的水印颜色范围越广 """ try: # 1. 打开图片并转换为RGB模式（确保有颜色通道） img = Image.open(input_path).convert('RGB') pixels = img.load() # 获取像素访问对象 width, height = img.size # 2. 估算背景色（用于替换水印像素） bg_color = estimate_background_color(img) # 3. 遍历每个像素，进行判断和替换 target_r, target_g, target_b = watermark_color_threshold for x in range(width): for y in range(height): r, g, b = pixels[x, y] # 判断当前像素颜色是否在水印颜色容差范围内 if (abs(r - target_r) <= tolerance and abs(g - target_g) <= tolerance and abs(b - target_b) <= tolerance): # 将水印像素替换为估算的背景色 pixels[x, y] = bg_color # 4. 保存处理后的图片 # 确保输出目录存在 os.makedirs(os.path.dirname(output_path), exist_ok=True) img.save(output_path) print(f" 成功处理: {input_path} -> {output_path}") return True except Exception as e: print(f" 处理失败 {input_path}: {e}") return False def batch_process_folders(root_directories, output_root, watermark_color_threshold, tolerance=30): """ 批量处理多个根文件夹下的所有图片。 :param root_directories: 包含多个根文件夹路径的列表 :param output_root: 处理后的图片统一输出根目录 :param watermark_color_threshold: 水印颜色阈值 (R, G, B) :param tolerance: 颜色容差 """ processed_count = 0 failed_count = 0 for root_dir in root_directories: root_path = Path(root_dir) if not root_path.exists(): print(f"警告：目录不存在，跳过 -> {root_path}") continue print(f"正在扫描目录: {root_path}") # 递归查找所有支持的图片文件 for file_path in root_path.rglob('*'): if file_path.suffix.lower() in SUPPORTED_EXTENSIONS: # 构建输出路径：保持原目录结构，放到新的输出根目录下 relative_path = file_path.relative_to(root_path) output_path = Path(output_root) / relative_path # 处理图片 success = remove_watermark_from_image( str(file_path), str(output_path), watermark_color_threshold, tolerance ) if success: processed_count += 1 else: failed_count += 1 print("\n" + "="*50) print(f"处理完成！") print(f"成功处理: {processed_count} 张图片") print(f"处理失败: {failed_count} 张图片") print(f"输出根目录: {output_root}") if __name__ == "__main__": """ 用户配置区：请根据你的实际情况修改以下三个变量。 """ # 1. 指定需要处理的多个文件夹路径（请替换为你的实际路径） root_dirs = [ "/path/to/your/first/image/folder", # 请替换 "/path/to/your/second/image/folder", # 请替换 # 可以继续添加更多文件夹路径... ] # 2. 指定水印的大致颜色 (R, G, B)。通常水印是灰色或浅色。 # 例如：浅灰色水印可以设为 (200, 200, 200) WATERMARK_COLOR = (200, 200, 200) # 请根据你的水印颜色调整 # 3. 指定处理后的图片输出根目录 OUTPUT_ROOT = "./cleaned_images" # 当前目录下的cleaned_images文件夹 # 4. (可选) 颜色容差，如果水印颜色有变化或深浅不一，可以适当调大此值（默认30） COLOR_TOLERANCE = 30 # 执行批量处理 batch_process_folders(root_dirs, OUTPUT_ROOT, WATERMARK_COLOR, COLOR_TOLERANCE) ``` ### 步骤三：使用方法（小白指南） 1. **修改配置**：用文本编辑器（如记事本、VS Code）打开 `batch_remove_watermark.py` 文件。 * 找到代码最底部 `if __name__ == "__main__":` 下面的部分。 * **`root_dirs`**：将列表中的 `"/path/to/your/first/image/folder"` 等示例路径，替换成你电脑上**存放带水印图片的文件夹**的真实路径。例如： ```python root_dirs = [ "C:\\Users\\你的用户名\\Pictures\\水印图片集", "D:\\工作资料\\待处理图片" ] ``` *注意：Windows路径中使用双反斜杠 `\\` 或单正斜杠 `/`。 * **`WATERMARK_COLOR`**：这是最关键的参数。你需要**确定水印的颜色RGB值**。一个简单的方法是： * 用Windows画图、Photoshop或在线工具打开一张有水印的图片。 * 使用“取色器”工具点击水印部分。 * 记录下显示的RGB数值，填入这里。例如，常见的浅灰色水印可能是 `(220, 220, 220)`。 * **`OUTPUT_ROOT`**：指定处理后的图片保存到哪里。默认 `"./cleaned_images"` 表示保存在与脚本同目录的 `cleaned_images` 文件夹内。可以改为其他路径，如 `"D:\\结果图片"`。 2. **运行脚本**： * 打开命令行（CMD 或 PowerShell）。 * 使用 `cd` 命令切换到你的脚本所在目录。例如： ```bash cd C:\Users\你的用户名\Desktop ``` * 运行脚本： ```bash python batch_remove_watermark.py ``` 3. **查看结果**：程序运行后，会在命令行中显示处理进度。所有处理成功的图片将按照原文件夹结构，保存到你指定的 `OUTPUT_ROOT` 目录下。 ### 核心原理与调整建议 | 关键点 | 说明与调整建议 | | :--- | :--- | | **水印颜色 (`WATERMARK_COLOR`)** | 这是去除效果的**关键**。颜色越准确，效果越好。若水印有渐变或多种颜色，你可能需要多次尝试或使用更复杂的算法[ref_4]。 | | **颜色容差 (`COLOR_TOLERANCE`)** | 默认值30意味着RGB每个通道允许±30的偏差。如果水印颜色不完全一致，**适当调大此值（如40、50）可以去除更多相近颜色的水印**，但过大会误伤正常图片区域。 | | **背景色估算** | 本脚本通过分析图片边缘像素来估算背景色[ref_6]。这适用于水印不在边缘的情况。若水印覆盖全图，此方法可能不准，你可以将其改为固定的 `(255, 255, 255)`（白色）或其他你认为的背景色。 | | **性能与局限性** | 此方法逐像素检查，处理大图或大量图片时可能较慢。它适用于**颜色单一、与背景对比明显**的简单水印。对于复杂背景上的半透明水印或PDF文件，需要更高级的技术（如OpenCV图像修复）[ref_4, ref_5]。 | ### 总结与扩展 以上代码提供了一个开箱即用的基础解决方案。其逻辑是：**遍历指定文件夹->识别特定颜色像素->替换为背景色**。通过调整水印颜色和容差，可以应对多种简单水印场景。如果效果不理想，可以考虑以下进阶方向： * **复杂水印**：结合OpenCV库，利用图像修复（`cv2.inpaint`）或更精确的掩模（Mask）检测技术[ref_4]。 * **PDF文件**：需要先将PDF页面转换为图片，处理后再合成PDF，可使用 `PyMuPDF` 或 `pdf2image` 库[ref_5, ref_6]。 * **图形界面**：如需更友好操作，可用 `tkinter` 包装上述功能，供用户选择文件夹、预览水印颜色和效果[ref_2]。