Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,570 Bytes
b56342d |
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 |
p = genpath('.\out\Stripformer_GoPro_results');% GoPro Deblur Results
gt = genpath('.\datasets\GoPro\test\sharp');% GoPro 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;
|