python实现离散傅里叶变换（DFT）进行图片水印嵌入

林青檎

已于 2024-04-26 08:57:28 修改

阅读量1k

点赞数 10

文章标签： python 开发语言

于 2024-01-22 09:15:54 首次发布

本文链接：https://blog.csdn.net/qq_30402535/article/details/135739987

版权

本文介绍了一个使用Python和OpenCV库实现的图像水印嵌入与提取方法，通过快速傅里叶变换（DFT）在频域中操作，展示了如何在图像中添加和提取水印，以及相关的示例应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import cv2
import numpy as np
import matplotlib.pyplot as plt
import matplotlib

# 设置matplotlib的字体为SimHei，支持中文显示
matplotlib.rcParams['font.family'] = 'SimHei'
matplotlib.rcParams['font.sans-serif'] = ['SimHei']  # 或者其他支持中文的字体
matplotlib.rcParams['axes.unicode_minus'] = False  # 用来正常显示负号

def add_watermark(image_path, watermark_path):
    # 读取原始图像和水印图像
    image = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)
    watermark = cv2.imread(watermark_path, cv2.IMREAD_GRAYSCALE)
    watermark = cv2.resize(watermark, (image.shape[1], image.shape[0]))

    # 将原始图像和水印图像转换到频域
    f_image = np.fft.fft2(image)
    f_image_shifted = np.fft.fftshift(f_image)
    f_watermark = np.fft.fft2(watermark)

    # 嵌入水印
    alpha = 0.1  # 水印强度因子
    f_image_shifted += alpha * f_watermark

    # 将嵌入水印后的频域表示转换回空间域
    f_ishift = np.fft.ifftshift(f_image_shifted)
    img_back =