cor_embed.m
上传用户:haoweisi88
上传日期:2010-02-03
资源大小:1950k
文件大小:3k
源码类别:

图形图象

开发平台:

Matlab

  1. %Name: Chris Shoemaker
  2. %Course: EER-280 - Digital Watermarking
  3. %Project:  Threshold-Based Correlation using block processing in the spatial domain
  4. %           Watermark embedding
  5. clear all;
  6. % save start time
  7. start_time=cputime;
  8. k=40;            % set the gain factor for embeding
  9. blocksize=8;   % set the size of the block in cover to be used for each bit in watermark
  10. % read in the cover object
  11. file_name='_lena_std_bw.bmp';
  12. cover_object=double(imread(file_name));
  13. % determine size of watermarked image
  14. Mc=size(cover_object,1); %Height
  15. Nc=size(cover_object,2); %Width
  16. % determine maximum message size based on cover object, and blocksize
  17. max_message=Mc*Nc/(blocksize*blocksize);
  18. % prefilter the cover image with a matched filter to reduce detection errors
  19. %prefilt=[-1,-1,-1;-1,10,-1;-1,-1,-1]./2;
  20. %cover_object=filter2(prefilt,cover_object);
  21. % read in the message image
  22. %file_name='_copyright.bmp';
  23. file_name='fig11.gif';
  24. message=double(imread(file_name,'gif'));
  25. Mm=size(message,1); %Height
  26. Nm=size(message,2); %Width
  27. % reshape the message to a vector
  28. %message=round(reshape(message,Mm*Nm,1)./256);
  29. message=fix(reshape(message,Mm*Nm,1)./2);
  30. % check that the message isn't too large for cover
  31. if (length(message) > max_message)
  32.     error('Message too large to fit in Cover Object.')
  33. end
  34. % pad the message out to the maximum message size with 0's
  35. message_pad=ones(1,max_message);
  36. message_pad(1:length(message))=message;
  37. % read in key for PN generator
  38. file_name='_key.bmp';
  39. key=double(imread(file_name))./256;
  40. % reset MATLAB's PN generator to state "key"
  41. rand('state',9);
  42. % generate the watermark equal to the size of one block
  43. pn_sequence=round(2*(rand(blocksize,blocksize)-0.5));
  44. % process the image in blocks
  45. % first construct the global watermark mask
  46. x=1;
  47. y=1;
  48. for (kk = 1:length(message_pad))    
  49.     
  50.     % if message bit contains zero, add PN sequence to that portion of mask
  51.     if (message_pad(kk) == 0)
  52.         watermark_mask(y:y+blocksize-1,x:x+blocksize-1) = pn_sequence;
  53.         
  54.     % otherwise mask is filled with zeros
  55.     else
  56.         watermark_mask(y:y+blocksize-1,x:x+blocksize-1) = zeros(blocksize,blocksize);
  57.     end
  58.     
  59.     % move to next block of mask along x; If at end of row, move to next row
  60.     if (x+blocksize) >= Nc
  61.         x=1;
  62.         y=y+blocksize;
  63.     else
  64.         x=x+blocksize;
  65.     end
  66. end
  67. % add watermark mask to cover image using gain factor k 
  68. watermarked_image_dbl=cover_object+k*watermark_mask;
  69. watermarked_image_int=uint8(cover_object+k*watermark_mask);
  70. % write the watermarked image out to a file
  71. imwrite(watermarked_image_int,'cor_watermarked.bmp','bmp');
  72. % display processing time
  73. elapsed_time=cputime-start_time,
  74. % calculate the PSNR
  75. psnr=psnr(cover_object,watermarked_image_dbl,Mc,Nc),
  76. % display watermarked image
  77. figure(1)
  78. imshow(watermarked_image_int,[])
  79. title('Watermarked Image')