nielsr HF Staff commited on
Commit
cc4b05e
·
verified ·
1 Parent(s): 8c1e736

Add comprehensive model card

Browse files

This 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.

Files changed (1) hide show
  1. README.md +52 -0
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
+ ```