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;