You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.5 KiB
61 lines
1.5 KiB
p = genpath('.\out\Stripformer_HIDE_results');% HIDE Deblur Results
|
|
gt = genpath('.\datasets\HIDE\sharp');% HIDE GT Results
|
|
|
|
length_p = size(p,2);
|
|
path = {};
|
|
temp = [];
|
|
for i = 1:length_p
|
|
if p(i) ~= ';'
|
|
temp = [temp p(i)];
|
|
else
|
|
temp = [temp '\'];
|
|
path = [path ; temp];
|
|
temp = [];
|
|
end
|
|
end
|
|
clear p length_p temp;
|
|
length_gt = size(gt,2);
|
|
path_gt = {};
|
|
temp_gt = [];
|
|
for i = 1:length_gt
|
|
if gt(i) ~= ';'
|
|
temp_gt = [temp_gt gt(i)];
|
|
else
|
|
temp_gt = [temp_gt '\'];
|
|
path_gt = [path_gt ; temp_gt];
|
|
temp_gt = [];
|
|
end
|
|
end
|
|
clear gt length_gt temp_gt;
|
|
|
|
file_num = size(path,1);
|
|
total_psnr = 0;
|
|
n = 0;
|
|
total_ssim = 0;
|
|
for i = 1:file_num
|
|
file_path = path{i};
|
|
gt_file_path = path_gt{i};
|
|
img_path_list = dir(strcat(file_path,'*.png'));
|
|
gt_path_list = dir(strcat(gt_file_path,'*.png'));
|
|
img_num = length(img_path_list);
|
|
if img_num > 0
|
|
for j = 1:img_num
|
|
image_name = img_path_list(j).name;
|
|
gt_name = gt_path_list(j).name;
|
|
image = imread(strcat(file_path,image_name));
|
|
gt = imread(strcat(gt_file_path,gt_name));
|
|
size(image);
|
|
size(gt);
|
|
peaksnr = psnr(image,gt);
|
|
ssimval = ssim(image,gt);
|
|
total_psnr = total_psnr + peaksnr;
|
|
total_ssim = total_ssim + ssimval;
|
|
n = n + 1
|
|
end
|
|
end
|
|
end
|
|
psnr = total_psnr / n
|
|
ssim = total_ssim / n
|
|
close all;clear all;
|
|
|