查看: 2609|回复: 0

MATLAB DCT水印源程序代码解析

字体大小: 正常放大

1188 主题	4 听众	2931 积分

该用户从未签到

电梯直达

1^#

发表于 2023-11-16 11:50 |只看该作者 |倒序浏览

|招呼Ta 关注Ta

离散余弦变换（Discrete Cosine Transform，DCT）是一种在信号处理和数据压缩领域广泛应用的数学变换方法。其核心思想是将一系列数据点通过一组余弦函数的加权和转换，从而将信号或图像从时域/空域表示转换为频域表示。DCT在信息编码和压缩中扮演着重要的角色，尤其在图像和音频领域，它通过能量集中的方式，更高效地表示信号的频率分量。
DCT的原理包括以下几个关键点：

1.变换域： DCT将输入的序列或图像分解为一组频率成分，每个分量对应不同频率，类似于傅里叶变换，但DCT的基函数是实数。
2.余弦函数： DCT使用一组余弦函数作为基函数，这些函数在不同频率上振荡，与傅里叶变换相似但更适合实际应用。
3.变换步骤：对于一维序列，DCT将N个数据点转换为N个频率分量。对于二维图像，DCT将图像分块并对每个块应用变换，得到每个块的频率分量。
4.能量集中：与傅里叶变换不同，DCT倾向于将信号或图像的能量集中在少数低频分量上，这在图像压缩中尤为有用。
5.类型：存在不同类型的DCT变换，其中类型II是最常用的，特别是在图像压缩中得到广泛应用。

DCT的应用广泛，其中包括：

6.图像压缩： JPEG图像压缩利用DCT将图像转换为频域表示，通过去除或量化高频分量，实现了高效的图像压缩。
7.音频处理： MP3等音频压缩格式使用DCT将音频信号转换为频域表示，以减小文件大小同时保留主要信息。
8.信号处理：在通信领域，DCT用于数据传输和信号处理，有助于分析信号的频率成分。

总体而言，DCT是一种强大的数学工具，能够以更适合分析和处理的方式表示信号或图像，为多个领域提供了有力的支持。现在，我们将通过DCT的原理和应用，探讨如何将水印嵌入图像的具体过程。

现在我们使用DCT将水印提案加到图片中
这段代码执行了图像水印嵌入的过程，以下是逐行代码的解释：

