仅为学习交流,侵删!转载请注明出处!
import os
import numpy as np
from glob import glob
import cv2
from skimage.measure import compare_mse,compare_ssim,compare_psnr
def read_img(path):
return cv2.imread(path,cv2.IMREAD_GRAYSCALE)
def mse(tf_img1, tf_img2):
return compare_mse(tf_img1,tf_img2)
def psnr(tf_img1, tf_img2):
return compare_psnr(tf_img1,tf_img2)
def ssim(tf_img1, tf_img2):
return compare_ssim(tf_img1,tf_img2)
def main(