Skip to content

Commit

Permalink
feat(Architecture): added ConvNeXt to architecture library
Browse files Browse the repository at this point in the history
  • Loading branch information
muellerdo committed Nov 29, 2022
1 parent 82c3e3d commit 5d94fbe
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions aucmedi/neural_network/architectures/image/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,11 @@
from aucmedi.neural_network.architectures.image.vit_b32 import ViT_B32
from aucmedi.neural_network.architectures.image.vit_l16 import ViT_L16
from aucmedi.neural_network.architectures.image.vit_l32 import ViT_L32
# ConvNeXt
from aucmedi.neural_network.architectures.image.convnext_base import ConvNeXtBase
from aucmedi.neural_network.architectures.image.convnext_tiny import ConvNeXtTiny
from aucmedi.neural_network.architectures.image.convnext_small import ConvNeXtSmall
from aucmedi.neural_network.architectures.image.convnext_large import ConvNeXtLarge

#-----------------------------------------------------#
# Access Functions to Architecture Classes #
Expand Down Expand Up @@ -107,6 +112,10 @@
"ViT_B32": ViT_B32,
"ViT_L16": ViT_L16,
"ViT_L32": ViT_L32,
"ConvNeXtBase": ConvNeXtBase,
"ConvNeXtTiny": ConvNeXtTiny,
"ConvNeXtSmall": ConvNeXtSmall,
"ConvNeXtLarge": ConvNeXtLarge,
}
""" Dictionary of implemented 2D Architectures Methods in AUCMEDI.
Expand Down Expand Up @@ -192,6 +201,10 @@
"ViT_B32": "tf",
"ViT_L16": "tf",
"ViT_L32": "tf",
"ConvNeXtBase": None,
"ConvNeXtTiny": None,
"ConvNeXtSmall": None,
"ConvNeXtLarge": None,
}
""" Dictionary of recommended [Standardize][aucmedi.data_processing.subfunctions.standardize] techniques for 2D Architectures Methods in AUCMEDI.
Expand Down

0 comments on commit 5d94fbe

Please sign in to comment.