mindspore2022/docs/api/api_python/nn/mindspore.nn.RootMeanSquareDistance.rst

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不同。