基于DCT的数字水印代码

以下是一个基于DCT的数字水印Python代码示例，其中使用了Python的numpy库进行DCT变换和矩阵操作：

import numpy as np
import cv2

# 读入原始图像和水印图像
img = cv2.imread('lena.png', 0)
watermark = cv2.imread('watermark.png', 0)

# 对原始图像进行DCT变换
dct_img = cv2.dct(np.float32(img))

# 将水印图像嵌入到DCT变换后的频域系数中
alpha = 0.1  # 水印强度
dct_watermark = alpha * cv2.dct(np.float32(watermark))
dct_img[0:8, 0:8] += dct_watermark

# 对修改后的DCT系数进行逆变换得到嵌入了水印的图像
watermarked_img = cv2.idct(np.float32(dct_img))

# 显示结果图像
cv2.imshow('Watermarked Image', watermarked_img)
cv2.waitKey(0)
cv2.destroyAllWindows()

在代码中，首先使用`cv2.imread`函数读入原始图像和水印图像，并使用`cv2.dct`函数对原始图像进行DCT变换，得到频域系数。然后将水印图像也进行DCT变换，并使用一个参数`alpha`表示水印的强度，将水印系数嵌入到原始图像的频域系数中。最后使用`cv2.idct`函数对修改后的DCT系数进行逆变换，得到嵌入了水印的图像。