Semantic Diffusion Posterior Sampling for Cardiac Ultrasound Dehazing

Eindhoven University of Technology, the Netherlands

Cardiac Ultrasound Dehazing Animation

The algorithm is implemented using Keras with JAX backend. Furthermore it heavily relies on the zea ultrasound library.

Either install the following in your Python environment, or use the Dockerfile provided in this repository.

# requires Python>=3.10
pip install tyro optuna zea==0.0.4
pip install -U "jax[cuda12]"

Although the code was primarily tested with JAX as the Keras backend, TensorFlow and PyTorch should also work.

Some example images are downloaded in the ./assets folder. The models are automatically downloaded from the Hugging Face Model Hub.

python main.py --input-folder ./assets --output-folder ./temp