批量psnr

import os
import cv2
import numpy as np
import math

def _psnr(img1, img2):
   mse = np.mean( (img1/255. - img2/255.) ** 2 )
   if mse < 1.0e-10:
      return 100
   PIXEL_MAX = 1
   return 20 * math.log10(PIXEL_MAX / math.sqrt(mse))

cnt=0
HD_path=r"C:\Users\Desktop\test\HD"
LC_path=r"C:\Users\Desktop\test\LC"
HD_files = os.listdir(HD_path)
for HD_file in  HD_files:
    cnt+=1
#print(cnt)

i=0
psnr=np.empty(cnt)
while(i<=cnt):
    img1=cv2.imread(os.path.join(HD_path,"%d.png"%i))
    img2=cv2.imread(os.path.join(LC_path,"%d.png"%i))
    psnr[j] = _psnr(img1,img2)
    i=i+1
    print (psnr[i])
avg = np.sum(psnr)/cnt
print(avg)

save_path = r'C:\Users\Desktop\test\psnr.txt'
with open (save_path,'w') as f:
    for i in range(cnt):
        f.write('%d.png: '%j+str(psnr[i])+'\n')
    f.write('avg: '+str(avg))