psnr = batch_PSNR(Out, ISource, 1.) print('去噪噪PSNR', psnr)怎么结合到一起
时间: 2024-01-16 16:02:52 浏览: 76
psnr.rar_PSNR_信噪比_数字水印psnr
你可以这样将它们结合起来:
```python
import torch
# 假设 Out 和 ISource 都是 torch.Tensor 类型
mse = torch.mean(torch.pow(Out - ISource, 2))
psnr = 10 * torch.log10(1.0 / mse)
print('去噪噪PSNR', psnr)
```
这里使用了 PyTorch 内置的函数计算MSE(均方误差),然后用它计算PSNR。最后打印出PSNR的值。你可以将这段代码添加到你的程序中,用于计算PSNR。
阅读全文