motionEstimationGUI/testMotionEstimatorClass3D.m at master · HendrikMuenster/motionEstimationGUI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
% Author Information:
% Hendrik Dirks
% Institute for Computational and Applied Mathematics
% University of Muenster, Germany
%
% Contact: hendrik.dirks@wwu.de
%
%
% Version 1.00
% Date: 2016-09-20

% MotionEstimator is copyright ©2016 by Hendrik Dirks and is distributed under the terms of the GNU General Public License (GPL) version 3 (or later).
%
% If you plan to distribute the software (commercially or not). Please contact me for more information.


clear all;close all;clc;
addpath(genpath(cd));
%%
sizeG = 20;

image1 = zeros(sizeG,sizeG,sizeG) + rand(sizeG,sizeG,sizeG)*0.1;
image2 = zeros(sizeG,sizeG,sizeG) + rand(sizeG,sizeG,sizeG)*0.1;

image1(5:15,5:15,5:15) = 1;
image2(5:15,5:15,8:18) = 1;

cut1 = squeeze(image1(:,10,:));
cut2 = squeeze(image2(:,10,:));

figure(1);imagesc(cut1);axis image;
figure(2);imagesc(cut2);axis image;

%%
u = cat(4,image1,image2);

tol = 1e-5;
alpha = 0.01;

verbose = 0;
dataTerm = 'L1';
regularizerTerm = 'Huber';
doGradientConstancy = 0;
steplength = 0.8;
numberOfWarps = 5;

%%

motionEstimator = motionEstimatorClass(u,tol,alpha,'verbose',verbose,'dataTerm',dataTerm,'regularizerTerm',regularizerTerm,'doGradientConstancy',doGradientConstancy,'steplength',steplength,'numberOfWarps',numberOfWarps);
motionEstimator.init;


%%

tic;motionEstimator.runPyramid;toc;

%%
v = motionEstimator.getResult;

%%
for i=1:3
    for j=1:size(v,3)
        figure(1);imagesc(v(:,:,j,1,i));axis image;title(['Field ',num2str(j),', component ',num2str(i)]);colorbar;pause
    end
end
error = absoluteError(squeeze(v(:,:,1,:)),flowGT)