lime_main_module.m

function [Ti,Tout,img_out,Iout] = lime_main_module(img_in,mu,rho,ds,ss,flag)
%main module incluidng all operations for LIME

%img_in is raw image
%alpha, mu, rho are constant parameters for the solver
%ds is degree of smoothing and ss is spatial smoothing for bilateral filter
%flag kept 1 for displaying outputs

%Ti is initial illumination map
%Tout is refined Ti
%img_out is enhanced image
%I_out is img_out after denoising

alpha=0.08; %same value for all images
gamma=0.8; %parameter for gamma correction

[Tout,Ti]=lime(img_in,alpha,mu,rho,gamma);
img_out=im2double(img_in)./Tout;

Iout=imbilatfilt(img_out,ds,ss);

if flag==1
    subplot(2,3,1);imshow(img_in);
    subplot(2,3,2);colormap('hot');imagesc(Ti(:,:,1));colorbar; %heat map
    subplot(2,3,3);colormap('hot');imagesc(Tout(:,:,1));colorbar;
    subplot(2,3,4);imshow(img_out);
    subplot(2,3,5);imshow(Iout);
end

end