DCT水印算法实现MATLAB

%图片加水印，提取出水印，并进行攻击测试 %林多 CSDN %http://blog.csdn.net/zxc024000 %%%%%%%%%%%% M=256; %原图像长度 N=32; %水印图像长度 K=8; %8x8的分块 I=zeros(M,M);%创建一个MxM矩阵，元素全是0 J=zeros(N,N); BLOCK = zeros(K,K); %显示原图像 subplot(5,2,1);%显示多幅图像，在第一个位置显示 I=imread('23.bmp');%将23.bmp读入I中 imshow(I);%显示图像 title('原始公开图像');%标题 %显示水印图像 subplot(5,2,2); J=imread('21.bmp'); imshow(J); title('水印图像'); %水印嵌入算法 for p=1:N %从1到32循环 for q=1:N %从1到32循环 x=(p-1)*K+1; y=(q-1)*K+1; BLOCK=I(x:x+K-1,y:y+K-1);%取相应元素保存到BLOCK中 BLOCK=dct2(BLOCK);%二维离散余弦变换 if J(p,q)==0 %如果元素为0 a=-1; else a=1; end BLOCK=BLOCK*(1+a*0.03); BLOCK=idct2(BLOCK);%反二维离散余弦变换 I(x:x+K-1,y:y+K-1)=BLOCK; end end %显示嵌入水印后的图像 subplot(5,2,3); imshow(I); title('嵌入水印后的图像'); imwrite(I,'watermarked.bmp','bmp');%将I中的数据保存成watermarked.bmp %从嵌入水印的图像中提取水印 I=imread('23.bmp');%未加水印的原图像 P=imread('watermarked.bmp');%水印图像 %提取水印算法 for p=1:N for q=1:N x=(p-1)*K+1; y=(q-1)*K+1; BLOCK1=I(x:x+K-1,y:y+K-1);%赋给BLOCK1元素 BLOCK2=P(x:x+K-1,y:y+K-1);%赋给BLOCK2元素 BLOCK1=idct2(BLOCK1);%对其本身进行反二维离散余弦变换 BLOCK2=idct2(BLOCK2);%对其本身进行反二维离散余弦变换 a=BLOCK2(1,1)/BLOCK1(1,1)-1; if a<0 W(p,q)=0; else W(p,q)=1; end end end %显示提取的水印 subplot(5,2,4); imshow(W); title('从含水印图像中提取的水印'); %----------攻击测试-----------------------% %----------中值攻击----------% P1=imread('watermarked.bmp'); P1=double(P1(:,:,1)); P1=medfilt2(P1);%中值滤波 subplot(5,2,5); imshow(P1,[]); title('中值滤波攻击'); I1=imread('23.bmp');%未加水印的原图像 %提取水印算法 for p=1:N for q=1:N x=(p-1)*K+1; y=(q-1)*K+1; BLOCK1=I1(x:x+K-1,y:y+K-1);%赋给BLOCK1元素 BLOCK2=P1(x:x+K-1,y:y+K-1);%赋给BLOCK2元素 BLOCK1=idct2(BLOCK1);%对其本身进行反二维离散余弦变换 BLOCK2=idct2(BLOCK2);%对其本身进行反二维离散余弦变换 a=BLOCK2(1,1)/BLOCK1(1,1)-1; if a<0 W1(p,q)=0; else W1(p,q)=1; end end end subplot(5,2,6); imshow(W1); title('从含中值滤波图像中提取的水印'); %----------变小攻击----------% P2=imread('watermarked.bmp'); P2=imresize(P2,0.5);%变为0.5倍，从256x256到128x128 subplot(5,2,7); imshow(P2,[]);%显示变小后图像 title('变小攻击，变为128x128像素'); I2=imread('23.bmp');%未加水印的原图像 I2=imresize(I2,0.5);%原图像一样变小 %提取水印算法 for p=1:32 for q=1:32 x=(p-1)*4+1; y=(q-1)*4+1; BLOCK1=I2(x:x+4-1,y:y+4-1);%赋给BLOCK1元素 BLOCK2=P2(x:x+4-1,y:y+4-1);%赋给BLOCK2元素 BLOCK1=idct2(BLOCK1);%对其本身进行反二维离散余弦变换 BLOCK2=idct2(BLOCK2);%对其本身进行反二维离散余弦变换 a=BLOCK2(1,1)/BLOCK1(1,1)-1; if a<0 W2(p,q)=0; else W2(p,q)=1; end end end subplot(5,2,8); imshow(W2); title('从128x128图像中提取水印'); %-------旋转攻击-------% P3=imread('watermarked.bmp'); P3=imrotate(P3,90);%逆时针旋转90度; subplot(5,2,9); imshow(P3,[]); title('旋转攻击'); I3=imread('23.bmp');%未加水印的原图像 I3=imrotate(I3,90); %原图像逆时针旋转90度; %提取水印算法 for p=1:N for q=1:N x=(p-1)*K+1; y=(q-1)*K+1; BLOCK1=I3(x:x+K-1,y:y+K-1);%赋给BLOCK1元素 BLOCK2=P3(x:x+K-1,y:y+K-1);%赋给BLOCK2元素 BLOCK1=idct2(BLOCK1);%对其本身进行反二维离散余弦变换 BLOCK2=idct2(BLOCK2);%对其本身进行反二维离散余弦变换 a=BLOCK2(1,1)/BLOCK1(1,1)-1; if a<0 W3(p,q)=0; else W3(p,q)=1; end end end subplot(5,2,10); imshow(W3); title('从含旋转图像中提取的水印');