mcosarinsky commited on
Commit
23636ce
·
verified ·
1 Parent(s): 15f6498

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +55 -0
README.md ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ datasets:
4
+ - mcosarinsky/CheXmask-U
5
+ tags:
6
+ - medical
7
+ ---
8
+
9
+ # CheXmask-U: Uncertainty-aware landmark-based anatomical segmentation for chest X-rays
10
+
11
+ 📄 [Paper](https://arxiv.org/abs/2512.10715) | 💻 [Code](https://github.com/mcosarinsky/CheXmask-U) | 🎛️ [Interactive Demo](https://huggingface.co/spaces/mcosarinsky/CheXmask-U) | 📦 [Dataset](https://huggingface.co/datasets/mcosarinsky/CheXmask-U)
12
+
13
+ ---
14
+
15
+ ## Model Description
16
+
17
+ CheXmask-U is a **landmark-based chest X-ray segmentation model** providing node-wise **uncertainty estimation**. It outputs:
18
+
19
+ - Anatomical landmarks for lung and heart structures
20
+ - Latent uncertainty from the learned variational latent space
21
+ - Predictive uncertainty via stochastic output sampling
22
+
23
+ The model is implemented using a hybrid graph-convolutional architecture (`HybridGNet`), combining convolutional encoders with graph-based decoders. For full usage and code, see the [GitHub repository](https://github.com/mcosarinsky/CheXmask-U).
24
+
25
+ ---
26
+
27
+ ## Usage
28
+
29
+ ```python
30
+ from models.HybridGNet2IGSC import HybridGNetHF
31
+
32
+ device = "cuda" # or "cpu"
33
+ model = HybridGNetHF.from_pretrained(
34
+ "mcosarinsky/CheXmask-U",
35
+ subfolder="v1_skip",
36
+ device=device
37
+ )
38
+
39
+ # xray_image: tensor or suitable input
40
+ landmarks, _, _ = model(xray_image)
41
+ ```
42
+
43
+ ## Citation
44
+ If you use this model, please cite:
45
+ ```bibtex
46
+ @misc{cosarinsky2025chexmaskuquantifyinguncertaintylandmarkbased,
47
+ title={CheXmask-U: Quantifying uncertainty in landmark-based anatomical segmentation for X-ray images},
48
+ author={Matias Cosarinsky and Nicolas Gaggion and Rodrigo Echeveste and Enzo Ferrante},
49
+ year={2025},
50
+ eprint={2512.10715},
51
+ archivePrefix={arXiv},
52
+ primaryClass={cs.CV},
53
+ url={https://arxiv.org/abs/2512.10715},
54
+ }
55
+ ```