Skip to content

Commit 3bd76e7

Browse files
committed
fix:CodeRabbit issues
Signed-off-by: Rusheel Sharma <rusheelhere@gmail.com>
1 parent 1114907 commit 3bd76e7

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

monai/metrics/utils.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -361,9 +361,15 @@ def get_edge_surface_distance(
361361
if use_subvoxels:
362362
edges_spacing = spacing if spacing is not None else ([1] * len(y_pred.shape))
363363

364-
edge_results = get_mask_edges(y_pred, y, crop=True, spacing=edges_spacing, always_return_as_numpy=False)
364+
edge_results = get_mask_edges(y_pred, y, crop=(mask is None), spacing=edges_spacing, always_return_as_numpy=False)
365365
edges_pred, edges_gt = edge_results[0], edge_results[1]
366366

367+
# Apply mask to edges if provided
368+
if mask is not None:
369+
mask = torch.as_tensor(mask, device=edges_pred.device, dtype=torch.bool)
370+
edges_pred = edges_pred & mask
371+
edges_gt = edges_gt & mask
372+
367373
distances_raw: tuple[torch.Tensor, torch.Tensor] | tuple[torch.Tensor]
368374
if symmetric:
369375
distances_raw = (

0 commit comments

Comments
 (0)