Demo_Dictionary_Training.m

% ========================================================================
% Demo codes for dictionary training by joint sparse coding
% 
% Reference
%   J. Yang et al. Image super-resolution as sparse representation of raw
%   image patches. CVPR 2008.
%   J. Yang et al. Image super-resolution via sparse representation. IEEE 
%   Transactions on Image Processing, Vol 19, Issue 11, pp2861-2873, 2010
%
% Jianchao Yang
% ECE Department, University of Illinois at Urbana-Champaign
% For any questions, send email to jyang29@uiuc.edu
% =========================================================================
function  Demo_Dictionary_Training(type, TR_IMG_PATH, dict_size, patch_size)


%clear all; clc; close all;
addpath(genpath('RegularizedSC'));

%TR_IMG_PATH = 'Data/Training';

%dict_size   = 512;          % dictionary size
lambda      = 0.15;         % sparsity regularization
%patch_size  = 5;            % image patch size
nSmp        = 100000;       % number of patches to sample
upscale     = 2;            % upscaling factor

% randomly sample image patches
[Xh, Xl] = rnd_smp_patch(TR_IMG_PATH, '*.bmp', patch_size, nSmp, upscale);

% prune patches with small variances, threshould chosen based on the
% training data
[Xh, Xl] = patch_pruning(Xh, Xl, 10);

% joint sparse coding 
[Dh, Dl] = train_coupled_dict(Xh, Xl, dict_size, lambda, upscale);
dict_path = ['Dictionary/D_' type '_' num2str(dict_size) '_' num2str(lambda) '_' num2str(patch_size) '_s' num2str(upscale) '_bp' num2str(patch_size) '.mat' ];
save(dict_path, 'Dh', 'Dl');

end