hs0.rar_HS光流法matlab实现_wheat5wz_光流法_光流矢量_运动矢量图

function [u, v] = HS(im1, im2, alpha, ite, uInitial, vInitial, displayFlow, displayImg) if nargin<1 || nargin<2 im1=imread('0329.jpg'); im2=imread('0335.jpg'); end if nargin<3 alpha=1; end if nargin<4 ite=100; end if nargin<5 || nargin<6 uInitial = zeros(size(im1(:,:,1))); vInitial = zeros(size(im2(:,:,1))); size(uInitial) elseif size(uInitial,1) ==0 || size(vInitial,1)==0 uInitial = zeros(size(im1(:,:,1))); vInitial = zeros(size(im2(:,:,1))); end if nargin<7 displayFlow=1; end if nargin<8 displayImg=im1; end if size(size(im1),2)==3 im1=rgb2gray(im1); end if size(size(im2),2)==3 im2=rgb2gray(im2); end im1=double(im1); im2=double(im2); im1=smoothImg(im1,1); im2=smoothImg(im2,1); tic; u = uInitial; v = vInitial; [fx, fy, ft] = computeDerivatives(im1, im2); kernel_1=[1/12 1/6 1/12;1/6 0 1/6;1/12 1/6 1/12]; for i=1:ite uAvg=conv2(u,kernel_1,'same'); vAvg=conv2(v,kernel_1,'same'); u= uAvg - ( fx .* ( ( fx .* uAvg ) + ( fy .* vAvg ) + ft ) ) ./ ( alpha^2 + fx.^2 + fy.^2); v= vAvg - ( fy .* ( ( fx .* uAvg ) + ( fy .* vAvg ) + ft ) ) ./ ( alpha^2 + fx.^2 + fy.^2); end if displayFlow==1 plotFlow(u, v, displayImg, 5, 5); end