close all
clear all
R = 4 ;
step = 1 ;
block_size = 16 ;
Anchor = save_yuvframe('foreman_qcif.yuv',[176 144],10,'foreman_qcif.jpg');
Tracked = save_yuvframe('foreman_qcif.yuv',[176 144],12,'foreman_qcif.jpg');
figure;
subplot(2,2,1);imshow(Anchor(R:size(Anchor,1)-R ,R:size(Anchor,2)-R ,1));title('Anchor Frame');
subplot(2,2,2);imshow(Tracked(R:size(Tracked,1)-R,R:size(Tracked,2)-R,1));title('Tracked Frame');
%%%***********************************************************
Anchor=double(Anchor);
Tracked=double(Tracked);
%%%%%**********************************************************
N_Y=size(Anchor,1)/block_size;
N_X=size(Anchor,2)/block_size;
for i = 1 + R :block_size : size(Anchor,1)- block_size - R
for j = 1 + R :block_size : size(Anchor,2)- block_size -R
MAD_min = 256 * block_size * block_size ;
% Mvx = 0 ;
% Mvy = 0 ;
for k = -R : 0.5: R
for l= -R : 0.5 : R
MAD = sum (sum ( abs ( Anchor (i:i+block_size-1 , j:j+block_size-1)- Tracked(i+k:i+k+block_size-1 , j+l:j+l+block_size-1))));
if MAD < MAD_min
MAD_min = MAD ;
dy = k;
dx = l;
end
end
end
Predicted (i:i+block_size-1 , j:j+block_size-1) = Tracked(i+dy:i+dy+block_size-1 , j+dx:j+dx+block_size-1);
i_Mv = floor ((i-1)/(block_size+1));
j_Mv = floor ((j-1)/(block_size+1));
Mvx(i_Mv+1,j_Mv+1)= dx ;
Mvy (i_Mv+1,j_Mv+1)= dy ;
end
end
subplot(2,2,3);imagesc (Predicted);title('Predicted Frame');
subplot(2,2,4); quiver (Mvx , Mvy )
%%%%%*********************calculate PSNR *********************************
R= 4 ;
Anchor1 = Anchor(R:size(Anchor,1)-R ,R:size(Anchor,2)-R);
Tracked1 = Tracked(R:size(Tracked,1)-R ,R:size(Tracked,2)-R);
N = size(Predicted,1) * size(Predicted,2) ;
max_p = 0 ;
for i = 1 : size(Predicted,1)
for j = 1 : size(Predicted,2)
error = ( 1/N ).* (sum (sum ( ( Tracked1 (i,j)- Predicted(i,j))^2)));
if Predicted(i,j) > max_p
max_p = Predicted(i,j);
end
error_image (i, j) = ( Tracked1 (i,j)- Predicted(i,j));
end
end
PSNR = 10 * log10 ((max_p^2)/error);
figure
imagesc (error_image);title('Predicted Error Image Frame');
评论0