基于块的全搜索运动估计算法实现matlab代码.docx

主文件motionEstAnalysis.m% This script uses all the Motion Estimation algorithms written for the% final project and save their results.close allclear all% imageName = caltrain.avi;VideoName = shaky_car.avi;video = aviread(VideoName);% movie(video);mbSize = 16;p = 7;fori = 1:6imgINumber = i;imgPNumber = i+2;videoI = video(imgINumber);videoP = video(imgPNumber);imgI = double(videoI.cdata);imgP = double(videoP.cdata); [row col] = size(imgI); % Exhaustive Search 基于块的全搜索算法 [BlockCenter, motionVect, computations] = motionEstES(imgP,imgI,mbSize,p); % P 帧当前重构图像imgPComp = motionComp(imgI, motionVect, mbSize); % P 帧当前图像和 P 帧当前重构图像的PSNR值ESpsnr(i+1) = imgPSNR(imgP, imgPComp, 255);EScomputations(i+1) = computations; % P 帧当前重构误差图像imagePDiff = imgP - imgPComp; ifi == 4figure; imageI = uint8(imgI);imageP = uint8(imgP);imagePComp = uint8(imgPComp);imagePDiff = uint8(imagePDiff);subplot(221);imshow(imageI); title(I 帧参考图像); subplot(222);imshow(imageP); title(P 帧当前图像); subplot(223);imshow(imagePComp); title(P 帧当前重构图像);subplot(224);imshow(imagePDiff); title(P 帧当前重构误差图像); % 画运动矢量图figure; quiver( BlockCenter(2,:), BlockCenter(1,:), motionVect(2,:), motionVect(1,:), .2,r);axis([0 320 0 240]);fori=mbSize:mbSize:col-mbSize x = [i,i]; y = [0,row];line(x,y,LineStyle,-,Marker,none);endfor j=mbSize:mbSize:row-mbSize x = [0,col]; y = [j,j];line(x,y,LineStyle,-,Marker,none);endxlabel(X);ylabel(Y);endend文件motionEstES.m% Computes motion vectors using exhaustive search method（全搜索法计算运动矢量）%% Input% imgP : The image for which we want to find motion vectors（当前图像）% imgI : The reference image（参考图像）% mbSize : Size of the macroblock（宏块尺寸）% p : Search parameter (read literature to find what this means)（搜索参数）%% Ouput% motionVect : the motion vectors for each integral macroblock in imgP（当前图像中每一个积分宏块的运动矢量）% EScomputations: The average number of points se