convolutional

Collection of Convolutional Models constructors.

Interfaces

interfaces.Conv2DInterface Primitive Interface to be used by all ashpy.models.

Decoders

decoders.Decoder Primitive Model for all decoder (i.e., transpose convolution) based architecture.
decoders.FCNNDecoder Fully Convolutional Decoder.

Encoders

encoders.Encoder Primitive Model for all encoder (i.e., convolution) based architecture.
encoders.FCNNEncoder Fully Convolutional Encoder.

Autoencoders

autoencoders.Autoencoder Primitive Model for all convolutional autoencoders.
autoencoders.FCNNAutoencoder Primitive Model for all fully convolutional autoencoders.

UNet

unet.UNet UNet Architecture.
unet.SUNet Semantic UNet.
unet.FUNet Functional UNET Implementation.

Discriminators

discriminators.MultiScaleDiscriminator Multi-Scale discriminator.
discriminators.PatchDiscriminator Pix2Pix discriminator.

Pix2PixHD

pix2pixhd.LocalEnhancer Local Enhancer module of the Pix2PixHD architecture.
pix2pixhd.GlobalGenerator Global Generator from pix2pixHD paper.

Modules

autoencoders Collection of Fully Convolutional Autoencoders.
discriminators Convolutional Discriminators.
decoders Collection of Decoders (i.e., GANs’ Generators) models.
encoders Collection of Encoders (i.e., GANs’ Discriminators) models.
interfaces Primitive Convolutional interfaces.
unet UNET implementations.
pix2pixhd Pix2Pix HD Implementation.