File tree 1 file changed +7
-6
lines changed
analysis/compression_analysis
1 file changed +7
-6
lines changed Original file line number Diff line number Diff line change 4
4
"""
5
5
6
6
import math
7
+ import os
7
8
8
9
import cv2
9
10
import numpy as np
@@ -18,18 +19,18 @@ def psnr(original, contrast):
18
19
19
20
20
21
def main ():
21
-
22
+ dir_path = os . path . dirname ( os . path . realpath ( __file__ ))
22
23
# 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 )
25
26
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 )
28
29
29
30
# Value expected: 29.73dB
30
31
print ("-- First Test --" )
31
32
print (f"PSNR value is { psnr (original , contrast )} dB" )
32
-
33
+
33
34
# # Value expected: 31.53dB (Wikipedia Example)
34
35
print ("\n -- Second Test --" )
35
36
print (f"PSNR value is { psnr (original2 , contrast2 )} dB" )
You can’t perform that action at this time.
0 commit comments