Python为图片和PDF去水印详解

前言

在许多实际应用中，我们需要处理一些图片和PDF文件，并且这些文件可能包含着不必要的水印。因此，本文将介绍如何使用Python去除这些水印。

需要提醒的是，本文所介绍的所有方法仅用于学习和技术研究，务必遵守各种法律法规和伦理道德规范。

去除图片水印

1. 使用Pillow库

Pillow是Python中一个非常常用的图像处理库。我们可以使用Pillow库中的Image模块来进行图像的处理，包括去除图片水印。以下是示例代码：

from PIL import Image

def remove_watermark(image_path):
    img = Image.open(image_path)
    width, height = img.size
    pixels = img.load()
    for i in range(width):
        for j in range(height):
            if pixels[i, j] == (0, 0, 0): # 假设水印为黑色
                pixels[i, j] = img.getpixel((i, j-20)) # 将当前像素点替换为上方像素点
    img.save(image_path)

在上述代码中，我们打开一张图片，遍历其中所有像素，如果发现某个像素点的颜色为黑色，就将其替换为上方像素点的颜色。这种方法主要用于去除比较简单的文本水印。

2. 使用OpenCV库

OpenCV是Python中另一个常用的图像处理库。我们可以使用OpenCV库中的图像处理函数来实现图片水印的去除。以下是示例代码：

import cv2

def remove_watermark(image_path):
    img = cv2.imread(image_path)
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    lower_black = np.array([0, 0, 0]) # 假设水印为黑色
    upper_black = np.array([180, 255, 40]) # 设定黑色颜色范围
    mask = cv2.inRange(hsv, lower_black, upper_black)
    res = cv2.bitwise_and(img, img, mask=mask)
    cv2.imwrite(image_path, res)

在上述代码中，我们首先读入一张图片，将其转换为HSV颜色空间，然后设定黑色的颜色范围，使用inRange函数提取出所有黑色的像素点，最后将其切割出来。这种方法可以处理比较复杂的图像水印。

去除PDF水印

1. 使用PyMuPDF库

PyMuPDF是Python中一个用于处理PDF文件的库。我们可以使用PyMuPDF库中的fitz模块来去除PDF中的水印。以下是示例代码：

import fitz

def remove_watermark(pdf_path):
    doc = fitz.open(pdf_path)
    for page in doc:
        blocks = page.getTextBlocks()
        for b in blocks:
            if '水印' in b[4]: # 假设水印的内容含有“水印”
                rect = fitz.Rect(b[:4])
                highlight = page.addHighlightAnnot(rect) # 添加高亮标注
                highlight.update()
    doc.save(pdf_path)

在上述代码中，我们首先读入一个PDF文件，依次处理其中每一个页面。对于每个页面，我们提取出其中所有文字块，寻找其中内容包含“水印”的块，然后使用addHighlightAnnot函数添加高亮标注并将其保存。

2. 使用pdftk工具

pdftk是一个用于处理PDF文件的命令行工具，我们可以使用它来去除PDF中的水印。以下是示例命令：

pdftk input.pdf output output.pdf stamp clear

在上述命令中，我们使用pdftk命令从输入文件input.pdf中生成新的输出文件output.pdf，并使用stamp和clear选项去除其中所有的水印。这种方法不需要对PDF文件进行解析，可以处理包含比较复杂的水印的PDF文件。

结语

本文介绍了使用Python去除图片和PDF中的水印的各种方法，每种方法都有其优缺点，需要根据具体情况进行选择。但不管采用何种方法，我们务必遵循法律法规和伦理道德规范，使用这些方法去除不合法水印是不被允许的。