mindspore.nn.MeanSurfaceDistance

class mindspore.nn.MeanSurfaceDistance(symmetric=False, distance_metric='euclidean')[源代码]

计算从 y_predy 的平均表面距离。通常情况下,用来衡量分割任务中,预测情况和真实情况之间的差异度。

给定两个集合A和B,S(A)表示A的表面像素,任意v到S(A)的最短距离定义为:

\[{\text{dis}}\left (v, S(A)\right ) = \underset{s_{A} \in S(A)}{\text{min }}\rVert v - s_{A} \rVert\]

从集合B到集合A的平均表面距离(Average Surface Distance)为:

\[AvgSurDis(B \rightarrow A) = \frac{\sum_{s_{B} \in S(B)}^{} {\text{dis} \left ( s_{B}, S(A) \right )} } {\left | S(B) \right |}\]

其中 ||*|| 表示距离度量。 |*| 表示元素的数量。

从集合B到集合A以及从集合A到集合B的表面距离平均值为:

\[MeanSurDis(A \leftrightarrow B) = \frac{\sum_{s_{A} \in S(A)}^{} {\text{dis} \left ( s_{A}, S(B) \right )} + \sum_{s_{B} \in S(B)}^{} {\text{dis} \left ( s_{B}, S(A) \right )} }{\left | S(A) \right | + \left | S(B) \right |}\]

参数:

  • distance_metric (string) - 支持如下三种距离计算方法:”euclidean”、”chessboard”或”taxicab”。默认值:”euclidean”。

  • symmetric (bool) - 是否计算 y_predy 之间的对称平均平面距离。如果为False,计算方式为 \(AvgSurDis(y_{pred} \rightarrow y)\) , 如果为True,计算方式为 \(MeanSurDis(y_{pred} \leftrightarrow y)\) 。默认值:False。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import nn, Tensor
>>> x = Tensor(np.array([[3, 0, 1], [1, 3, 0], [1, 0, 2]]))
>>> y = Tensor(np.array([[0, 2, 1], [1, 2, 1], [0, 0, 1]]))
>>> metric = nn.MeanSurfaceDistance(symmetric=False, distance_metric="euclidean")
>>> metric.clear()
>>> metric.update(x, y, 0)
>>> mean_average_distance = metric.eval()
>>> print(mean_average_distance)
0.8047378541243649
clear()[源代码]

内部评估结果清零。

eval()[源代码]

计算平均表面距离。

返回:

numpy.float64,计算得到的平均表面距离值。

异常:

  • RuntimeError - 如果没有先调用update方法。

update(*inputs)[源代码]

使用 y_predylabel_idx 更新内部评估结果。

参数:

  • inputs - y_predylabel_idxy_predy 为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_predy 的shape不同。