Skip to content

Commit fbbb64b

Browse files
authored
Implemented VoxelMorph (#7178)
Fixes #5484. ### Description Implemented VoxelMorph and added some docstrings. Checked coding style locally. ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] 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). - [x] 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`. - [x] In-line docstrings updated. - [x] Documentation updated, tested `make html` command in the `docs/` folder. --------- Signed-off-by: kaibo <ktang@unc.edu>
1 parent 4847df2 commit fbbb64b

File tree

4 files changed

+775
-0
lines changed

4 files changed

+775
-0
lines changed

docs/source/networks.rst

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -710,6 +710,18 @@ Nets
710710
.. autoclass:: Quicknat
711711
:members:
712712

713+
`VoxelMorph`
714+
~~~~~~~~~~~~
715+
.. autoclass:: VoxelMorphUNet
716+
:members:
717+
718+
.. autoclass:: voxelmorphunet
719+
:members:
720+
721+
.. autoclass:: VoxelMorph
722+
:members:
723+
724+
.. autoclass:: voxelmorph
713725

714726
Utilities
715727
---------

monai/networks/nets/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,3 +111,4 @@
111111
from .vit import ViT
112112
from .vitautoenc import ViTAutoEnc
113113
from .vnet import VNet
114+
from .voxelmorph import VoxelMorph, VoxelMorphUNet

0 commit comments

Comments
 (0)