robust watermaking.rar_Robust watermaking_watermaking_图像分块_鲁棒 水印

clear all; close all;clc; I=double(imread('lena.bmp')); [H,W]=size(I); h=8;w=8; block_size=8; n=h*w; K=(H/h)*(W/w); M=[1 -1 1 -1 1 -1 1 -1; 1 -1 1 -1 1 -1 1 -1; 1 -1 1 -1 1 -1 1 -1; 1 -1 1 -1 1 -1 1 -1; 1 -1 1 -1 1 -1 1 -1; 1 -1 1 -1 1 -1 1 -1; 1 -1 1 -1 1 -1 1 -1; 1 -1 1 -1 1 -1 1 -1; ]; %% Calculate the distribution of Alpha of image 'Lena' %% B=mat2cell(I,ones(H/h,1)*block_size,ones(W/w,1)*block_size); B=reshape(B,1,(H/h)*(W/w)); for k=1:(H/h)*(W/w) for m = 1:h for n = 1:w B{k}(m,n)=M(m,n)* B{k}(m,n); end end end Alpha=cellfun(@sum,cellfun(@sum,B,'UniformOutput',false)); k=size(Alpha); figure minv = min(Alpha); maxv = max(Alpha); num = maxv-minv + 1; tmp = minv; for i = 1:num numv(i,1) = tmp; numv(i,2) =length(find(Alpha == tmp)); tmp = tmp + 1; end plot(numv(:,1),numv(:,2)),title('The distribution of Alpha of image'); %% Calculate the distribution of Alpha of Shifted image 'Lena' %% T=128; G=64; Beta1=ceil((T+2*G)*2/(h*w)); for k=1:(H/h)*(W/w) for i = 1:h for j = 1:w if and (Alpha(k)>T, mod(i+j,2)==0) S1{k}(i,j)=B{k}(i,j)+Beta1; else if and (Alpha(k)<-T, mod(i+j,2)~=0) S1{k}(i,j)=B{k}(i,j)-Beta1; else S1{k}(i,j)=B{k}(i,j); end end end end end Alpha_s=cellfun(@sum,cellfun(@sum,S1,'UniformOutput',false)); figure, minv = min(Alpha_s); maxv = max(Alpha_s); num = maxv-minv + 1; tmp = minv; for i = 1:num numv(i,1) = tmp; numv(i,2) =length(find(Alpha_s == tmp)); tmp = tmp + 1; end plot(numv(:,1),numv(:,2)),title('The distribution of Alpha of Shifted image'); %% data bits %% X = randperm(floor(H/h)*floor(W/w)+1); WM_bits = zeros(1, floor(H/h)*floor(W/w)+1); wm_cnt = 1; for i = 1:H/h for j = 1:W/w WM_bits(wm_cnt) = mod(X(H/h*(i-1)+j),2); wm_cnt = wm_cnt + 1; end end %% Embedding %% Beta2=ceil((T+G)*2/(h*w)); wm_cnt = 1; for k=1:(H/h)*(W/w) wwm = WM_bits(wm_cnt); wm_cnt = wm_cnt + 1; for i = 1:h for j = 1:w %% Alpha_s(k)=[-T,T],data embeded if (Alpha_s(k)>=-T && Alpha_s(k)<=T) % bit 0 to be embedded ,the block remains intact if wwm==0 S{k}(i,j)=S1{k}(i,j); elseif wwm==1 if ((Alpha_s(k)>=0 && Alpha_s(k)<=T) && mod(i+j,2)==0) S{k}(i,j)=S1{k}(i,j)+Beta2; elseif ((Alpha_s(k)>=-T && Alpha_s(k)<0) && mod(i+j,2)~=0) S{k}(i,j)=S1{k}(i,j)-Beta2; else S{k}(i,j)=S1{k}(i,j); end end %% Alpha_s(k)~=[-T,T],No data embeded else if (Alpha_s(k)<-T || Alpha_s(k)>T) S{k}(i,j)=S1{k}(i,j); end end end end end Alpha_E=cellfun(@sum,cellfun(@sum,S,'UniformOutput',false)); figure, minv = min(Alpha_E); maxv = max(Alpha_E); num = maxv-minv + 1; tmp = minv; for i = 1:num numv(i,1) = tmp; numv(i,2) =length(find(Alpha_E == tmp)); tmp = tmp + 1; end plot(numv(:,1),numv(:,2)),title('The distribution of Alpha of Embeded image'); %% Extration %% Ori_bits = zeros(1, floor(H/h)*floor(W/w)); wm_cnt = 1; for k=1:(H/h)*(W/w) wwm = WM_bits(wm_cnt); wm_cnt = wm_cnt + 1; for i = 1:h for j = 1:w if (Alpha_E(k)>=-T && Alpha_E(k)<=T) wwm==0; Ori_bits(wm_cnt)= wwm; R{k}(i,j)=S{k}(i,j); elseif ((Alpha_E(k)>=-(2*T+G) && Alpha_E(k)<-T)||(Alpha_E(k)>T && Alpha_E(k)<=(2*T+G))) wwm==1; Ori_bits(wm_cnt)= wwm; if ( (Alpha_E(k)>T && Alpha_E(k)<=(2*T+G)) && mod(i+j,2)==0) R{k}(i,j)=S{k}(i,j)-Beta2; elseif ((Alpha_E(k)>=-(2*T+G) && Alpha_E(k)<-T) && mod(i+j,2)~=0) R{k}(i,j)=S{k}(i,j)+Beta2; end end if (Alpha_E(k)>(2*T+G)&& mod(i+j,2)==0) R{k}(i,j)=S{k}(i,j)-Beta1; elseif (Alpha_E(k)<(-2*T+G)&& mod(i+j,2)~=0) R{k}(i,j)=S{k}(i,j)+Beta1; else R{k}(i,j)=S{k}(i,j); end end end end Alpha_R=cellfun(@sum,cellfun(@sum,R,'UniformOutput',false)); figure, minv = min(Alpha_R); maxv = max(Alpha_R); num = maxv-minv + 1; tmp = minv; for i = 1:num numv(i,1) = tmp; numv(i,2) =length(find(Alpha_R == tmp)); tmp = tmp + 1; end plot(numv(:,1),numv(:,2)),title('The distribution of Alpha of Original image'); % clear all; close all;clc; % I=double(imread('lena.bmp')); % [H,W]=size(I); % sb_x =8;sb_y=8; % Rnum=floor(H/sb_x); % Cnum=floor(W/sb_y); % M=[1 -1 1 -1 1 -1 1 -1; % 1 -1 1 -1 1 -1 1 -1; % 1 -1 1 -1 1 -1 1 -1; % 1 -1 1 -1 1 -1 1 -1; % 1 -1 1 -1 1 -1 1 -1; % 1 -1 1 -1 1 -1 1 -1; % 1 -1 1 -1 1 -1 1 -1; % 1 -1 1 -1 1 -1 1 -1; % ]; % T=128; % G=64; % Beta1=ceil((T+2*G)*2/(sb_x*sb_y)); % for ri = 1 : Rnum % for cj = 1 : Cnum % sb = I(((ri - 1) * sb_x + 1) : ri * sb_x, ((cj - 1) * sb_y + 1) : cj * sb_y); % Alpha = sum(sum(sb)); % wm_sb = zeros(size(sb)); % for i = 1: sb_x % for j = 1: sb_y % if (Alpha>T&& mod(i+j,2)==0) % wm_sb(i,j) = sb(i,j)+Beta1; % elseif (Alpha<-T&&mod(i+j,2)~=0) % wm_sb(i,j) = sb(i,j) - Beta1; % else % wm_sb(i,j) = sb(i,j); % end % end % end % wm_I(((ri - 1) * sb_x + 1) : ri * sb_x, ((cj - 1) * sb_y + 1) : cj * sb_y) = wm_sb; % end % end