python 求图像SNR代码

今天花了一上午写了图像SNR的Python代码 （发现csdn全是psnr的
保证正确 有用matlab内置函数测试过
场景：有真值图像
单位：分贝

import os
import numpy as np
import math
def cal_snr(noise_img,clean_img):
    noise_signal=noise_img-clean_img
    clean_signal=clean_img
    noise_signal_2=noise_signal**2
    clean_signal_2=clean_signal**2
    sum1=np.sum(clean_signal_2)
    sum2=np.sum(noise_signal_2)
    snrr=20*math.log10(math.sqrt(sum1)/math.sqrt(sum2))
    return snrr

接口部分

####################### calculate and store snr for test image################################
    clean_im=clean_im.astype(np.float32)#float32改成别的数据格式算出来的值是有较明显区别的
    denoise_im=tiff.imread(save_name)
    denoise_im=denoise_im.astype(np.float32)
    snrr=cal_snr(denoise_im[100-pre_post_frame:400-pre_post_frame,11:500,11:500],clean_im[100:400,:,:])
    print(snrr)
    with open(path_output+ '//'+'result_' + run_uid + '.txt', 'a', encoding='utf-8') as f:
         f.write(str(snrr) + '\n')