Downloads · 30 days
348
3% of all-time downloads
zeahub/hvae
hvae is a machine learning model from zeahub. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Generative model for 2D B-mode Ultrasound Images. This model can be used for unconditional sampling as well as inpainting tasks.
Downloads · 30 days
348
3% of all-time downloads
All-time downloads
10.6K
Public
Repo size
3.3 GB
Likes
0
Public
Click a slice to open those files.
.h5908 MB · 100%
From the Hugging Face model README
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 datasetur: encoder retrained with UniformRandomLines subsampling zea.agentge: encoder retrained with GreedyEntropy subsampling zea.agent24: 24 lines/columns used when subsampling📚 A usage tutorial can be found on Collab