PSNR 计算函数

下面的 mse2psnr 是 MSE 转换位 PSNR 的代码，注意此时像素需要归一化到 0-1 的范围内。

Cal_PSNR 函数第三个参数可以设置 像素强度的最大值是 255 还是1。默认是 1。

import torch
import numpy as np
import cv2
import math

mse2psnr = lambda x: -10. * torch.log(x) / torch.log(torch.Tensor