AutoEncoder_Transformer

class PatchEmbedding

Convert image into patches and embed them

Methods:

class TransformerEncoder

Transformer Encoder for image patches

Methods:

class TransformerDecoder

Decode embedded features back to image

class TransformerAutoencoder

Complete Transformer-based Autoencoder

Methods: