Skip to content

Commit 90ead4b

Browse files
SimoneBendazzoli93pre-commit-ci[bot]ericspod
authored
8328 nnunet bundle integration (#8329)
Fixes #8328 ### Description ### 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). - [x] Breaking change (fix or new feature that would cause existing functionality to change). - [X] New tests added to cover the changes. - [x] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`. - [x] 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: simben <[email protected]> Signed-off-by: Simone Bendazzoli <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Eric Kerfoot <[email protected]>
1 parent 4986d7f commit 90ead4b

File tree

6 files changed

+766
-0
lines changed

6 files changed

+766
-0
lines changed

docs/source/apps.rst

+12
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

+1
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

monai/apps/nnunet/__init__.py

+8
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)