We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent af83422 commit 187da14Copy full SHA for 187da14
1 file changed
monai/metrics/utils.py
@@ -359,7 +359,7 @@ def get_edge_surface_distance(
359
if len(edge_results) > 2 and isinstance(edge_results[2], tuple):
360
slices = edge_results[2]
361
mask = mask[slices]
362
- mask = mask.to(edges_pred.device).bool()
+ mask = torch.as_tensor(mask, device=edges_pred.device, dtype=torch.bool)
363
edges_pred = edges_pred & mask
364
edges_gt = edges_gt & mask
365
0 commit comments