Skip to content

Commit fac754d

Browse files
marksgrahamericspodKumoLiu
authored
6676 port generative networks autoencoderkl (#7260)
Partially fixes #6676 ### Description Implements the AutoencoderKL network from MONAI Generative. NB this network is subject to a planned refactor once the porting is complete, [see here](#7227). ### 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. - [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: Mark Graham <[email protected]> Co-authored-by: Eric Kerfoot <[email protected]> Co-authored-by: YunLiu <[email protected]>
1 parent 962aa3c commit fac754d

File tree

4 files changed

+1089
-0
lines changed

4 files changed

+1089
-0
lines changed

docs/source/networks.rst

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -595,6 +595,11 @@ Nets
595595
.. autoclass:: AutoEncoder
596596
:members:
597597

598+
`AutoEncoderKL`
599+
~~~~~~~~~~~~~~~
600+
.. autoclass:: AutoencoderKL
601+
:members:
602+
598603
`VarAutoEncoder`
599604
~~~~~~~~~~~~~~~~
600605
.. autoclass:: VarAutoEncoder

monai/networks/nets/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
from .ahnet import AHnet, Ahnet, AHNet
1515
from .attentionunet import AttentionUnet
1616
from .autoencoder import AutoEncoder
17+
from .autoencoderkl import AutoencoderKL
1718
from .basic_unet import BasicUNet, BasicUnet, Basicunet, basicunet
1819
from .basic_unetplusplus import BasicUNetPlusPlus, BasicUnetPlusPlus, BasicunetPlusPlus, basicunetplusplus
1920
from .classifier import Classifier, Critic, Discriminator

0 commit comments

Comments
 (0)