WIP: optional moving_seg/fixed_seg support in VoxelMorph.forward

[Kheil-Z]

Fixes #8540 .

Description

This is a work-in-progress PR to extend VoxelMorph.forward for optional label (moving_seg, fixed_seg, keypoints) support, enabling semi-supervised deformable image registration.

Current State

No functional changes yet.

Next steps :

• Implement warping of segmentation maps using self.warp.
• Update the function signature to include optional moving_seg and fixed_seg.
• Update tests and docstrings accordingly.

Types of changes

• [] Non-breaking change (fix or new feature that would not break existing functionality).
• Breaking change (fix or new feature that would cause existing functionality to change).
• New tests added to cover the changes.
• Integration tests passed locally by running ./runtests.sh -f -u --net --coverage.
• Quick tests passed locally by running ./runtests.sh --quick --unittests --disttests.
• In-line docstrings updated.
• Documentation updated, tested make html command in the docs/ folder.

[coderabbitai]

Important

Review skipped

Ignore keyword(s) in the title.

⛔ Ignored keywords (2)

• WIP
• DO NOT MERGE

Please check the settings in the CodeRabbit UI or the .coderabbit.yaml file in this repository. To trigger a single review, invoke the @coderabbitai review command.

You can disable this status message by setting the reviews.review_status to false in the CodeRabbit configuration file.

✨ Finishing touches

🧪 Generate unit tests (beta)

• Create PR with unit tests
• Post copyable unit tests in a comment

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}