Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[WIP]Ports generative networks #7196

Closed
wants to merge 12 commits into from
Closed

[WIP]Ports generative networks #7196

Show file tree
Hide file tree
Changes from 4 commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
994d0e7
Initial commit of diffusion unet
marksgraham Nov 3, 2023
cdd8b7f
Adds autoencoderKL (including torchscript errors, for now)
marksgraham Nov 8, 2023
844f7bd
Removes torchscript test
marksgraham Nov 9, 2023
4eba0e7
Removes torchscript tests
marksgraham Nov 9, 2023
d5d247b
Adds VQ-VAE
marksgraham Nov 13, 2023
b2a19d1
Makes blocks private and adds warning
marksgraham Nov 13, 2023
4237233
Adds transformer
marksgraham Nov 14, 2023
103e6e0
Adds controlnet
marksgraham Nov 14, 2023
e9c483a
Adds patchgan discriminator
marksgraham Nov 14, 2023
b9ff39c
Adds spade network
marksgraham Nov 20, 2023
2cac4a6
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Nov 20, 2023
8602212
Makes vector quantizier compatible with mypy
marksgraham Nov 22, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions monai/networks/nets/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
from .ahnet import AHnet, Ahnet, AHNet
from .attentionunet import AttentionUnet
from .autoencoder import AutoEncoder
from .autoencoderkl import AutoencoderKL
from .basic_unet import BasicUNet, BasicUnet, Basicunet, basicunet
from .basic_unetplusplus import BasicUNetPlusPlus, BasicUnetPlusPlus, BasicunetPlusPlus, basicunetplusplus
from .classifier import Classifier, Critic, Discriminator
Expand All @@ -34,6 +35,7 @@
densenet201,
densenet264,
)
from .diffusion_model_unet import DiffusionModelUNet
from .dints import DiNTS, TopologyConstruction, TopologyInstance, TopologySearch
from .dynunet import DynUNet, DynUnet, Dynunet
from .efficientnet import (
Expand Down
Loading
Loading