数字水印(DWT算法Python实现)

最新推荐文章于 2026-04-21 05:07:53 发布

原创

最新推荐文章于 2026-04-21 05:07:53 发布 · 3.5k 阅读

标签

#算法 #计算机视觉 #python

收录于

本文介绍了一种基于离散小波变换(DWT)的数字水印嵌入与提取方法。该方法通过Arnold变换增强安全性，并利用不同分辨率的小波系数进行水印嵌入。实验结果显示，此方案能有效实现水印的隐藏与提取。

全部代码：

"""-------------------------
Project:     DWT数字水印嵌入、提取
Author:      Tysay
Environment: Python3.8 + VsCode
History:     2022/12/01
-------------------------"""
import cv2
import pywt
import numpy as np
import matplotlib.pyplot as plt

plt.rcParams['font.sans-serif'] = ['SimHei']  # 替换sans-serif字体
plt.rcParams['axes.unicode_minus'] = False  # 解决坐标轴负数的负号显示问题


class WaterMarkDWT:
    def __init__(self, origin: str, watermark: str, key: int, weight: list):
        self.key = key
        self.img = cv2.imread(origin)
        self.mark = cv2.imread(watermark)
        self.coef = weight

    def imshow(self, img, title='Img', size: tuple = None):
        if isinstance(img, str):
            img = cv2.imread(img)
        if size:
            img = cv2.resize(img, size)
        plt.imshow(img, cmap='gray')
        plt.title(title)
        plt.axis('off')
        plt.show()

    def arnold(self, img):
        # 获取图片的行、列
        r, c = img.