Initial commit

Author: csjunxu

.DS_Store

@@ -0,0 +1,2 @@
+# Auto detect text files and perform LF normalization
+* text=auto

.gitattributes

@@ -0,0 +1,44 @@
+clc;
+clear;
+close all;
+%%% choose test dataset
+datasets = {'LowLight', 'NPE', 'VV', 'NASA', 'LDR'};
+Testset = datasets{1}; % select test dataset
+Test_dir  = fullfile('/home/csjunxu/Paper/Enhancement/Dataset', ['Images_' Testset]);
+%%% read images
+ext         =  {'*.jpg','*.jpeg','*.JPG','*.png','*.bmp'};
+im_dir   =  [];
+for i = 1 : length(ext)
+    im_dir = cat(1,im_dir, dir(fullfile(Test_dir,ext{i})));
+end
+im_num = length(im_dir);
+ 
+%%% save results
+N=11;
+name = regexp(im_dir(N).name, '\.', 'split');
+I=im2double( imread(fullfile(Test_dir, im_dir(N).name)) );
+%I=im2double(imread('correction/img/books.png'));
+for scale=1.2
+    for alpha=0.001
+        for beta=0.0001
+            for pI=2
+                for pR=1
+                    for ch=1:size(I,3)
+                        [L(:,:,ch),R(:,:,ch)]=STAR(I(:,:,ch),alpha,beta,pI,pR);
+                    end
+                    r=mean(reshape(R,size(R,1)*size(R,2),size(R,3)));
+                    for i=1:3
+                        temp=L(:,:,i);
+                        meanLi=mean(temp(:));
+                        L2(:,:,i)=L(:,:,i)./meanLi.*(scale*mean(L(:))*r(i));
+                    end
+                    I_c=L2.*R;
+                    %imshow(R);
+                    %imshow(L2);
+                    %imshow(I_c);
+                    imwrite(I_c,['/home/csjunxu/Paper/Enhancement/Results_Color/books_star_' num2str(scale) '_' num2str(alpha) '_' num2str(beta) '_' num2str(pI) '_' num2str(pR) '.png']);
+                end
+            end
+        end
+    end
+end

AblationStudy/29.png

