mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2024-11-23 21:11:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

make images' path specific (#671)

Browse Source 
fixed wrong image's path while debuggin in VSCode
This commit is contained in:
raksa 2018-12-26 20:06:34 +07:00 committed by Libin Yang
parent f6d241e52d
commit ad0bc2bb8b
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
analysis/compression_analysis/psnr.py
View File

@ -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")