使用Python轻松实现彩色水印去除

在数字图像处理领域，水印去除一直是一个热门且具有挑战性的课题。特别是在处理彩色图像时，水印的去除不仅需要保持原始图像的完整性和清晰度，还需要确保去除水印后的图像色彩真实、自然。随着Python编程语言在数据处理和图像分析方面的广泛应用，利用其强大的库和工具来实现彩色水印去除变得尤为可行。本文将介绍如何使用Python轻松实现彩色水印去除，通过结合图像处理的基本原理与Python的实用技巧，为读者提供一个清晰、易懂的实践指南。无论是对于图像处理初学者，还是对于希望深入探索水印去除技术的专业人士，本文都将是一个宝贵的资源。

Python去除彩色水印

有一张下面这样的彩色图片，白底黑字上有蓝色水印：

用photoshop之类软件处理特别麻烦，上网找去背景网站消除或者相关AI工具效果也不一定好，特别是如果这种图片有一整本数百页的书，那就更麻烦。考虑到这种水印的颜色与文字相比差异很大，因此，可以考虑利用OpenCV将其转换为hsv模式，这时候会发现水印部分的v值比文字部分的v值（可以在gimp或者photoshop中查看）大得多，用python代码处理起来就很方便了。如下代码可将上面的示例图片中的蓝色水印消除：

import cv2
import numpy as np
 
def remove_color_watermark(image_path, thres):
 
    """    
        :param image_path: 图片路径
        :param thres: V值的阈值
        :return: 除掉水印后的图片
    """
 
    # 读取图片
    img = cv2.imread(image_path)
    if img is None:
        print("无法读取图片，请检查图片路径。")
        return
 
    # 转换到HSV颜色空间
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    # 某像素点的V值大于阈值则将其转换为白色
    img[hsv[:, :, 2] > thres ] = [255, 255, 255]
    return img
 
# 请替换为你的图片路径
image_path = f"path\\to\\orig\\img.jpg"
result = remove_color_watermark(image_path, 230)
 
if result is not None:
    # 显示原始图片和处理后的图片
    cv2.imshow('Original Image', cv2.imread(image_path))
    cv2.imshow('Processed Image', result)
    cv2.waitKey(0)
    cv2.destroyAllWindows()
 
    # 保存处理后的图片
    cv2.imwrite('processed_image.jpg', result)

处理效果如下：

似乎还是很不错的。下面的代码可以将整个文件夹中所有相同水印的图片处理后输出为一个pdf文件：

import cv2
import os
 
from PIL import Image
 
def remove_color_watermark(image_path, thres):
 
    # 读取图片
    img = cv2.imread(image_path)
    if img is None:
        print("无法读取图片，请检查图片路径。")
        return
 
    # 转换到HSV颜色空间
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    # 某像素点的v值大于阈值则将其转换为白色
    img[hsv[:, :, 2] > thres] = [255, 255, 255]
    return img
 
 
if __name__ == '__main__':
    image_path = 'folder\\path\\to\\orig'
    out_path = 'path\\to\\output\\folder'
    thresh = 230
    images = []
    for filename in os.listdir(image_path):
        if filename.endswith(('.jpg', '.jpeg', '.png', '.bmp', '.tiff')):
            img = remove_color_watermark(os.path.join(image_path, filename), thresh)
            img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
            # OpenCV图片转换为pillow图片
            pillow_image = Image.fromarray(img)
            print(f'{filename}已经处理完毕。')
            # 加入pillow图片数组，为输出PDF文件做准备
            images.append(pillow_image)
    # 将处理后的图片输出为PDF文件
    images[0].save(os.path.join(out_path,'output.pdf'), save_all=True, append_images=images[1:])

总结

本文通过详细介绍和使用Python及其相关图像处理库，展示了如何轻松实现彩色水印去除的方法。从基础的图像读取与显示，到复杂的水印检测与去除算法，我们逐步深入探讨了彩色水印去除的整个过程。通过实践案例的演示，读者不仅能够掌握水印去除的基本原理，还能够学会如何运用Python编写代码来实现这一功能。彩色水印去除作为图像处理中的一个重要应用，不仅具有理论价值，更在实际生活中具有广泛的应用前景。希望本文能够为读者提供一个良好的起点，激发大家对图像处理和Python编程的进一步探索与兴趣。