通过深度矩阵生成雾图算法（HazeRD）

目录

 

官方：

matlab代码：

完整文件百度云下载：

python版本更新

---

---

官方：

 论文： HazeRD: An Outdoor Scene Dataset and Benchmark for Single Image Dehazing

网址： https://labsites.rochester.edu/gsharma/research/computer-vision/hazerd/

matlab代码：

demo_simu_haze.m主程序

% This is a demo script for haze simulation with the dataset and parameters given by
% the paper: 
% HAZERD: an outdoor scene dataset and benchmark for single image dehazing
% IEEE Internation Conference on Image Processing, Sep 2017
% The paper and additional information on the project are available at:
% https://labsites.rochester.edu/gsharma/research/computer-vision/hazerd/
% If you use this code, please cite our paper.
%
% Set different values to variables (pert_perlin, airlight, and
% visual_range) to generate different hazy images
%
% The simulated images will be save at /data/simu/
%
% Authors: 
% Yanfu Zhang: yzh185@ur.rochester.edu
% Li Ding: l.ding@rochester.edu
% Gaurav Sharma: gaurav.sharma@rochester.edu
%
% Last update: May 2017

clear;clc;

img_dir = 'data/img/'; img_list = dir(img_dir);
depth_dir = 'data/depth/'; depth_list = dir(depth_dir);
save_dir = 'data/simu/';

addpath('./utils');

% set parameters
pert_perlin = 0;    % 1 for add perlin noise
airlight = 0.76*ones(3,1);  % airlight in the range [0,1]
visual_range = [0.05,0.1,0.2,0.5,1]; % visual range in km

for j = 1:length(img_list)-2
    img_name= [img_dir,img_list(j+2).name];
    depth_name = [depth_dir,depth_list(j+2).name];
    hazy_img = hazy_simu(img_name,depth_name,save_dir,pert_perlin,airlight,visual_range);
end

hazy_simu.m函数

function save_name = hazy_simu(img_name,depth_name,save_dir,varargin)
% This is the function for haze simulation with the parameters given by
% the paper: 
% HAZERD: an outdoor scene dataset and benchmark for single image dehazing
% IEEE Internation Conference on Image Processing, Sep 2017
% The paper and additional information on the project are available at:
% https://labsites.rochester.edu/gsharma/research/computer-vision/hazerd/
% If you use this code, please cite our paper.
%
% Input:
%   img_name: the directory and name of a haze-free RGB image, the name 
%             should be in the format of ..._RGB.jpg 
%   depth_name: the corresponding directory and name of the depth map, in
%               .mat file, the name should be in the format of ..._depth.mat
%   save_dir: the directory to save the simulated images
%   varargin: optional
%   varargin{1}: 1 for adding perlin noise, default 0
%   varargin{2}: airlight, 3*1 matrix in the range [0,1]
%   varargin{3}: visual range, a vector of any size 
%
% Output:
%   save_name: image name of hazy image
%
% IMPORTANT NOTE: The code uses the convention that pixel locations with a
% depth value of 0 correspond to objects that are very far and for the
% simulation of haze these are placed a distance of 2 times the visual
% range.
%
% Authors: 
%   Yanfu Zhang: yzh185@ur.rochester.edu
%   Li Ding: l.ding@rochester.edu
%   Gaurav Sharma: gaurav.sharma@rochester.edu
% 
% Last update: May 2017

% parse inputs and set default values
% Set default parameter values. Some of these are used only if they are not
% passed in
pert_perlin = 0;
airlight=0.76;
beta_param = 3.912; % Default beta parameter corresponding to visual range of 1000m
A = zeros(1,1,3);
A(1,1,1) = airlight;A(1,1,2) = airlight;A(1,1,3) = airlight;
visual_range = [0.05,0.1,0.2,0.5,1]; % km
if ~isempty(varargin)
    pert_perlin = varargin{1};
end
if length(varargin) > 1
    A(1,1,1) = varargin{2}(1);A(1,1,2) = varargin{2}(2);A(1,1,3) = varargin{2}(3);
