起因

我找了个免费生成图标的网站，但下载是需要掏钱了，直接屏幕截图有水印，这难得倒技术宅吗？我在人工智能的帮助下写了个去水印的Python程序。

程序

import os
import cv2
import numpy as np

def remove_watermark(image, gray_color, white_color, tolerance=10):
    # 确保图像是 BGR 格式
    if len(image.shape) == 2:
        image = cv2.cvtColor(image, cv2.COLOR_GRAY2BGR)
    elif image.shape[2] == 4:
        image = cv2.cvtColor(image, cv2.COLOR_BGRA2BGR)
    
    # 创建灰色和白色的掩码
    lower_gray = np.array([max(0, c - tolerance) for c in gray_color])
    upper_gray = np.array([min(255, c + tolerance) for c in gray_color])
    lower_white = np.array([max(0, c - tolerance) for c in white_color])
    upper_white = np.array([min(255, c + tolerance) for c in white_color])
    
    mask_gray = cv2.inRange(image, lower_gray, upper_gray)
    mask_white = cv2.inRange(image, lower_white, upper_white)
    
    # 合并灰色和白色掩码
    mask_combined = cv2.bitwise_or(mask_gray, mask_white)
    
    # 反转掩码以获取非灰色和非白色区域
    mask_content = cv2.bitwise_not(mask_combined)
    
    # 创建白色背景
    white_bg = np.ones_like(image) * 255
    
    # 将内容放到白色背景上
    result = cv2.bitwise_and(image, image, mask=mask_content)
    result = cv2.add(result, cv2.bitwise_and(white_bg, white_bg, mask=cv2.bitwise_not(mask_content)))
    
    return result

# 设置水印颜色（BGR 格式）
gray_color = (239, 239, 239)  # 灰色水印
white_color = (255, 255, 255)  # 白色背景

# 获取当前文件夹中的所有PNG图片文件
image_files = [f for f in os.listdir('.') if f.lower().endswith('.png')]

for image_file in image_files:
    try:
        print(f"正在处理: {image_file}")
        
        # 读取图像
        image = cv2.imread(image_file, cv2.IMREAD_UNCHANGED)
        
        if image is None:
            print(f"无法读取图像: {image_file}")
            continue
        
        print(f"成功读取图像，形状: {image.shape}, 数据类型: {image.dtype}")
        
        # 去除水印
        result = remove_watermark(image, gray_color, white_color)
        
        # 生成新的文件名
        new_filename = f"{os.path.splitext(image_file)[0]}_no_watermark.png"
        
        # 保存新图像
        cv2.imwrite(new_filename, result)
        print(f"处理完成: {new_filename}")
    except Exception as e:
        print(f"处理 {image_file} 时出错: {str(e)}")
        import traceback
        traceback.print_exc()

print("所有图像处理完毕")

用法

得把要去水印的图片名字整得简单一些，然后和python脚本放在同一个文件夹，然后运行脚本就行啦。

代码的解释

这个脚本的主要目的是从图像中移除水印，特别是针对白底图像上的灰色水印。以下是代码的主要部分及其功能：

1. `remove_watermark` 函数：

- 确保输入图像是 BGR 格式。

- 创建灰色（水印）和白色（背景）的颜色范围掩码。

- 合并这两个掩码，然后反转，得到图像中非水印和非背景的部分。

- 创建一个纯白色背景。

- 将图像的主要内容（非水印和非背景部分）提取出来，放到白色背景上。

2. 主程序：

- 定义水印颜色（灰色）和背景颜色（白色）。

- 获取当前文件夹中所有的 PNG 图片文件。

- 对每个图片文件进行处理：

* 读取图像。

* 调用 `remove_watermark` 函数去除水印。

* 生成新的文件名（添加 "_no_watermark" 后缀）。

* 保存处理后的图像。

3. 错误处理：

- 使用 try-except 块来捕获和处理可能出现的错误。

- 如果无法读取图像或处理过程中出现错误，会打印相应的错误信息。

这个脚本的核心思想是通过颜色识别来分离图像的主要内容和水印/背景，然后只保留主要内容，并将其放置在新的白色背景上。这种方法特别适用于有简单背景（如白色）和单一颜色水印的图像。

脚本的优点是它能批量处理文件夹中的所有 PNG 图像，并且使用了 OpenCV 库来进行高效的图像处理。然而，它可能不适用于有复杂背景或水印的图像，在这些情况下可能需要更高级的图像处理技术。