Skip to content

Commit 43bc023

Browse files
authored
6676 port generative networks controlnet (#7312)
Part of #6676 . ### Description Ports the ControlNet. ### 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: Mark Graham <[email protected]>
1 parent de0a476 commit 43bc023

File tree

4 files changed

+604
-0
lines changed

4 files changed

+604
-0
lines changed

docs/source/networks.rst

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -588,6 +588,11 @@ Nets
588588
.. autoclass:: DiffusionModelUNet
589589
:members:
590590

591+
`ControlNet`
592+
~~~~~~~~~~~~
593+
.. autoclass:: ControlNet
594+
:members:
595+
591596
`RegUNet`
592597
~~~~~~~~~
593598
.. autoclass:: RegUNet

monai/networks/nets/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
from .basic_unet import BasicUNet, BasicUnet, Basicunet, basicunet
1919
from .basic_unetplusplus import BasicUNetPlusPlus, BasicUnetPlusPlus, BasicunetPlusPlus, basicunetplusplus
2020
from .classifier import Classifier, Critic, Discriminator
21+
from .controlnet import ControlNet
2122
from .daf3d import DAF3D
2223
from .densenet import (
2324
DenseNet,

0 commit comments

Comments
 (0)