@@ -0,0 +1,59 @@
+clc;clear;
+%%% choose test dataset
+datasets = {'35images', '200images'};
+%%% metrics
+addpath(genpath('metrics'));
+metrics = {'NIQE', 'VIF'};
+% NIQE and VIF: input image is in uint8;
+
+method = 'STAR';
+for d = 1%:length(datasets)
+    Testset = datasets{d}; % select test dataset
+    Test_dir = fullfile('C:\Users\csjunxu\Desktop\star\', Testset);
+    %%% read images
+    ext      =  {'*.jpg','*.jpeg','*.png','*.bmp'};
+    im_dir   =  [];
+    for i = 1 : length(ext)
+        im_dir = cat(1,im_dir, dir(fullfile(Test_dir,ext{i})));
+    end
+    im_num = length(im_dir);
+    write_mat_dir = ['/home/csjunxu/Paper/Enhancement/Results_' Testset '/'];
+    write_img_dir = [write_mat_dir method '/'];
+    if ~isdir(write_img_dir)
+        mkdir(write_img_dir);
+    end 
+    gamma=2.2; % fixed parameter suggested by other researchers
+    alpha = 0.001;
+    beta = 0.0001;
+    for pI = 1% [1.6:0.1:1.9]  
+        for pR = 1 % [0.5 1]
+            NIQEs = zeros(im_num,1);
+            VIFs = zeros(im_num,1);
+            for i = 23%1:im_num 
+                name = regexp(im_dir(i).name, '\.', 'split');
+                Im=im2double( imread(fullfile(Test_dir, im_dir(i).name)) );
+                [I, R] = STAR(Im, alpha, beta, pI, pR);
+                hsv = rgb2hsv(Im);
+                I_gamma = I.^(1/gamma);
+                S_gamma = R .* I_gamma;
+                hsv(:,:,3) = S_gamma;
+                eIm = hsv2rgb(hsv); 
+                % convert Im and eIm to uint8
+                Im = uint8(Im*255);
+                eIm = uint8(eIm*255);
+                % metrics 
+                NIQEs(i) = niqe(eIm);
+                VIFs(i) = VIF(Im,eIm);
+                fprintf('%s : NIQE = %2.2f, VIF = %2.2f\n', im_dir(i).name, NIQEs(i), VIFs(i));
+                imwrite(eIm, [method '_aIpI=' num2str(pI) '_RpR=' num2str(pR) '_alpha=' ...
+                  num2str(alpha) '_beta=' num2str(beta) '_' name{1} '.png'])
+            end
+            matname = [write_mat_dir '/Our_aIpI=' num2str(pI) '_RpR=' num2str(pR) '_alpha=' ...
+                num2str(alpha) '_beta=' num2str(beta) '.mat'];
+            mNIQEs = mean(NIQEs);
+            mVIFs = mean(VIFs);
+            fprintf('mNIQE = %2.4f, mVIF = %2.4f\n', mNIQEs, mVIFs);
+            %save(matname, 'NIQEs', 'mNIQEs', 'VIFs', 'mVIFs');
+        end
+    end
+end

AblationStudy/29_I_Gray_STAR.png

@@ -0,0 +1,93 @@
+% Retinex theory is a color perception model of human vision and is
+% used to remove illumination effects in images. The primary goal
+% of Retinex is to decompose the observed images into illumination
+% and reflectance.
+
+clc;clear;
+%%% methods
+addpath(genpath('methods'));
+%%% choose test dataset
+datasets = {'LowLight', 'NPE', 'VV', 'NASA', 'LDR'};
+Testset = datasets{1}; % select test dataset
+Test_dir  = fullfile('/home/csjunxu/Paper/Enhancement/Dataset', ['Images_' Testset]);
+%%% read images
+ext         =  {'*.jpg','*.jpeg','*.JPG','*.png','*.bmp'};
+im_dir   =  [];
+for i = 1 : length(ext)
+    im_dir = cat(1,im_dir, dir(fullfile(Test_dir,ext{i})));
+end
+im_num = length(im_dir);
+ 
+%%% save results
+N=32;
+name = regexp(im_dir(N).name, '\.', 'split');
+Im=im2double( imread(fullfile(Test_dir, im_dir(N).name)) );
+write_dir = './';
+if ~isdir(write_dir)
+    mkdir(write_dir);
+end
+imwrite(Im, [write_dir name{1} '.png'])
+
+% STAR   
+method = 'STAR';
+alpha = 0.001;
+beta = 0.0001;  
+for pI = [1.5] 
+    for pR = [1]  
+        [I, R] = STAR(Im, alpha, beta, pI, pR);
+        hsv = rgb2hsv(Im);
+        subplot(2,2,1); imshow(I);  title('Illumination (Gray)');
+        imwrite(I, [write_dir name{1} '_I_Gray_' method '_pI' num2str(pI) '_pR' num2str(pR) '.png'])
+        hsv(:,:,3) = I;
+        subplot(2,2,2); imshow(hsv2rgb(hsv));  title('Illumination (RGB)');
+        imwrite(hsv2rgb(hsv), [write_dir name{1} '_I_RGB_' method '_pI' num2str(pI) '_pR' num2str(pR) '.png'])
+        subplot(2,2,3); imshow(I);  title('Reflectance (Gray)');
+        imwrite(R, [write_dir name{1} '_R_Gray_' method '_pI' num2str(pI) '_pR' num2str(pR) '.png'])
+        hsv(:,:,3) = R;
+        subplot(2,2,4); imshow(hsv2rgb(hsv));  title('Reflectance (RGB)');
+        imwrite(hsv2rgb(hsv), [write_dir name{1} '_R_RGB_' method '_pI' num2str(pI) '_pR' num2str(pR) '.png'])
+    end
+end     
+
+% JIEP ICCV2017
+method = 'JieP';
+[I, R] = jiep(Im);
+hsv = rgb2hsv(Im);
+subplot(2,2,1); imshow(I);  title('Illumination (Gray)');
+imwrite(I, [write_dir name{1} '_I_Gray_' method '.png'])
+hsv(:,:,3) = I;
+subplot(2,2,2); imshow(hsv2rgb(hsv));  title('Illumination (RGB)');
+imwrite(hsv2rgb(hsv), [write_dir name{1} '_I_RGB_' method '.png'])
+subplot(2,2,3); imshow(I);  title('Reflectance (Gray)');
+imwrite(R, [write_dir name{1} '_R_Gray_' method '.png'])
+hsv(:,:,3) = R;
+subplot(2,2,4); imshow(hsv2rgb(hsv));  title('Reflectance (RGB)');
+imwrite(hsv2rgb(hsv), [write_dir name{1} '_R_RGB_' method '.png'])
+%subplot(2,2,1); imshow(Im);  title('Input');
+
+% WVM CVPR2016
+method = 'WVM';
+Im = 255*Im;
+if size(Im,3)>1
+    HSV = rgb2hsv(Im);   % RGB space to HSV  space
+    S = HSV(:,:,3);       % V layer
+else
+    S = Im;              % gray image
+end
+c_1 = 0.01; c_2 = 0.1; lambda = 1;     % set parameters
+epsilon_stop = 1e-3;  % stopping criteria
+[ R, I, epsilon_R, epsilon_L ] = WVM_CVPR2016( S, c_1, c_2, lambda, epsilon_stop );
+Im = Im/255;
+I = I/255;
+hsv = rgb2hsv(Im);
+subplot(2,2,1); imshow(I);  title('Illumination (Gray)');
+imwrite(I, [write_dir name{1} '_I_Gray_' method '.png'])
+hsv(:,:,3) = I;
+subplot(2,2,2); imshow(hsv2rgb(hsv));  title('Illumination (RGB)');
+imwrite(hsv2rgb(hsv), [write_dir name{1} '_I_RGB_' method '.png'])
+subplot(2,2,3); imshow(I);  title('Reflectance (Gray)');
+imwrite(R, [write_dir name{1} '_R_Gray_' method '.png'])
+hsv(:,:,3) = R;
+subplot(2,2,4); imshow(hsv2rgb(hsv));  title('Reflectance (RGB)');
+imwrite(hsv2rgb(hsv), [write_dir name{1} '_R_RGB_' method '.png'])
+%subplot(2,2,1); imshow(Im);  title('Input');

AblationStudy/29_I_Gray_STAR_L1.png

@@ -0,0 +1,44 @@
+function [ outputimage ] = DrawBox( image, top, left, width, rgbarray)
+%UNTITLED Summary of this function goes here
+%   Detailed explanation goes here
+
+[m, n, c] = size(image);
+if(c==3)
+    outputimage = image;
+else
+    outputimage(:,:,1) = image;
+    outputimage(:,:,2) = image;
+    outputimage(:,:,3) = image;
+end
+r = rgbarray(1);
+g = rgbarray(2);
+b = rgbarray(3);
+
+outputimage(top:top+width,left,1) = r;
+outputimage(top:top+width,left,2) = g;
+outputimage(top:top+width,left,3) = b;
+outputimage(top:top+width,left+1,1) = r;
+outputimage(top:top+width,left+1,2) = g;
+outputimage(top:top+width,left+1,3) = b;
+outputimage(top:top+width,left+width,1) = r;
+outputimage(top:top+width,left+width,2) = g;
+outputimage(top:top+width,left+width,3) = b;
+outputimage(top:top+width,left+width-1,1) = r;
+outputimage(top:top+width,left+width-1,2) = g;
+outputimage(top:top+width,left+width-1,3) = b;
+
+outputimage(top,left:left+width,1) = r;
+outputimage(top,left:left+width,2) = g;
+outputimage(top,left:left+width,3) = b;
+outputimage(top+1,left:left+width,1) = r;
+outputimage(top+1,left:left+width,2) = g;
+outputimage(top+1,left:left+width,3) = b;
+outputimage(top+width,left:left+width,1) = r;
+outputimage(top+width,left:left+width,2) = g;
+outputimage(top+width,left:left+width,3) = b;
+outputimage(top+width-1,left:left+width,1) = r;
+outputimage(top+width-1,left:left+width,2) = g;
+outputimage(top+width-1,left:left+width,3) = b;
+
+end
+

AblationStudy/29_I_Gray_STAR_L2.png

@@ -0,0 +1,13 @@
+clear;
+Original_image_dir  =    './';
+fpath = fullfile(Original_image_dir, '29*L2.png');
+im_dir  = dir(fpath);
+im_num = length(im_dir);
+Im1 = im2double(imread(fullfile(Original_image_dir, im_dir(1).name)));
+Im2 = im2double(imread(fullfile(Original_image_dir, im_dir(2).name)));
+RIm1 = imresize(Im1, 0.5);
+RIm2 = imresize(Im2, 0.5);
+RIm = zeros(size(RIm1));
+RIm(:,1:125)=RIm1(:,1:125);
+RIm(:,126:250)=RIm2(:,126:250);
+imwrite(RIm, ['resize_29_IR_STAR_L2.png']);

AblationStudy/29_I_Gray_STAR_noT.png

@@ -0,0 +1,15 @@
+clear;
+scale = [256, 256];
+ii = 100;
+Original_image_dir  =    '../Priors/';
+fpath = fullfile(Original_image_dir, '*.png');
+im_dir  = dir(fpath);
+im_num = length(im_dir);
+for i=1:im_num
+    Im = im2double(imread(fullfile(Original_image_dir, im_dir(i).name)));
+    [hh, ww, cc] = size(Im);
+    ll = min(hh,ww);
+    image = imresize(Im(1:ll,ii:ll+ii-1,:), scale);
+    imname = ['rs' im_dir(i).name];
+    imwrite(image,imname,'png');
+end

AblationStudy/29_I_Gray_STAR_pI05_pR15.png

@@ -0,0 +1,15 @@
+clear;
+scale = [256, 256];
+ii = 100;
+Original_image_dir  =    '../Priors/';
+fpath = fullfile(Original_image_dir, '*.png');
+im_dir  = dir(fpath);
+im_num = length(im_dir);
+for i=1:im_num
+    Im = im2double(imread(fullfile(Original_image_dir, im_dir(i).name)));
+    [hh, ww, cc] = size(Im);
+    ll = min(hh,ww);
+    image = imresize(Im(1:ll,ii:ll+ii-1,:), scale);
+    imname = ['rs');
+    imwrite(image,imname,'png');
+end

AblationStudy/29_I_Gray_STAR_pI1_pR1.png

@@ -0,0 +1,23 @@
+function [ outputimage ] = boxandresize( image, top, left, width, scale, model )
+%UNTITLED3 Summary of this function goes here
+%   Detailed explanation goes here
+% top 框顶坐标
+% left 框左边坐标
+% width 框宽度
+% scale 放大倍数
+% model 模式
+outputimage = DrawBox( image, top, left, width, [0,255,0]);
+
+smallimage = outputimage(top:top+width,left:left+width,:);
+bigimage = imresize(smallimage,scale, 'bic');
+[m ,n, c] = size(bigimage);
+if(model==1)    %右下角
+    outputimage(end-m+1:end,end-n+1:end,:) = bigimage;
+elseif(model==2)   %左下角
+    outputimage(end-m+1:end,1:n,:) = bigimage;
+else
+        outputimage(1:m,end-n+1:end,:) = bigimage;
+end
+
+end
+

AblationStudy/29_I_Gray_noS.png

@@ -0,0 +1,55 @@
+function createfigure(X1, Y1, Y2)
+%CREATEFIGURE1(X1, Y1, Y2)
+%  X1:  x 数据的矢量
+%  Y1:  y 数据的矢量
+%  Y2:  y 数据的矢量
+
+%  由 MATLAB 于 15-Oct-2016 13:14:45 自动生成
+
+% 创建 figure
+figure1 = figure;
+
+% 创建 subplot
+subplot1 = subplot(2,1,1,'Parent',figure1);
+%% 取消以下行的注释以保留坐标轴的 X 范围
+% xlim(subplot1,[0.001 0.004]);
+%% 取消以下行的注释以保留坐标轴的 Y 范围
+% ylim(subplot1,[38 38.6]);
+box(subplot1,'on');
+hold(subplot1,'on');
+
+% 创建 ylabel
+ylabel('PSNR (dB)');
+
+% 创建 xlabel
+xlabel('\lambda');
+
+% 创建 plot
+plot(X1,Y1,'Parent',subplot1,'MarkerFaceColor',[0 0 0],'MarkerSize',3,...
+    'Marker','o',...
+    'LineWidth',1,...
+    'LineStyle','-.',...
+    'Color',[0 0 1]);
+
+% 创建 subplot
+subplot2 = subplot(2,1,2,'Parent',figure1);
+%% 取消以下行的注释以保留坐标轴的 X 范围
+% xlim(subplot2,[0.001 0.004]);
+%% 取消以下行的注释以保留坐标轴的 Y 范围
+% ylim(subplot2,[0.96 0.97]);
+box(subplot2,'on');
+hold(subplot2,'on');
+
+% 创建 ylabel
+ylabel('SSIM');
+
+% 创建 xlabel
+xlabel('\lambda');
+
+% 创建 plot
+plot(X1,Y2,'Parent',subplot2,'MarkerFaceColor',[0 0 0],'MarkerSize',3,...
+    'Marker','o',...
+    'LineWidth',1,...
+    'LineStyle','-.',...
+    'Color',[1 0 0]);
+