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

图形图象

开发平台:

Matlab

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