end
if length(varargin) > 2
    visual_range = varargin{3};
end

if exist(save_dir,'dir') ~= 7
    mkdir(save_dir);        % make dir if not exists
end

fprintf('Simulating hazy image for: %s\n',img_name);
for VR = visual_range
    fprintf('    Viusal range: %.3f km\n',VR);
    I0 = im2double(imread(img_name));
    
    % convert sRGB to linear RGB
    I = srgb2lrgb(I0);
    
    load(depth_name);   
    d = imDepth/1000;   % convert meter to kilometer
    
%   Set regions where depth value is set to 0 to indicate no valid depth to
%   a distance of two times the visual range. These regions typically
%   correspond to sky areas
    d(d==0) = 2*VR;
    	
%	Add perlin noise for visual vividness(see reference 16 in the HazeRD paper). The noise is added to the depth 
%	map, which is equivalent to change the optic distance. 
    if pert_perlin
        d = d.*((perlin_noise(zeros(size(d)))-0.5)+1);
    end
    
    % convert depth map to transmission
    beta = [beta_param/VR,beta_param/VR,beta_param/VR];beta = reshape(beta,[1,1,3]);
    transmission = exp(bsxfun(@times,-beta,d));
    
    % Obtain simulated linear RGB hazy image. Eq. 3 in the HazeRD paper
    Ic = bsxfun(@times,transmission,I)+bsxfun(@times,1-transmission,A);
    
    % convert linear RGB to sRGB
    I2 = lrgb2srgb(Ic);
    
    % save result
    [~,name,ext] = fileparts(img_name); 
    save_name = strrep(name,'_RGB',['_',num2str(VR*1000),ext]);
    imwrite(I2,[save_dir,save_name]);
    [~,name,ext] = fileparts(depth_name); 
    save_name_trans = strrep(name,'_depth',['_trans_',num2str(VR*1000),ext]);
    save([save_dir,save_name_trans],'transmission');
end
fprintf('\n');

end

function I = srgb2lrgb(I0)
    I = ((I0+0.055)/1.055).^(2.4);  
    I(I0<=0.04045) = I0(I0<=0.04045)/12.92;
end

function I2 = lrgb2srgb(I1)
    I2 = zeros(size(I1));
    for k = 1:3
        temp = I1(:,:,k);
        I2(:,:,k) = 12.92*temp.*(temp<=0.0031308)+(1.055*temp.^(1/2.4)-0.055).*(temp>0.0031308);
    end
end

perlin_noise.m函数

function im = perlin_noise(im,varargin)
% This is the function for adding perlin noise to the depth map. It is a 
% simplified implementation of the paper: 
% an image sunthesizer
% Ken Perlin, SIGGRAPH, Jul. 1985
% The bicubic interpolation is used, compared to the original version.
%
% Reference:
% HAZERD: an outdoor scene dataset and benchmark for single image dehazing
% IEEE International Conference on Image Processing, Sep 2017
% The paper and additional information on the project are available at:
% https://labsites.rochester.edu/gsharma/research/computer-vision/hazerd/
% If you use this code, please cite our paper.
%
% Input:
%   im: depth map
%   varargin{1}: decay term
% Output:
%   im: result of transmission with perlin noise added
%
% Authors: 
%   Yanfu Zhang: yzh185@ur.rochester.edu
%   Li Ding: l.ding@rochester.edu
%   Gaurav Sharma: gaurav.sharma@rochester.edu
% 
% Last update: May 2017


    [h,w] = size(im);
    i = 1;
    if nargin == 1
        decay = 2;
    else
        decay = varargin{1};
    end
    l_bound = min([h,w]);
    while i <= l_bound
        d = imresize(randn(i, i)*decay, size(im), 'bicubic');
        im = im+d;
        i = i*2;
    end
    im = mat2gray(im);
end

完整文件百度云下载：

链接：https://pan.baidu.com/s/1DgfwitXkSh1U24Ax24ii9g 
提取码：Haze

python版本更新

【python】通过深度图生成雾图（HAZERD）_Alocus的博客-CSDN博客