test_robustpca_l2sparsity_mxm_block_splitting.m

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
% A test wrapper for RobustPCA implementation using splitting method
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

% Data can be arranged from the video frames 
% y = dir('*.bmp');
% 
% for i = 1:200
%     x = imread(y(i).name);
%     M(:,:,i) = rgb2gray(x);
% end

% For reproduciton of the result we have stored the data in a mat file
% loading airport video image sequence of first 200 frames
load('./Data/Airport data/200frames.mat');

M = double(M);
s = size(M);

mu = 1;
bdim = 10;
lambda = 1 / (bdim * sqrt(s(1)*s(2)));

[L,z,history] = robustpca_l2sparsity_mxm_block_splitting(M, mu,lambda,bdim);
% for warm start
% [L,z,history] = robustpca_l2sparsity_mxm_block_splitting(M, mu,lambda,bdim,L,z);

% reporting
K = length(history.objval);

h = figure;
subplot(2,1,1);
plot(1:K, history.objval, 'k', 'MarkerSize', 10, 'LineWidth', 2);
ylabel('f(x^k) + g(z^k)'); xlabel('iter (k)');

subplot(2,1,2);
semilogy(1:K, max(1e-8, history.r_norm), 'k', ...
    1:K, history.eps_pri, 'k--',  'LineWidth', 2);
ylabel('||r||_2');

% Results from the paper
figure;
subplot(3,3,1);imshow(M(:,:,18),[]);
subplot(3,3,2);imshow(L(:,:,18),[]);
subplot(3,3,3);imshow(abs(z(:,:,18)),[]);
subplot(3,3,4);imshow(M(:,:,100),[]);
subplot(3,3,5);imshow(L(:,:,100),[]);
subplot(3,3,6);imshow(abs(z(:,:,100)),[]);
subplot(3,3,7);imshow(M(:,:,199),[]);
subplot(3,3,8);imshow(L(:,:,199),[]);
subplot(3,3,9);imshow(abs(z(:,:,199)),[]);