forked from mindspore-Ecosystem/mindspore
31 lines
1.2 KiB
ReStructuredText
31 lines
1.2 KiB
ReStructuredText
|
mindspore.nn.PSNR
|
|||
|
|
==================
|
|||
|
|
|
|||
|
|
.. py:class:: mindspore.nn.PSNR(max_val=1.0)
|
|||
|
|
|
|||
|
|
在批处理中计算两个图像的峰值信噪比（PSNR）。
|
|||
|
|
|
|||
|
|
它为批处理中的每个图像生成PSNR值。假设输入为 :math:`I` 和 :math:`K` ，且shape都为 :math:`h*w` 。 :math:`MAX` 表示像素值的动态范围。
|
|||
|
|
|
|||
|
|
.. math::
|
|||
|
|
MSE&=\frac{1}{hw}\sum\limits_{i=0}^{h-1}\sum\limits_{j=0}^{w-1}[I(i,j)-K(i,j)]^2\\
|
|||
|
|
PSNR&=10*log_{10}(\frac{MAX^2}{MSE})
|
|||
|
|
|
|||
|
|
**参数：**
|
|||
|
|
|
|||
|
|
- **max_val** (Union[int, float]) - 像素的动态范围（8位灰度图像为255）。该值必须大于0。默认值：1.0。
|
|||
|
|
|
|||
|
|
**输入：**
|
|||
|
|
|
|||
|
|
- **img1** (Tensor) - 格式为'NCHW'的第一批图像。shape和数据类型必须与 `img2` 相同。
|
|||
|
|
- **img2** (Tensor) - 格式为'NCHW'的第二批图像。shape和数据类型必须与 `img1` 相同。
|
|||
|
|
|
|||
|
|
**输出：**
|
|||
|
|
|
|||
|
|
Tensor，使用数据类型mindspore.float32。shape为N的一维Tensor，其中N是 `img1` 的批次大小。
|
|||
|
|
|
|||
|
|
**异常：**
|
|||
|
|
|
|||
|
|
- **TypeError** - `max_val` 既不是int也不是float。
|
|||
|
|
- **ValueError** - `max_val` 小于或等于0。
|
|||
|
|
- **ValueError** - `img1` 或 `img2` 的shape长度不等于4。
|