From ad0bc2bb8b0a6f4d947998b1c428c336fa1b8a30 Mon Sep 17 00:00:00 2001 From: raksa Date: Wed, 26 Dec 2018 20:06:34 +0700 Subject: [PATCH] make images' path specific (#671) fixed wrong image's path while debuggin in VSCode --- analysis/compression_analysis/psnr.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/analysis/compression_analysis/psnr.py b/analysis/compression_analysis/psnr.py index abc98fbfe..0f21aac07 100644 --- a/analysis/compression_analysis/psnr.py +++ b/analysis/compression_analysis/psnr.py @@ -4,6 +4,7 @@ """ import math +import os import cv2 import numpy as np @@ -18,18 +19,18 @@ def psnr(original, contrast): def main(): - + dir_path = os.path.dirname(os.path.realpath(__file__)) # Loading images (original image and compressed image) - original = cv2.imread('original_image.png') - contrast = cv2.imread('compressed_image.png', 1) + original = cv2.imread(os.path.join(dir_path, 'original_image.png')) + contrast = cv2.imread(os.path.join(dir_path, 'compressed_image.png'), 1) - original2 = cv2.imread('PSNR-example-base.png') - contrast2 = cv2.imread('PSNR-example-comp-10.jpg', 1) + original2 = cv2.imread(os.path.join(dir_path, 'PSNR-example-base.png')) + contrast2 = cv2.imread(os.path.join(dir_path, 'PSNR-example-comp-10.jpg'), 1) # Value expected: 29.73dB print("-- First Test --") print(f"PSNR value is {psnr(original, contrast)} dB") - + # # Value expected: 31.53dB (Wikipedia Example) print("\n-- Second Test --") print(f"PSNR value is {psnr(original2, contrast2)} dB")