数字水印python

数字水印是一种在数字媒体中嵌入信息的技术，可以用于版权保护、身份验证等方面。在Python中，可以使用第三方库实现数字水印的嵌入和提取。

一个常用的数字水印嵌入方法是将信息嵌入到载体图像的频域中。具体步骤如下：

1. 读取载体图像和水印信息。
2. 对载体图像进行离散余弦变换（DCT）。
3. 将水印信息嵌入到载体图像的低频系数中。
4. 对嵌入水印后的载体图像进行反离散余弦变换（IDCT）。
5. 保存嵌入水印后的载体图像。

提取水印的方法与嵌入相似，具体步骤如下：

1. 读取嵌入水印后的载体图像。
2. 对载体图像进行离散余弦变换（DCT）。
3. 提取载体图像的低频系数，得到水印信息。
4. 保存提取出的水印信息。

下面是一个使用Python实现数字水印嵌入和提取的示例代码：

# 导入必要的库
import cv2
import numpy as np

# 定义嵌入水印的函数
def embed_watermark(img_path, watermark, alpha=0.1):
    # 读取载体图像和水印信息
    img = cv2.imread(img_path)
    watermark = cv2.imread(watermark, cv2.IMREAD_GRAYSCALE)

    # 对载体图像进行离散余弦变换（DCT）
    img_dct = cv2.dct(np.float32(img[:,:,0]))

    # 将水印信息嵌入到载体图像的低频系数中
    watermark_dct = cv2.dct(np.float32(watermark))
    watermark_dct = cv2.resize(watermark_dct, (img.shape[1]//8, img.shape[0]//8))
    img_dct[:watermark_dct.shape[0], :watermark_dct.shape[1]] += alpha * watermark_dct

    # 对嵌入水印后的载体图像进行反离散余弦变换（IDCT）
    img_watermark = cv2.idct(img_dct)

    # 保存嵌入水印后的载体图像
    cv2.imwrite('watermarked.png', img_watermark)

# 定义提取水印的函数
def extract_watermark(img_path, alpha=0.1):
    # 读取嵌入水印后的载体图像
    img = cv2.imread(img_path)

    # 对载体图像进行离散余弦变换（DCT）
    img_dct = cv2.dct(np.float32(img[:,:,0]))

    # 提取载体图像的低频系数，得到水印信息
    watermark_dct = img_dct[:img.shape[0]//8, :img.shape[1]//8]
    watermark_dct /= alpha
    watermark = cv2.idct(watermark_dct)

    # 保存提取出的水印信息
    cv2.imwrite('extracted.png', watermark)

# 嵌入水印
embed_watermark('input.png', 'python-digital-watermark.png')

# 提取水印
extract_watermark('watermarked.png')