clear;clc
% h' u\" J0 F7 j; e0 v( D$ R0 M
+ v7 j3 P7 d: Y) c
size=512;
# a( }8 w3 X; z\" B% Q! u) G
block=8;
- B! ?- l* g- j: c/ q6 ]
blockno=size/block;. N9 B9 F9 T$ L* @) |
LENGTH=size*size/64;\" D: K7 m) Y% V9 @0 [
Alpha1=0.02; Alpha2=0.02;3 D/ M1 @4 w9 [' q% o
T1=100;
8 `: t1 J/ } ^. }\" P
I=zeros(size,size);! ` p ^4 X; S
D=zeros(size,size);
* q\" H\" c/ q: F+ ?7 c! `. z0 h
BW=zeros(size,size);. R\" ]2 a1 L, u8 T
block_dct1=zeros(block,block);, i, h+ y$ O2 e2 x9 p
% o) b% C. Y* {; ]2 S7 ]
i=imread('watermark64by64.png');
- S/ Y; e3 S7 U/ ~
mark=reshape(i,1,LENGTH);
. Y\" l% m4 ~# C% ^\" M) T, W) E
figure;imshow(i);title('水印');4 ]) ~! R# ~: [8 `
%subplot(3,2,1);plot(mark);title('水印');
5 U ?4 [/ z0 {0 d. T0 H
%显示原图0 t: p; D8 d. h5 N3 s' V) C: ~
figure;I=imread('lena512512.jpg');
+ p3 h/ f8 P; V
' z* c# t# g5 ^# s% h# H6 j8 @
I0=I;
1 o, z/ z3 L7 o+ M) Z
imshow(I);title('原始图像');
8 t0 @) @( Y& C& m
5 P8 ~, O0 g! M! v! r, y' P& a
% BW=edge(I,'canny');figure;;imshow(BW);
, h% E7 o: L, n8 ?3 D7 E
% title('edge of origine image');+ b1 ]0 G% z2 G2 A7 a
%嵌入水印
4 w8 m5 l/ @* V1 F
k=1;
. s8 F a1 w. T7 j
for m=1:blockno/ x: z3 S2 d4 f* i0 O
for n=1:blockno) w7 @- A! `2 y2 O) ~1 [5 `: V
x=(m-1)block+1; y=(n-1)block+1;2 {: \ N9 V( U1 r' }3 M\" i
block_dct1=I(x:x+block-1,y:y+block-1);$ b6 R9 ~/ L% a8 F
block_dct1=dct2(block_dct1);8 @8 V. B+ t( b; p; {3 A0 T
BW_8_8=BW(x:x+block-1,y:y+block-1);7 q\" E+ A& m' t* Q. v2 z
if m<=1|n<=1
d: f8 S: i6 O$ X: z* ~
T=0;
8 J3 a% ?) {1 C1 W7 z
else9 A* Z; z7 U# P7 J: B0 y
T=sum(BW_8_8); T=sum(T);
2 Z, \% }8 d3 `3 k! }
end6 d; K! Y2 g- a& p% ]$ ?) n& T0 N
if T>T1
- s d8 x0 C! P
Alpha=Alpha2;* J/ p7 P! S% C( T5 q
else& c0 }0 g# c0 @/ s; o$ v\" v
Alpha=Alpha1;- g' S9 m M4 W) H! n0 n5 N
end
, z- ?# H8 V; O& v
block_dct1(1,1)=block_dct1(1,1)(1+Alphamark(k));
% N1 k% Z) v8 v, u, R2 J
block_dct1=idct2(block_dct1);4 h9 ]& U/ q1 i; E, O3 b0 d B$ h
D(x:x+block-1,y:y+block-1)=block_dct1;
k=k+1;
- Q; s& g$ S, Q% f( B9 V5 N1 R
end: v, P. }; q0 s7 |$ b
end
. v; a6 z3 Y; f5 l
& [% y+ v! ?; |3 c1 v8 G
figure;imshow(D,[]);title('嵌入水印后的图像');

复制代码

1.clear; clc;: 这两个命令清空了 MATLAB 的工作区和命令窗口，以确保在开始之前没有残留的变量或输出。
2.size = 512; block = 8; blockno = size / block; LENGTH = size * size / 64; Alpha1 = 0.02; Alpha2 = 0.02; T1 = 100;: 这些是设置图像处理所需的变量和参数。size 定义了图像的尺寸，block 是用于分块的大小，blockno 是图像分块后的数量，LENGTH 是图像长度，Alpha1 和 Alpha2 是用于水印嵌入的参数，T1 是一个阈值。
3.I = zeros(size, size); D = zeros(size, size); BW = zeros(size, size); block_dct1 = zeros(block, block);: 这些命令初始化了一些矩阵和数组，用于存储图像、水印、DCT 变换后的块等。
4.i = imread('watermark64by64.png'); mark = reshape(i, 1, LENGTH);: 从文件 'watermark64by64.png' 中读取水印图像，然后将其重新形状化为一个向量 mark。
5.figure; imshow(i); title('水印');: 这段代码显示了读取的水印图像。
6.figure; I = imread('lena512512.jpg'); I0 = I; imshow(I); title('原始图像');: 从文件 'lena512512.jpg' 中读取原始图像，并显示它。
7.接下来的部分是对原始图像进行边缘检测，但是注释掉了：BW = edge(I, 'canny'); figure; imshow(BW); title('edge of origine image');。因为这部分代码被注释掉了，所以不会实际执行。
8.for m = 1:blockno for n = 1:blockno ... end end: 这是一个嵌套循环，用于对图像进行分块处理和水印嵌入。
9.在嵌入水印的循环中，它首先将图像按照设定的块大小进行处理，对每个块应用 DCT 变换，然后根据水印像素的值和设定的阈值 T1 进行水印嵌入的处理。
10.最后，它显示了嵌入水印后的图像：figure; imshow(D, []); title('嵌入水印后的图像');。

%提取水印 y2 p! j- Y) j! p2 \
k=1;
% c/ V# v- S$ ]! b9 C7 c
mark1=[];( G- H& }* Z! e6 i |3 C
for m=1:blockno% w/ S# E* ^8 G$ s4 D# t& G
for n=1:blockno
. }0 \0 w& |8 g8 n$ x# j6 w
x=(m-1)*block+1; y=(n-1)*block+1;
: Y3 L! l! F3 E# i7 L( g1 {
block_dct1=D(x:x+block-1,y:y+block-1);
$ P& Q1 X D6 A6 g
block_dct0=I0(x:x+block-1,y:y+block-1);3 G, A: h5 d& w! c3 H' N) b
block_dct1=dct2(block_dct1);
* e3 q% x9 L. @
block_dct0=dct2(block_dct0);( _# w! r9 E. x/ k
BW_8_8=BW(x:x+block-1,y:y+block-1);& X' g# u) j( f
if m<=1|n<=1
, P. Z9 l5 X0 l8 q* H+ o
T=0;
* p( r\" m% w. s& ?2 o$ T\" Z
else
$ v5 m4 f' H* D m& ?
T=sum(BW_8_8); T=sum(T);
. k% d6 Z6 Q! i9 G7 ~ P' _( G
end
2 W/ @ x' D! ~; @5 s
if T>T1
% W: j+ l# p* G
Alpha=Alpha2;
* A ~6 n- c( v& `' }! x$ K. g8 q
else\" A; a; ~; m2 K1 f6 s
Alpha=Alpha1;; z- R; X& E! Y- l8 }
end
& z T( b) z1 B4 `5 B; s
5 _* B3 w1 } a
mark1(k)=(block_dct1(1,1)/block_dct0(1,1)-1)/Alpha;
1 z4 \% F; F0 F( \% K2 F
k=k+1;2 ] }0 D+ e7 d; J/ o
end
J, D* ?, |8 P! b
end
9 ]6 {0 L: n% {' w& ]* V
mark1=reshape(mark1,64,64);
$ y; l: ?8 W& T
figure h& e3 B% z5 t) ]
imshow(mark1)$ i1 Y( |* n4 A% k
title('提取的水印')

复制代码

这部分代码是关于从嵌有水印的图像中提取水印信息的过程。让我为你解释一下：

1.k=1; mark1=[];: 这两个语句初始化了一个索引变量 k 和一个空数组 mark1，用于存储提取的水印信息。
2.for m=1:blockno for n=1:blockno ... end end: 这是一个嵌套循环，用于遍历图像的分块。
3.在循环内部，它首先确定当前块的位置 x 和 y，然后分别从嵌有水印的图像 D 和原始图像 I0 中提取对应的块。
4.对提取的块应用 DCT 变换：block_dct1 = dct2(block_dct1); block_dct0 = dct2(block_dct0);。这里对嵌有水印的块和原始图像的块都进行了 DCT 变换。
5.计算了当前块的边缘信息 BW_8_8。
6.根据设定的阈值 T1，确定 Alpha 值。这与嵌入水印的过程中相似。
7.接下来的代码是用于从嵌有水印图像和原始图像中提取水印信息的关键步骤：

mark1(k) = (block_dct1(1,1) / block_dct0(1,1) - 1) / Alpha;

这个式子计算了水印信息，并将其存储到 mark1 数组中。

8.循环结束后，mark1 存储了提取出的水印信息。接下来，代码对 mark1 进行重新形状化为 64x64 的矩阵，并显示提取的水印图像。

最后的 imshow(mark1) 代码会显示提取出的水印图像。请确保之前的嵌入过程已经执行，并且使用相同的图像来提取水印，以获取正确的结果。