X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=eval_voxelflow.m;fp=eval_voxelflow.m;h=413f8d74031567c40780d3461a772e0c428a6e40;hb=6bfc2f72ff168a9f9ed2eb1a014cfb4420ae5993;hp=0000000000000000000000000000000000000000;hpb=bc1177902e6abfae7a65cf68b3172ba96c07ce05;p=voxel-flow

diff --git a/eval_voxelflow.m b/eval_voxelflow.m
new file mode 100644
index 0000000..413f8d7
--- /dev/null
+++ b/eval_voxelflow.m
@@ -0,0 +1,57 @@
+clear;clc;
+
+num_img = length(subdir);
+flag_valid = zeros(1, num_img, 'single');
+mat_psnr = zeros(1, num_img, 'single');
+mat_ssim = zeros(1, num_img, 'single');
+
+for id_img = 1:num_img
+
+	dir_img_cur = [dir_data, subdir(id_img).name, '/'];
+
+	% read images
+	img_pred = imread([dir_img_cur, 'pred_01.png']);
+	img_target = imread([dir_img_cur, 'target_01.png']);
+	img_prev = imread([dir_img_cur, 'frame_00.png']);
+	img_next = imread([dir_img_cur, 'frame_01.png']);
+
+	img_pred_ycbcr = rgb2ycbcr(uint8(img_pred));
+	img_target_ycbcr = rgb2ycbcr(uint8(img_target));
+
+	img_pred_gray = img_pred_ycbcr(:, :, 1);
+	img_target_gray = img_target_ycbcr(:, :, 1);
+
+	img_pred = single(img_pred);
+	img_target = single(img_target);
+	img_prev = single(img_prev);
+
+	img_pred_gray = single(img_pred_gray);x
+	img_target_gray = single(img_target_gray);
+
+	% check validity
+	if sum(mask_flow(:)) > 0
+		
+		flag_valid(id_img) = 1;
+
+		img_pred_mask = repmat(mask_flow, [1, 1, 3]) .* img_pred;
+		img_target_mask = repmat(mask_flow, [1, 1, 3]) .* img_target;
+
+		mse = sum((img_pred_mask(:) - img_target_mask(:)).^2) ./ (3 .* sum(mask_flow(:)));
+		psnr_cur = 20.0 .* log10(255.0) - 10.0 .* log10(mse);
+		
+		[ssim_cur, ~] = ssim(img_pred_gray, img_target_gray);
+
+		mat_psnr(id_img) = psnr_cur;
+		mat_ssim(id_img) = ssim_cur;
+
+	end
+
+	disp(['Processing Img ', num2str(id_img), '...']);
+
+end
+
+flag_valid(find(mat_psnr == inf)) = 0;
+mat_psnr(find(mat_psnr == inf)) = 0;
+
+mean_psnr = sum(flag_valid .* mat_psnr) ./ sum(flag_valid)
+mean_ssim = sum(flag_valid .* mat_ssim) ./ sum(flag_valid)