伪随机数DFT水印代码

以下是一个简单的Python代码示例，用于生成伪随机数DFT水印：

import numpy as np
import cv2

# 加载图像
img = cv2.imread('lena.jpg', cv2.IMREAD_GRAYSCALE)
height, width = img.shape

# 生成伪随机数矩阵
np.random.seed(0)
rand_matrix = np.random.rand(height, width)

# 将伪随机数矩阵转换为二进制
binary_rand_matrix = np.zeros((height, width), dtype=np.uint8)
for i in range(height):
    for j in range(width):
        if rand_matrix[i, j] > 0.5:
            binary_rand_matrix[i, j] = 1

# 进行DFT变换
dft_matrix = cv2.dft(np.float32(binary_rand_matrix), flags=cv2.DFT_COMPLEX_OUTPUT)

# 将水印嵌入到频域图像中
watermark = cv2.imread('watermark.png', cv2.IMREAD_GRAYSCALE)
watermark = cv2.resize(watermark, (width, height))
watermark = np.float32(watermark)
for i in range(height):
    for j in range(width):
        dft_matrix[i, j] = dft_matrix[i, j] * (1 + watermark[i, j] / 255)

# 进行IDFT变换
idft_matrix = cv2.idft(dft_matrix, flags=cv2.DFT_SCALE)

# 提取水印
extracted_watermark = np.zeros((height, width), dtype=np.uint8)
for i in range(height):
    for j in range(width):
        extracted_watermark[i, j] = np.round(abs(idft_matrix[i, j])).astype(np.uint8)

# 显示图像和水印
cv2.imshow('Original Image', img)
cv2.imshow('Watermark', watermark)
cv2.imshow('Extracted Watermark', extracted_watermark)
cv2.waitKey(0)
cv2.destroyAllWindows()

在这个示例中，我们首先加载了一张灰度图像，然后生成了一个伪随机数矩阵。接下来，我们将伪随机数矩阵转换为二进制形式，并对其进行DFT变换。然后，我们将一个水印嵌入到频域图像中。在进行IDFT变换之后，我们可以提取出水印并将其显示出来。

请注意，这只是一个简单的示例，真实的水印算法可能需要更复杂的处理步骤，以提高水印的鲁棒性和安全性。