Skip to content

Commit d22e4fc

Browse files
committed
WIP: placeholder for optional moving_seg/fixed_seg support in VoxelMorph.forward
1 parent a8a7e57 commit d22e4fc

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

monai/networks/nets/voxelmorph.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -441,6 +441,7 @@ def __init__(
441441
self.warp = Warp(mode="bilinear", padding_mode="zeros")
442442

443443
def forward(self, moving: torch.Tensor, fixed: torch.Tensor) -> tuple[torch.Tensor, torch.Tensor]:
444+
# TODO: add optional moving_seg, fixed_seg arguments and handle warping of segmentation maps
444445
if moving.shape != fixed.shape:
445446
raise ValueError(
446447
"The spatial shape of the moving image should be the same as the spatial shape of the fixed image."

0 commit comments

Comments
 (0)