alif-munim/echo-vqgan

ViT-VQGAN for echocardiogram quality enhancement.

stars

commits

Python

primary language

Oct 17, 2023

updated

README

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.

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.

10 commits

RootYuan/vit-s-vqgan

patil-suraj/vqgan-jax

JAX implementation of VQGAN

Qiyuan-Ge/PaintMind

Fast and controllable text-to-image model.

Emma02/vqvae_ckpts

dalle-mini/vqgan_imagenet_f16_16384

araachie/river

Efficient Video Prediction via Sparsely Conditioned Flow Matching. In ICCV, 2023.

lavonda-li/MIMICEchoQAInfer

Visual question answering on echocardiogram images using MedGemma and LLaVA-Med models

huggingface/open-muse

Open reproduction of MUSE for fast text2image generation.

358

Python

100.0%

README

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.