'''MSE''' squaredError = []absError = [] for val in error: squaredError.append(val * val) # target-prediction之差平方 a = sum(squaredError) / len(squaredError) print("MSE = ", a) # 均方误差MSE '''SNR''' snfang = [] for s in target: snfang.append(s*s)ss = math.log10(sum(snfang)/sum(squaredError)) print("SNR=",10*ss) """PSNR""" p = math.log10((2**len(squaredError)-1)**2/a) print("PSNR=",10*p)
评论