fanwuyang
/
Stripformer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
main
pp00704831 3 years ago committed by GitHub
parent c76e6afef8
commit f8ac53b7a7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 120 additions and 120 deletions
Show Stats Download Patch File Download Diff File

120
evaluation_GoPro.m
Unescape View File

@ -1,60 +1,60 @@
p = genpath('.\out\stripformer_10_patches');% 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;
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;

120
evaluation_HIDE.m
Unescape View File

@ -1,60 +1,60 @@
p = genpath('.\out\Stripformer_HIDE_result');% 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;
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;

Write Preview
Loading…
Cancel
Save