Add comprehensive model card
Browse filesThis PR adds a comprehensive model card for CoCoLIT, ensuring better discoverability and documentation on the Hugging Face Hub.
It includes:
- The `license` (`cc-by-nc-4.0`)
- The `pipeline_tag` (`image-to-image`)
- A link to the paper ([CoCoLIT: ControlNet-Conditioned Latent Image Translation for MRI to Amyloid PET Synthesis](https://huggingface.co/papers/2508.01292))
- A link to the GitHub repository
- Installation instructions
- Usage examples
- The academic citation
Please review and merge this PR if everything looks good.
README.md
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: cc-by-nc-4.0
|
| 3 |
+
pipeline_tag: image-to-image
|
| 4 |
+
---
|
| 5 |
+
|
| 6 |
+
# CoCoLIT: ControlNet-Conditioned Latent Image Translation for MRI to Amyloid PET Synthesis
|
| 7 |
+
|
| 8 |
+
This model, CoCoLIT, presents a diffusion-based latent generative framework for synthesizing amyloid PET scans from structural MRI. It addresses challenges in 3D neuroimaging data translation through a novel Weighted Image Space Loss (WISL), Latent Average Stabilization (LAS), and ControlNet-based conditioning for improved synthesis quality and inference consistency.
|
| 9 |
+
|
| 10 |
+
**Paper**: [CoCoLIT: ControlNet-Conditioned Latent Image Translation for MRI to Amyloid PET Synthesis](https://huggingface.co/papers/2508.01292)
|
| 11 |
+
**Code**: [https://github.com/brAIn-science/CoCoLIT](https://github.com/brAIn-science/CoCoLIT)
|
| 12 |
+
|
| 13 |
+
<div align="center">
|
| 14 |
+
<img src="https://github.com/brAIn-science/CoCoLIT/raw/main/docs/assets/preview.gif"/>
|
| 15 |
+
</div>
|
| 16 |
+
|
| 17 |
+
## Installation
|
| 18 |
+
|
| 19 |
+
This repository requires Python 3.10 and PyTorch 2.0 or later. To install the latest version, run:
|
| 20 |
+
|
| 21 |
+
```bash
|
| 22 |
+
pip install cocolit
|
| 23 |
+
```
|
| 24 |
+
|
| 25 |
+
## Usage
|
| 26 |
+
|
| 27 |
+
After installing the package, you can convert a T1-weighted MRI to a Florbetapir SUVR map by running:
|
| 28 |
+
|
| 29 |
+
```bash
|
| 30 |
+
mri2pet --i /path/to/t1.nii.gz --o /path/to/output.nii.gz
|
| 31 |
+
```
|
| 32 |
+
|
| 33 |
+
To replicate the results presented in the paper, include the `--m 64` flag.
|
| 34 |
+
|
| 35 |
+
<img width="100%" src="https://github.com/brAIn-science/CoCoLIT/raw/main/docs/assets/cocolit-cli.svg">
|
| 36 |
+
|
| 37 |
+
## Disclaimer
|
| 38 |
+
|
| 39 |
+
This software is not intended for clinical use. The code is not available for commercial applications. For commercial inquiries, please contact the corresponding authors.
|
| 40 |
+
|
| 41 |
+
## Citing
|
| 42 |
+
|
| 43 |
+
Arxiv Preprint:
|
| 44 |
+
|
| 45 |
+
```bib
|
| 46 |
+
@article{sargood2025cocolit,
|
| 47 |
+
title={CoCoLIT: ControlNet-Conditioned Latent Image Translation for MRI to Amyloid PET Synthesis},
|
| 48 |
+
author={Sargood, Alec and Puglisi, Lemuel and Cole, James H and Oxtoby, Neil P and Rav{\`\i}, Daniele and Alexander, Daniel C},
|
| 49 |
+
journal={arXiv preprint arXiv:2508.01292},
|
| 50 |
+
year={2025}
|
| 51 |
+
}
|
| 52 |
+
```
|