MRI Segmentation Neural Networks
--------------------------------

.. autoclass:: atommic.collections.segmentation.nn.base.BaseMRISegmentationModel
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.segmentationnet.BaseSegmentationNet
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.attentionunet.SegmentationAttentionUNet
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.attentionunet_base.attentionunet_block.AttentionGate
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.attentionunet_base.attentionunet_block.AttentionUnet
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.dynunet.SegmentationDYNUNet
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.dynunet_base.dynunet_block.DynUNetSkipLayer
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.dynunet_base.dynunet_block.DynUNet
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.lambdaunet.SegmentationLambdaUNet
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.lambdaunet_base.lambdaunet_block.LambdaLayer
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.lambdaunet_base.lambdaunet_block.LambdaBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.lambdaunet_base.lambdaunet_block.LambdaUNet
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unet.SegmentationUNet
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unet3d.Segmentation3DUNet
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unet3d_base.unet3d_block.Conv3dBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unet3d_base.unet3d_block.TransposeConv3dBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unet3d_base.unet3d_block.UNet3D
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unetr.SegmentationUNetR
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unetr_base.unetr_block.UnetOutBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unetr_base.unetr_block.UnetrBasicBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unetr_base.unetr_block.UnetrPrUpBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unetr_base.unetr_block.UnetrUpBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unetr_base.unetr_block.UnetResBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unetr_base.unetr_block.UnetUpBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unetr_base.unetr_block.UnetBasicBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.unetr_base.unetr_block.UNETR
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.vit_block.ViT
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.patchembedding.PatchEmbeddingBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.patchembedding.PatchEmbed
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.transformer_block.MLPBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.transformer_block.SABlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.transformer_block.TransformerBlock
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.utils.Convolution
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.utils.stride_minus_kernel_padding
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.utils.get_padding
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.utils.get_output_padding
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.utils.same_padding
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.utils.get_conv_layer
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.utils._no_grad_trunc_normal_
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vit_base.utils.trunc_normal_
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vnet.SegmentationVNet
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vnet_base.vnet_block.LUConv
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vnet_base.vnet_block._make_nconv
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vnet_base.vnet_block.InputTransition
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vnet_base.vnet_block.DownTransition
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vnet_base.vnet_block.UpTransition
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vnet_base.vnet_block.OutputTransition
    :special-members: __init__


.. autoclass:: atommic.collections.segmentation.nn.vnet_base.vnet_block.VNet
    :special-members: __init__