o  i%@sddlmZddlmZddlmZddlZddlZddl m Z m Z m Z m Z ddlmZmZddlmZGd d d eZ ddddZdS)) annotations)Sequence)AnyN)do_metric_reductionget_edge_surface_distanceignore_backgroundprepare_spacing)MetricReductionconvert_data_type)CumulativeIterationMetriccsHeZdZdZdddejdfdfdd ZdddZ ddddZZ S) SurfaceDistanceMetrica Compute Surface Distance between two tensors. It can support both multi-classes and multi-labels tasks. It supports both symmetric and asymmetric surface distance calculation. Input `y_pred` is compared with ground truth `y`. `y_preds` is expected to have binarized predictions and `y` should be in one-hot format. You can use suitable transforms in ``monai.transforms.post`` first to achieve binarized values. `y_preds` and `y` can be a list of channel-first Tensor (CHW[D]) or a batch-first Tensor (BCHW[D]). Example of the typical execution steps of this metric class follows :py:class:`monai.metrics.metric.Cumulative`. Args: include_background: whether to include distance computation on the first channel of the predicted output. Defaults to ``False``. symmetric: whether to calculate the symmetric average surface distance between `seg_pred` and `seg_gt`. Defaults to ``False``. distance_metric: : [``"euclidean"``, ``"chessboard"``, ``"taxicab"``] the metric used to compute surface distance. Defaults to ``"euclidean"``. reduction: define mode of reduction to the metrics, will only apply reduction on `not-nan` values, available reduction modes: {``"none"``, ``"mean"``, ``"sum"``, ``"mean_batch"``, ``"sum_batch"``, ``"mean_channel"``, ``"sum_channel"``}, default to ``"mean"``. if "none", will not do reduction. get_not_nans: whether to return the `not_nans` count, if True, aggregate() returns (metric, not_nans). Here `not_nans` count the number of not nans for the metric, thus its shape equals to the shape of the metric. F euclideaninclude_backgroundbool symmetricdistance_metricstr reductionMetricReduction | str get_not_nansreturnNonecs,t||_||_||_||_||_dSN)super__init__rrrrr)selfrrrrr __class__`/home/dell461/cl/sdc2/last_ska_mid/HISourceFinder-master-l/src/monai/metrics/surface_distance.pyr4s  zSurfaceDistanceMetric.__init__y_pred torch.Tensorykwargsrc Ks4|dkr tdt|||j|j|j|ddS)a Args: y_pred: input data to compute, typical segmentation model output. It must be one-hot format and first dim is batch, example shape: [16, 3, 32, 32]. The values should be binarized. y: ground truth to compute the distance. It must be one-hot format and first dim is batch. The values should be binarized. kwargs: additional parameters, e.g. ``spacing`` should be passed to correctly compute the metric. ``spacing``: spacing of pixel (or voxel). This parameter is relevant only if ``distance_metric`` is set to ``"euclidean"``. If a single number, isotropic spacing with that value is used for all images in the batch. If a sequence of numbers, the length of the sequence must be equal to the image dimensions. This spacing will be used for all images in the batch. If a sequence of sequences, the length of the outer sequence must be equal to the batch size. If inner sequence has length 1, isotropic spacing with that value is used for all images in the batch, else the inner sequence length must be equal to the image dimensions. If ``None``, spacing of unity is used for all images in batch. Defaults to ``None``. Raises: ValueError: when `y_pred` has less than three dimensions. z-y_pred should have at least three dimensions.spacing)r!r#rrrr&)dim ValueError compute_average_surface_distancerrrget)rr!r#r$rrr _compute_tensorCs z%SurfaceDistanceMetric._compute_tensorNMetricReduction | str | None0torch.Tensor | tuple[torch.Tensor, torch.Tensor]cCsB|}t|tjstdt||p|j\}}|jr||fS|S)a Execute reduction logic for the output of `compute_average_surface_distance`. Args: reduction: define mode of reduction to the metrics, will only apply reduction on `not-nan` values, available reduction modes: {``"none"``, ``"mean"``, ``"sum"``, ``"mean_batch"``, ``"sum_batch"``, ``"mean_channel"``, ``"sum_channel"``}, default to `self.reduction`. if "none", will not do reduction. z-the data to aggregate must be PyTorch Tensor.) get_buffer isinstancetorchTensorr(rrr)rrdatafnot_nansrrr aggregatefs  zSurfaceDistanceMetric.aggregate) rrrrrrrrrrrr)r!r"r#r"r$rrr"r)rr,rr-) __name__ __module__ __qualname____doc__r MEANrr+r5 __classcell__rrrr r s $r Frr!np.ndarray | torch.Tensorr#rrrrrr&\int | float | np.ndarray | Sequence[int | float | np.ndarray | Sequence[int | float]] | Nonerr"c Cs6|s t||d\}}t|tjtjdd}t|tjtjdd}|j|jkr3td|jd|jd|jdd\}}tj||ftj|j d }|j d} t ||| d } t ||D]3\} } t|| | f|| | f|| | || d \} }} t|}|jd krtt jn||| | f<qZt|tj|j tjd dS)a This function is used to compute the Average Surface Distance from `y_pred` to `y` under the default setting. In addition, if sets ``symmetric = True``, the average symmetric surface distance between these two inputs will be returned. The implementation refers to `DeepMind's implementation `_. Args: y_pred: input data to compute, typical segmentation model output. It must be one-hot format and first dim is batch, example shape: [16, 3, 32, 32]. The values should be binarized. y: ground truth to compute mean the distance. It must be one-hot format and first dim is batch. The values should be binarized. include_background: whether to include distance computation on the first channel of the predicted output. Defaults to ``False``. symmetric: whether to calculate the symmetric average surface distance between `seg_pred` and `seg_gt`. Defaults to ``False``. distance_metric: : [``"euclidean"``, ``"chessboard"``, ``"taxicab"``] the metric used to compute surface distance. Defaults to ``"euclidean"``. spacing: spacing of pixel (or voxel). This parameter is relevant only if ``distance_metric`` is set to ``"euclidean"``. If a single number, isotropic spacing with that value is used for all images in the batch. If a sequence of numbers, the length of the sequence must be equal to the image dimensions. This spacing will be used for all images in the batch. If a sequence of sequences, the length of the outer sequence must be equal to the batch size. If inner sequence has length 1, isotropic spacing with that value is used for all images in the batch, else the inner sequence length must be equal to the image dimensions. If ``None``, spacing of unity is used for all images in batch. Defaults to ``None``. )r!r#) output_typedtyperz*y_pred and y should have same shapes, got z and .N)r?device)r& batch_sizeimg_dim)rr&r class_index)r)r>rBr?)rr r0r1floatshaper(emptyfloat32rBndimrnpndindexrcattensornanmean)r!r#rrrr&rCn_classZasdrD spacing_listbc_ distancessurface_distancerrr r){s,$     (r))FFrN)r!r<r#r<rrrrrrr&r=rr") __future__rcollections.abcrtypingrnumpyrKr0monai.metrics.utilsrrrr monai.utilsr r metricr r r)rrrr s   d