runDehazing.m

close all;clear;clc;

inputImagePath = 'inputImgs/tree2.png';
I = imread(inputImagePath);
r = 15;
beta = 1.0;

%----- Parameters for Guided Image Filtering -----
gimfiltR = 60;
eps = 10^-3;
%-------------------------------------------------
tic;
[dR, dP] = calVSMap(I, r);
refineDR = fastguidedfilter_color(double(I)/255, dP, r, eps, r/4);
%refineDR = imguidedfilter(dR, double(I)/255, 'NeighborhoodSize', [gimfiltR, gimfiltR], 'DegreeOfSmoothing', eps);
tR = exp(-beta*refineDR);
tP = exp(-beta*dP);

imwrite(dR, 'res/originalDepthMap.png');
imwrite(refineDR, 'res/refineDepthMap.png');

figure;
imshow([dP dR refineDR]);
title('depth maps');

figure;
imshow([tP tR]);
title('transmission maps');
imwrite(tR, 'res/transmission.png');

a = estA(I, dR);
t0 = 0.05;
t1 = 1;
I = double(I)/255;
[h w c] = size(I);
J = zeros(h,w,c);
J(:,:,1) = I(:,:,1)-a(1);
J(:,:,2) = I(:,:,2)-a(2);
J(:,:,3) = I(:,:,3)-a(3);

t = tR;
[th tw] = size(t);
for y=1:th
    for x=1:tw
        if t(y,x)<t0
            t(y,x)=t0;
        end
    end
end

for y=1:th
    for x=1:tw
        if t(y,x)>t1
            t(y,x)=t1;
        end
    end
end

J(:,:,1) = J(:,:,1)./t;
J(:,:,2) = J(:,:,2)./t;
J(:,:,3) = J(:,:,3)./t;

J(:,:,1) = J(:,:,1)+a(1);
J(:,:,2) = J(:,:,2)+a(2);
J(:,:,3) = J(:,:,3)+a(3);

toc;
figure;
imshow([I J]);
title('hazy image and dehazed image');

saveName = ['res/' num2str(r) '_beta' num2str(beta) '.png'];
imwrite(J, saveName);