Skip to content

Commit e7d7f66

Browse files
Merge branch 'dev' into enable-hf-bundle-hosting
2 parents 9a6a1b6 + 90ead4b commit e7d7f66

File tree

20 files changed

+2144
-55
lines changed

20 files changed

+2144
-55
lines changed

docs/source/apps.rst

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -277,3 +277,15 @@ FastMRIReader
277277

278278
.. autoclass:: monai.apps.nnunet.nnUNetV2Runner
279279
:members:
280+
281+
`nnUNet Bundle`
282+
---------------
283+
.. autoclass:: monai.apps.nnunet.ModelnnUNetWrapper
284+
:members:
285+
:special-members:
286+
287+
.. autofunction:: monai.apps.nnunet.get_nnunet_trainer
288+
.. autofunction:: monai.apps.nnunet.get_nnunet_monai_predictor
289+
.. autofunction:: monai.apps.nnunet.convert_nnunet_to_monai_bundle
290+
.. autofunction:: monai.apps.nnunet.convert_monai_bundle_to_nnunet
291+
.. autofunction:: monai.apps.nnunet.get_network_from_nnunet_plans

docs/source/bundle.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ Model Bundle
3434
:members:
3535
:special-members:
3636

37+
3738
`Scripts`
3839
---------
3940
.. autofunction:: ckpt_export

docs/source/networks.rst

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,16 @@ Blocks
109109
.. autoclass:: SABlock
110110
:members:
111111

112+
`CABlock Block`
113+
~~~~~~~~~~~~~~~
114+
.. autoclass:: CABlock
115+
:members:
116+
117+
`FeedForward Block`
118+
~~~~~~~~~~~~~~~~~~~
119+
.. autoclass:: FeedForward
120+
:members:
121+
112122
`Squeeze-and-Excitation`
113123
~~~~~~~~~~~~~~~~~~~~~~~~
114124
.. autoclass:: ChannelSELayer
@@ -173,6 +183,16 @@ Blocks
173183
.. autoclass:: Subpixelupsample
174184
.. autoclass:: SubpixelUpSample
175185

186+
`Downsampling`
187+
~~~~~~~~~~~~~~
188+
.. autoclass:: DownSample
189+
:members:
190+
.. autoclass:: Downsample
191+
.. autoclass:: SubpixelDownsample
192+
:members:
193+
.. autoclass:: Subpixeldownsample
194+
.. autoclass:: SubpixelDownSample
195+
176196
`Registration Residual Conv Block`
177197
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
178198
.. autoclass:: RegistrationResidualConvBlock
@@ -625,6 +645,11 @@ Nets
625645
.. autoclass:: ViT
626646
:members:
627647

648+
`Restormer`
649+
~~~~~~~~~~~
650+
.. autoclass:: restormer
651+
:members:
652+
628653
`ViTAutoEnc`
629654
~~~~~~~~~~~~
630655
.. autoclass:: ViTAutoEnc

monai/apps/nnunet/__init__.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,5 +11,13 @@
1111

1212
from __future__ import annotations
1313

14+
from .nnunet_bundle import (
15+
ModelnnUNetWrapper,
16+
convert_monai_bundle_to_nnunet,
17+
convert_nnunet_to_monai_bundle,
18+
get_network_from_nnunet_plans,
19+
get_nnunet_monai_predictor,
20+
get_nnunet_trainer,
21+
)
1422
from .nnunetv2_runner import nnUNetV2Runner
1523
from .utils import NNUNETMode, analyze_data, create_new_data_copy, create_new_dataset_json

0 commit comments

Comments
 (0)