首页python中opencv中的傅立叶变换fft实现图像加水印

好的，我可以为您解答这个问题。首先，对于图像加水印，我们可以使用傅里叶变换（FFT）来实现。傅里叶变换可以将图像从空间域转换到频域，使得我们可以对其进行频域操作，例如在频域中添加水印。接下来，我将为您提供一个Python示例代码，演示如何使用OpenCV的FFT函数来实现图像加水印。

import cv2
import numpy as np

# 加载图像
img = cv2.imread("lena.png", cv2.IMREAD_GRAYSCALE)

# 获取图像尺寸
rows, cols = img.shape

# 傅里叶变换
f = np.fft.fft2(img)

# 移动零频分量到频谱中心
fshift = np.fft.fftshift(f)

# 创建水印
watermark = np.zeros((rows, cols), dtype=np.uint8)
cv2.putText(watermark, "Watermark", (100, 100), cv2.FONT_HERSHEY_SIMPLEX, 1, 255)

# 将水印转换为频域
w = np.fft.fft2(watermark)

# 移动零频分量到频谱中心
wshift = np.fft.fftshift(w)

# 将水印添加到图像中
fwatermark = fshift + 0.1 * wshift

# 移动零频分量回到左上角
fwatermarkshift = np.fft.ifftshift(fwatermark)

# 反傅里叶变换
iwatermark = np.fft.ifft2(fwatermarkshift)

# 取实部
iwatermark = np.real(iwatermark)

# 将结果转换为0-255的灰度图像
iwatermark = cv2.normalize(iwatermark, dst=None, alpha=0, beta=255, norm_type=cv2.NORM_MINMAX, dtype=cv2.CV_8U)

# 显示结果
cv2.imshow("Original Image", img)
cv2.imshow("Watermarked Image", iwatermark)
cv2.waitKey(0)
cv2.destroyAllWindows()

在这个示例中，我们首先加载了一张灰度图像，并对其进行了傅里叶变换。然后，我们创建了一个水印图像，并将其转换为频域。接下来，我们将水印添加到原始图像的频域表示中，并将结果转换回空间域。最后，我们显示了原始图像和添加水印后的图像。

请注意，这个示例只是演示了如何使用傅里叶变换来添加水印。在实际应用中，可能需要采用更复杂的算法来保证水印的鲁棒性和不可见性。