-
Notifications
You must be signed in to change notification settings - Fork 0
/
Evaluation_for_Single_Algorithm.m
136 lines (127 loc) · 6.34 KB
/
Evaluation_for_Single_Algorithm.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
clc
clear all
easy = 0;
dataset = 'TNO'
row_name1 = 'row1';
row_data1 = 'row2';
Method_name = 'SeAFusion';
row = 'A';
row_name = strrep(row_name1, 'row', row);
row_data = strrep(row_data1, 'row', row);
fileFolder=fullfile('../Image/Source-Image/TNO/ir'); % 源图像A所在文件夹 此处是'Evaluation\Image\Source-Image\TNO\ir'
dirOutput=dir(fullfile(fileFolder,'*.*'));
fileNames = {dirOutput.name};
[m, num] = size(fileNames);
ir_dir = fullfile('../Image/Source-Image/TNO/ir'); % 源图像A所在文件夹 此处是'Evaluation\Image\Source-Image\TNO\ir'
vi_dir = fullfile('../Image/Source-Image/TNO/vi'); % 源图像B所在文件夹 此处是'Evaluation\Image\Source-Image\TNO\vi'
Fused_dir = fullfile('../Image/Algorithm/SeAFusion_TNO'); % 融合结果所在文件夹 此处是 'Evaluation\Image\Algorithm\SeAFusion_TNO'
EN_set = []; SF_set = [];SD_set = [];PSNR_set = [];
MSE_set = [];MI_set = [];VIF_set = []; AG_set = [];
CC_set = [];SCD_set = []; Qabf_set = [];
SSIM_set = []; MS_SSIM_set = [];
Nabf_set = [];FMI_pixel_set = [];
FMI_dct_set = []; FMI_w_set = [];
for j = 1:num
if (isequal(fileNames{j}, '.') || isequal(fileNames{j}, '..'))
continue;
else
fileName_source_ir = fullfile(ir_dir, fileNames{j});
fileName_source_vi = fullfile(vi_dir, fileNames{j});
fileName_Fusion = fullfile(Fused_dir, fileNames{j});
ir_image = imread(fileName_source_ir);
vi_image = imread(fileName_source_vi);
fused_image = imread(fileName_Fusion);
if size(ir_image, 3)>2
ir_image = rgb2gray(ir_image);
end
if size(vi_image, 3)>2
vi_image = rgb2gray(vi_image);
end
if size(fused_image, 3)>2
fused_image = rgb2gray(fused_image);
end
[m, n] = size(fused_image);
% fused_image = fused_image(7:m-6, 7:n-6);
ir_size = size(ir_image);
vi_size = size(vi_image);
fusion_size = size(fused_image);
if length(ir_size) < 3 && length(vi_size) < 3
[EN, SF,SD,PSNR,MSE, MI, VIF, AG, CC, SCD, Qabf, Nabf, SSIM, MS_SSIM, FMI_pixel, FMI_dct, FMI_w] = analysis_Reference(fused_image,ir_image,vi_image, easy);
EN_set = [EN_set, EN];SF_set = [SF_set,SF];SD_set = [SD_set, SD];PSNR_set = [PSNR_set, PSNR];
MSE_set = [MSE_set, MSE];MI_set = [MI_set, MI]; VIF_set = [VIF_set, VIF];
AG_set = [AG_set, AG]; CC_set = [CC_set, CC];SCD_set = [SCD_set, SCD];
Qabf_set = [Qabf_set, Qabf]; Nabf_set = [Nabf_set, Nabf];
SSIM_set = [SSIM_set, SSIM]; MS_SSIM_set = [MS_SSIM_set, MS_SSIM];
FMI_pixel_set = [FMI_pixel_set, FMI_pixel]; FMI_dct_set = [FMI_dct_set,FMI_dct];
FMI_w_set = [FMI_w_set, FMI_w];
else
disp('unsucessful!')
disp( fileName_Fusion)
end
fprintf('Fusion Method:%s, Image Name: %s\n', Method_name, fileNames{j})
end
end
save_dir = '../Metric'; %存放Excel结果的文件夹
if exist(save_dir,'dir')==0
mkdir(save_dir);
end
file_name = fullfile(save_dir, strcat('Metric_', Method_name, '.xlsx')); %存放Excel文件的文件名
%% 将测试结果写入 Excel， 此处采用writetable， 第一行可能会有问题，算法名在第二行，评估结果从第三行开始
if easy ==1
SD_table = table(SD_set');
PSNR_table = table(PSNR_set');
MSE_table = table(MSE_set');
MI_table = table(MI_set');
VIF_table = table(VIF_set');
AG_table = table(AG_set');
CC_table = table(CC_set');
SCD_table = table(SCD_set');
EN_table = table(EN_set');
Qabf_table = table(Qabf_set');
SF_table = table(SF_set');
method_name = cellstr(Method_name);
method_table = table(method_name);
writetable(SD_table,file_name,'Sheet','SD','Range',row_data);
writetable(PSNR_table,file_name,'Sheet','PSNR','Range',row_data);
writetable(MSE_table,file_name,'Sheet','MSE','Range',row_data);
writetable(MI_table,file_name,'Sheet','MI','Range',row_data);
writetable(VIF_table,file_name,'Sheet','VIF','Range',row_data);
writetable(AG_table,file_name,'Sheet','AG','Range',row_data);
writetable(CC_table,file_name,'Sheet','CC','Range',row_data);
writetable(SCD_table,file_name,'Sheet','SCD','Range',row_data);
writetable(EN_table,file_name,'Sheet','EN','Range',row_data);
writetable(Qabf_table,file_name,'Sheet','Qabf','Range',row_data);
writetable(SF_table,file_name,'Sheet','SF','Range',row_data);
writetable(method_table,file_name,'Sheet','SD','Range',row_name);
writetable(method_table,file_name,'Sheet','PSNR','Range',row_name);
writetable(method_table,file_name,'Sheet','MSE','Range',row_name);
writetable(method_table,file_name,'Sheet','MI','Range',row_name);
writetable(method_table,file_name,'Sheet','VIF','Range',row_name);
writetable(method_table,file_name,'Sheet','AG','Range',row_name);
writetable(method_table,file_name,'Sheet','CC','Range',row_name);
writetable(method_table,file_name,'Sheet','SCD','Range',row_name);
writetable(method_table,file_name,'Sheet','EN','Range',row_name);
writetable(method_table,file_name,'Sheet','Qabf','Range',row_name);
writetable(method_table,file_name,'Sheet','SF','Range',row_name);
else
Nabf_table = table(Nabf_set');
SSIM_table = table(SSIM_set');
MS_SSIM_table = table(MS_SSIM_set');
FMI_pixel_table = table(FMI_pixel_set');
FMI_dct_table = table(FMI_dct_set');
FMI_w_table = table(FMI_w_set');
method_name = cellstr(Method_name);
method_table = table(method_name);
writetable(Nabf_table,file_name,'Sheet','Nabf','Range',row_data);
writetable(SSIM_table,file_name,'Sheet','SSIM','Range',row_data);
writetable(MS_SSIM_table,file_name,'Sheet','MS_SSIM','Range',row_data);
writetable(FMI_pixel_table,file_name,'Sheet','FMI_pixel','Range',row_data);
writetable(FMI_dct_table,file_name,'Sheet','FMI_dct','Range',row_data);
writetable(FMI_w_table,file_name,'Sheet','FMI_w','Range',row_data);
writetable(method_table,file_name,'Sheet','Nabf','Range',row_name);
writetable(method_table,file_name,'Sheet','SSIM','Range',row_name);
writetable(method_table,file_name,'Sheet','MS_SSIM','Range',row_name);
writetable(method_table,file_name,'Sheet','FMI_pixel','Range',row_name);
writetable(method_table,file_name,'Sheet','FMI_dct','Range',row_name);
writetable(method_table,file_name,'Sheet','FMI_w','Range',row_name);
end