Skip to content

Commit ad0bc2b

Browse files
raksayanglbme
authored andcommitted
make images' path specific (TheAlgorithms#671)
fixed wrong image's path while debuggin in VSCode
1 parent f6d241e commit ad0bc2b

File tree

1 file changed

+7
-6
lines changed
  • analysis/compression_analysis

1 file changed

+7
-6
lines changed

analysis/compression_analysis/psnr.py

+7-6
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
"""
55

66
import math
7+
import os
78

89
import cv2
910
import numpy as np
@@ -18,18 +19,18 @@ def psnr(original, contrast):
1819

1920

2021
def main():
21-
22+
dir_path = os.path.dirname(os.path.realpath(__file__))
2223
# Loading images (original image and compressed image)
23-
original = cv2.imread('original_image.png')
24-
contrast = cv2.imread('compressed_image.png', 1)
24+
original = cv2.imread(os.path.join(dir_path, 'original_image.png'))
25+
contrast = cv2.imread(os.path.join(dir_path, 'compressed_image.png'), 1)
2526

26-
original2 = cv2.imread('PSNR-example-base.png')
27-
contrast2 = cv2.imread('PSNR-example-comp-10.jpg', 1)
27+
original2 = cv2.imread(os.path.join(dir_path, 'PSNR-example-base.png'))
28+
contrast2 = cv2.imread(os.path.join(dir_path, 'PSNR-example-comp-10.jpg'), 1)
2829

2930
# Value expected: 29.73dB
3031
print("-- First Test --")
3132
print(f"PSNR value is {psnr(original, contrast)} dB")
32-
33+
3334
# # Value expected: 31.53dB (Wikipedia Example)
3435
print("\n-- Second Test --")
3536
print(f"PSNR value is {psnr(original2, contrast2)} dB")

0 commit comments

Comments
 (0)