基于DWT數(shù)字水印源碼

1、精選優(yōu)質(zhì)文檔-傾情為你奉上DWT水印源碼1%DWT水印源碼%完全根據(jù)DCT1修改，只為了拋磚引玉%人生在于嘗試，快樂。%有錯誤的地方希望大家指正%嵌入源碼clcclear all;% 保存開始時間start_time=cputime;k=20;       blocksize=8;   % 設置塊的大小% 讀入原圖像file_name='_lena_std_bw.bmp'cover_object=double(imread(file_name);% 原圖像的行數(shù)與列數(shù)Mc=size(cover_object,1);   

2、0;     %原圖像行數(shù)Nc=size(cover_object,2);         %原圖像列數(shù)% 最大嵌入信息量max_message=Mc*Nc/(blocksize2);% 讀入水印圖像file_name='c.bmp'message=double(imread(file_name);%水印圖像的行數(shù)與列數(shù)Mm=size(message,1);               %水印行數(shù)Nm=size(message,2);  

3、            %水印列數(shù)message_vector=reshape(message,1,Mm*Nm);% 檢查水印信息是否過大if (Mm*Nm> max_message)  error('水印太大')end%將隨機數(shù)發(fā)生器的狀態(tài)置為1100randn('state',1100);% 產(chǎn)生watermarked_image,并寫入原圖信息watermarked_image=cover_object;% 將圖像分塊嵌入% 當 (2,2) > (2,3) 且 message

4、_pad(kk)=0% 當 (2,2) < (2,3) 且 message_pad(kk)=1%經(jīng)過分析可以得出結(jié)論：在提取水印時，如果cD1(2,2)>cD1(2,3)便是嵌入了水印的黑色像素，%反之則是嵌入了白色像素x=1;y=1;h=waitbar(0,'嵌入水印，請等待');for (kk = 1:length(message_vector)  % 對塊進行DWT變換  cA1,cH1,cV1,cD1 = dwt2(cover_object(x:x+blocksize-1,y:y+blocksize-1),'haar');

5、  a=cD1;  % 如果 message_pad(kk)= 0  if (message_vector(kk) = 0)    % 且(2,2) < (2,3) ，交換它們    if (cD1(2,2) < cD1(2,3)          temp=cD1(2,3);          cD1(2,3)=cD1(2,2);          cD1(2,2

6、)=temp;    end    % 如果message_pad(kk) = 1,   elseif (message_vector(kk) = 1)        % 且(2,2) > (2,3) ，交換它們    if (cD1(2,2) >= cD1(2,3)          temp=cD1(2,3);          cD1(2,3)=cD1(2,2);

7、60;         cD1(2,2)=temp;    end  end    % 檢查(2,2) ， (2,3)的差是否>= k  if cD1(2,2) > cD1(2,3)    if cD1(2,2) - cD1(2,3) < k        cD1(2,2)=cD1(2,2)+(k/2);        cD1(2,3)=cD1(2,3)-(k/2);  

8、;           end  else       if cD1(2,3) - cD1(2,2) < k        cD1(2,3)=cD1(2,3)+(k/2);         cD1(2,2)=cD1(2,2)-(k/2);    end  end      %IDWT  watermarked_image(x:x+blocksize-1

9、,y:y+blocksize-1)= idwt2(cA1,cH1,cV1,cD1,'haar',Mc,Nc);  % 移動到下一塊  if (x+blocksize) >= Nc    x=1;    y=y+blocksize;  else    x=x+blocksize;  end  waitbar(kk/length(message_vector),h);endclose(h);% 轉(zhuǎn)換為uint8并寫入dwt_watermarked.bmpwaterma

10、rked_image_uint=uint8(watermarked_image);imwrite(watermarked_image_uint,'dwt_watermarked.bmp','bmp');% 顯示運行時間elapsed_time=cputime-start_time,% 計算psnr值psnr=psnr(cover_object,watermarked_image), % 顯示圖像figure(1)imshow(message,);title('水印');figure(2)subplot(1,2,1);imshow(watermar

11、ked_image,)title('嵌入水印圖像')subplot(1,2,2);imshow(cover_object,);title('原圖');%提取源碼clear all;% 保存開始時間start_time=cputime;blocksize=8;   % 設置塊的大小% 讀入嵌入水印圖像file_name='dwt_watermarked.bmp'watermarked_image=double(imread(file_name);% 嵌入水印圖像的行數(shù)與列數(shù)Mw=size(watermarked_image,1);

12、60;       %嵌入水印圖像的行數(shù)Nw=size(watermarked_image,2);         %嵌入水印圖像的列數(shù)% 最大嵌入信息量max_message=Mw*Nw/(blocksize2);% 讀入原始水印file_name='c.bmp'orig_watermark=double(imread(file_name);% 原始水印的行數(shù)與列數(shù)Mo=size(orig_watermark,1);    %原始水印的行數(shù)No=size(orig_watermark,2

13、);    %原始水印的列數(shù)% 將圖像分塊提取水印x=1;y=1;h=waitbar(0,'水印提取中，請等待');for (kk = 1:max_message)  % 對塊進行dwt變換  cA1,cH1,cV1,cD1 = dwt2(watermarked_image(x:x+blocksize-1,y:y+blocksize-1),'haar');    % 如果cD1(2,2) > cD1(3,3) 那么 message_vector(kk)=0  % 否則 message_v

14、ector(kk)=1  if cD1(2,2)>cD1(3,3)    message_vector(kk)=0;  else    message_vector(kk)=1;  end  % 移動到下一塊  if (x+blocksize) >= Mw    x=1;    y=y+blocksize;  else    x=x+blocksize;  end  waitbar(kk/max_message,h);endclose(h);% 將message重新排列message_vector=message_vector(1,1:Mo*No); %只需前Mo*N