forked from codeslake/PVDNet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun_evaluation.m
executable file
·38 lines (30 loc) · 1.59 KB
/
run_evaluation.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
%% init
clear all; close all;
addpath('evaluation_code');
addpath('evaluation_code\image_quality_algorithms');
addpath('evaluation_code\image_quality_algorithms\metrix_mux');
addpath('evaluation_code\image_quality_algorithms\metrix_mux\metrix');
addpath('evaluation_code\image_quality_algorithms\metrix_mux\metrix\ssim');
delete(gcp('nocreate'));
parpool(4);
%% gt path
% test_offet = './datasets';
test_offset = '\\mango.postech.ac.kr\Users\junyonglee\hub\datasets\video_deblur';
%% result path
% result_root = './logs/PVDNet_TOG2021';
result_root = '\\jarvis.postech.ac.kr\logs\junyonglee\PVDNet_TOG2021';
% PVDNet_DVD
test_model = 'PVDNet_DVD';
folder_result = fullfile(result_root, 'PVDNet_DVD\result\quanti_quali\PVDNet_DVD\DVD\2021_04_15_0131\png\output');
psnr_mean, ssim_mean = evaluation(test_model, folder_result, 'DVD', test_offset);
fprintf('PSNR (Koehler): %f, SSIM (Koehler): %f', psnr_mean, ssim_mean);
% % PVDNet_nah
% test_model = 'PVDNet_nah';
% folder_result = fullfile(result_root, 'PVDNet_nah\result\quanti_quali\PVDNet_nah\nah\2021_04_15_0131\png\output');
% psnr_mean, ssim_mean = evaluation(test_model, folder_result, 'nah', test_offset);
% fprintf('PSNR (Koehler): %f, SSIM (Koehler): %f', psnr_mean, ssim_mean);
% % PVDNet_large_nah
% test_model = 'PVDNet_large_nah';
% folder_result = fullfile(result_root, 'PVDNet_large_nah\result\quanti_quali\PVDNet_large_nah\nah\2021_04_15_0132\png\output');
% psnr_mean, ssim_mean = evaluation(test_model, folder_result, 'nah', test_offset);
% fprintf('PSNR (Koehler): %f, SSIM (Koehler): %f', psnr_mean, ssim_mean);