Improve model card with metadata and GitHub link (#1)
Browse files- Improve model card with metadata and GitHub link (88d7e7731999487c6fa80d1fa56ec950bef6354e)
Co-authored-by: Niels Rogge <[email protected]>
README.md
CHANGED
|
@@ -1,7 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# Medical Image VAE Models (Brain MRI)
|
| 2 |
|
| 3 |
This repository contains Variational Autoencoder (VAE) models specifically adapted and fine-tuned for medical imaging tasks, particularly for brain MRI. Two different VAE model with different compression factor are provided (klf8, and klf4). Models are trained with KL loss, LIPIPs, and MSE loss.
|
| 4 |
|
|
|
|
| 5 |
|
| 6 |
## Original Model Source
|
| 7 |
|
|
@@ -42,9 +48,4 @@ If you use this model in your research, please cite this work.
|
|
| 42 |
journal={arXiv preprint arXiv:2502.16943},
|
| 43 |
year={2025}
|
| 44 |
}
|
| 45 |
-
```
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
---
|
| 49 |
-
license: mit
|
| 50 |
-
---
|
|
|
|
| 1 |
+
---
|
| 2 |
+
pipeline_tag: image-to-image
|
| 3 |
+
license: mit
|
| 4 |
+
---
|
| 5 |
+
|
| 6 |
# Medical Image VAE Models (Brain MRI)
|
| 7 |
|
| 8 |
This repository contains Variational Autoencoder (VAE) models specifically adapted and fine-tuned for medical imaging tasks, particularly for brain MRI. Two different VAE model with different compression factor are provided (klf8, and klf4). Models are trained with KL loss, LIPIPs, and MSE loss.
|
| 9 |
|
| 10 |
+
This model is used as part of the MAD-AD model, described in the paper [MAD-AD: Masked Diffusion for Unsupervised Brain Anomaly Detection](https://huggingface.co/papers/2502.16943). The code for the MAD-AD model can be found at https://github.com/farzad-bz/MAD-AD.
|
| 11 |
|
| 12 |
## Original Model Source
|
| 13 |
|
|
|
|
| 48 |
journal={arXiv preprint arXiv:2502.16943},
|
| 49 |
year={2025}
|
| 50 |
}
|
| 51 |
+
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|