zeahub/hvae

Hierarchical Variational Autoencoder (HVAE)

Generative model for 2D B-mode Ultrasound Images. This model can be used for unconditional sampling as well as inpainting tasks.

Current architecture in this folder is only trained on resolution 256x256x3 (3 video frames per input/output).

🌽 The zea implementation is a snippet, adapted from the source.

The weights that are currently present can be read as follows:

• lvh: trained on EchonetLVH dataset
• ur: encoder retrained with UniformRandomLines subsampling zea.agent
• ge: encoder retrained with GreedyEntropy subsampling zea.agent
• 24: 24 lines/columns used when subsampling

📚 A usage tutorial can be found on Collab