Echo ViT-VQGAN

Echocardiogram quality enhancement using ViT-VQGANs, introduced in the paper Vector-quantized Image Modeling with Improved VQGAN. Heavily inspired by the repositories Qiyuan-Ge/PaintMind and CompVis/taming-transformers.

Notes

• To run inference on a pretrained model, cd into the scripts directory and run inference.py.
• To train a model from scratch, cd into the scripts directory and run train.py.