Architectures
=============

Autoencoder
-----------

.. autofunction:: antspynet.architectures.create_autoencoder_model
.. autofunction:: antspynet.architectures.create_convolutional_autoencoder_model_2d
.. autofunction:: antspynet.architectures.create_convolutional_autoencoder_model_3d


Image classification/regression
-------------------------------

.. autofunction:: antspynet.architectures.create_alexnet_model_2d
.. autofunction:: antspynet.architectures.create_alexnet_model_3d

.. autofunction:: antspynet.architectures.create_densenet_model_2d
.. autofunction:: antspynet.architectures.create_densenet_model_3d

.. autofunction:: antspynet.architectures.create_resnet_model_2d
.. autofunction:: antspynet.architectures.create_resnet_model_3d

.. autofunction:: antspynet.architectures.create_simple_classification_with_spatial_transformer_network_model_2d
.. autofunction:: antspynet.architectures.create_simple_classification_with_spatial_transformer_network_model_3d

Image super-resolution
----------------------

.. autofunction:: antspynet.architectures.create_deep_back_projection_network_model_2d
.. autofunction:: antspynet.architectures.create_deep_back_projection_network_model_3d

.. autofunction:: antspynet.architectures.create_deep_denoise_super_resolution_model_2d
.. autofunction:: antspynet.architectures.create_deep_denoise_super_resolution_model_3d

.. autofunction:: antspynet.architectures.create_denoising_auto_encoder_super_resolution_model_2d
.. autofunction:: antspynet.architectures.create_denoising_auto_encoder_super_resolution_model_3d

.. autofunction:: antspynet.architectures.create_expanded_super_resolution_model_2d
.. autofunction:: antspynet.architectures.create_expanded_super_resolution_model_3d

.. autofunction:: antspynet.architectures.create_image_super_resolution_model_2d
.. autofunction:: antspynet.architectures.create_image_super_resolution_model_3d

.. autofunction:: antspynet.architectures.create_resnet_super_resolution_model_2d
.. autofunction:: antspynet.architectures.create_resnet_super_resolution_model_3d

.. autofunction:: antspynet.architectures.create_vgg_model_2d
.. autofunction:: antspynet.architectures.create_vgg_model_3d

.. autofunction:: antspynet.architectures.create_wide_resnet_model_2d
.. autofunction:: antspynet.architectures.create_wide_resnet_model_3d

Image voxelwise segmentation
----------------------------

.. autofunction:: antspynet.architectures.create_unet_model_2d
.. autofunction:: antspynet.architectures.create_unet_model_3d

.. autofunction:: antspynet.architectures.create_resunet_model_2d
.. autofunction:: antspynet.architectures.create_resunet_model_3d

.. autofunction:: antspynet.architectures.create_denseunet_model_2d
.. autofunction:: antspynet.architectures.create_denseunet_model_3d

.. autofunction:: antspynet.architectures.create_nobrainer_unet_model_3d

.. autofunction:: antspynet.architectures.create_hippmapp3r_unet_model_3d

.. autofunction:: antspynet.architectures.create_sysu_media_unet_model_2d

.. autofunction:: antspynet.create_hypothalamus_unet_model_3d

Custom
------

.. autofunction:: antspynet.architectures.create_simple_fully_convolutional_network_model_3d

Generative adverserial networks
-------------------------------

.. autoclass:: antspynet.architectures.VanillaGanModel
.. autoclass:: antspynet.architectures.DeepConvolutionalGanModel
.. autoclass:: antspynet.architectures.WassersteinGanModel
.. autoclass:: antspynet.architectures.ImprovedWassersteinGanModel
.. autoclass:: antspynet.architectures.CycleGanModel
.. autoclass:: antspynet.architectures.SuperResolutionGanModel