【Python】基于blind-watermark库添加图片盲水印

news2025/1/8 2:30:14

blind-watermark 是一个用于在图像中添加和提取盲水印的 Python 库。盲水印是一种嵌入信息（如水印）到图像中的方法，使得水印在视觉上不可见，但在需要时可以通过特定的算法进行提取。以下是如何使用 blind-watermark 库来添加和提取盲水印的示例。

首先，你需要安装 blind-watermark 库。你可以通过以下命令来安装它：

pip install blind-watermark

添加盲水印

下面是一个示例代码，用于在图像中添加盲水印：

from blind_watermark import Watermark
import cv2

# 读取原始图像和水印图像
original_image_path = 'path_to_your_original_image.png'
watermark_image_path = 'path_to_your_watermark_image.png'

# 加载图像
original_image = cv2.imread(original_image_path, cv2.IMREAD_GRAYSCALE)
watermark_image = cv2.imread(watermark_image_path, cv2.IMREAD_GRAYSCALE)

# 创建水印对象
wm = Watermark(alpha=32)  # alpha 是水印嵌入强度

# 添加水印到原始图像
marked_image = wm.embed(original_image, watermark_image)

# 保存带有水印的图像
cv2.imwrite('marked_image.png', marked_image)

提取盲水印

下面是一个示例代码，用于从带有水印的图像中提取盲水印：

# 读取带有水印的图像
marked_image_path = 'marked_image.png'
marked_image = cv2.imread(marked_image_path, cv2.IMREAD_GRAYSCALE)

# 使用相同的水印对象（注意要使用相同的 alpha 值）
wm = Watermark(alpha=32)

# 提取水印
extracted_watermark = wm.extract(marked_image, original_image)

# 保存提取的水印图像
cv2.imwrite('extracted_watermark.png', extracted_watermark)

注意事项

图像格式：确保原始图像和水印图像都是灰度图像（即单通道图像）。你可以使用 OpenCV 的 cv2.imread 函数，并指定 cv2.IMREAD_GRAYSCALE 标志来读取灰度图像。
图像大小：通常，水印图像应该比原始图像小得多。如果水印图像太大，可能会导致嵌入水印后的图像质量下降或水印提取效果不佳。
alpha 参数：alpha 参数控制水印的嵌入强度。较大的 alpha 值会使水印更难以去除，但也可能导致原始图像质量下降。
图像路径：确保提供的图像路径是正确的，并且图像文件存在。