|
- mindspore.nn.RootMeanSquareDistance
- ======================================
-
- .. py:class:: mindspore.nn.RootMeanSquareDistance(symmetric=False, distance_metric='euclidean')
-
- 计算从 `y_pred` 到 `y` 的均方根表面距离。
-
- 给定两个集合A和B,S(A)表示A的表面像素,任意v到S(A)的最短距离定义为:
-
- .. math::
- {\text{dis}}\left (v, S(A)\right ) = \underset{s_{A} \in S(A)}{\text{min }}\rVert v - s_{A} \rVert
-
- 从集合B到集合A的均方根表面距离(Root Mean Square Surface Distance)为:
-
- .. math::
- RmsSurDis(B \rightarrow A) = \sqrt{\frac{\sum_{s_{B} \in S(B)}^{} {\text{dis}^2 \left ( s_{B}, S(A)
- \right )} }{\left | S(B) \right |}}
-
- 其中 \|\|\*\|\| 表示距离度量。 \|\*\| 表示元素的数量。
-
- 从集合B到集合A以及从集合A到集合B的表面距离平均值为:
-
- .. math::
- RmsSurDis(A \leftrightarrow B) = \sqrt{\frac{\sum_{s_{A} \in S(A)}^{} {\text{dis} \left ( s_{A},
- S(B) \right ) ^{2}} + \sum_{s_{B} \in S(B)}^{} {\text{dis} \left ( s_{B}, S(A) \right ) ^{2}}}{\left | S(A)
- \right | + \left | S(B) \right |}}
-
- **参数:**
-
- - **distance_metric** (string) - 支持如下三种距离计算方法:"euclidean"、"chessboard" 或 "taxicab"。默认值:"euclidean"。
- - **symmetric** (bool) - 是否计算 `y_pred` 和 `y` 之间的对称平均平面距离。如果为False,计算方式为 :math:`RmsSurDis(y_pred, y)`, 如果为True,计算方式为 :math:`RmsSurDis(y_pred \leftrightarrow y)`。默认值:False。
-
- .. py:method:: clear()
-
- 内部评估结果清零。
-
- .. py:method:: eval()
-
- 计算均方根表面距离。
-
- **返回:**
-
- numpy.float64,计算得到的均方根表面距离值。
-
- **异常:**
-
- - **RuntimeError** - 如果没有先调用update方法,则会报错。
-
-
-
- .. py:method:: update(*inputs)
-
- 使用 `y_pred`、`y` 和 `label_idx` 更新内部评估结果。
-
- **参数:**
-
- - **inputs** - `y_pred`、`y` 和 `label_idx`。`y_pred` 和 `y` 为Tensor,list或numpy.ndarray,`y_pred` 是预测的二值图像。`y` 是实际的二值图像。`label_idx` 数据类型为int或float,表示像素点的类别值。
-
- **异常:**
-
- - **ValueError** - 输入的数量不等于3。
- - **TypeError** - `label_idx` 的数据类型不是int或float。
- - **ValueError** - `label_idx` 的值不在y_pred或y中。
- - **ValueError** - `y_pred` 和 `y` 的shape不同。
|