Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
---
|
| 3 |
+
pipeline_tag: image-segmentation
|
| 4 |
+
tags:
|
| 5 |
+
- medical-imaging
|
| 6 |
+
- echocardiography
|
| 7 |
+
- ultrasound
|
| 8 |
+
- multi-view-segmentation
|
| 9 |
+
- text-prompted-segmentation
|
| 10 |
+
- pytorch
|
| 11 |
+
- arxiv:2611.13425
|
| 12 |
+
---
|
| 13 |
+
|
| 14 |
+
# Prompt2Echo (P2Echo)
|
| 15 |
+
### Text-Prompted Multi-View Echocardiography Segmentation with One Model
|
| 16 |
+
|
| 17 |
+
Official pretrained weights for **P2Echo**, a unified framework for text-prompted segmentation of multiple cardiac structures across standard echocardiographic views.
|
| 18 |
+
|
| 19 |
+
P2Echo uses structured, view-aware text prompts to control segmentation through a single model. Its **Dynamic Image-Text Alignment (DyITA)** module adaptively conditions decoder features on the requested view and anatomical structure.
|
| 20 |
+
|
| 21 |
+
## Resources
|
| 22 |
+
|
| 23 |
+
- 📄 **Paper:** https://arxiv.org/abs/2611.13425
|
| 24 |
+
- 💻 **Code:** https://github.com/moeinheidari7829/P2Echo
|
| 25 |
+
- 🌐 **Project Page:** https://moeinheidari7829.github.io/p2echo/
|
| 26 |
+
- 🤗 **Model Repository:** https://huggingface.co/moein99/P2Echo
|
| 27 |
+
|
| 28 |
+
## Pretrained Weights
|
| 29 |
+
|
| 30 |
+
**[Download P2Echo Checkpoint (Google Drive)](https://drive.google.com/file/d/1wKF_MnWlMcpfYZTurb2wHOobENmhHQuj/view?usp=sharing)**
|
| 31 |
+
|
| 32 |
+
Download the checkpoint and place it at:
|
| 33 |
+
|
| 34 |
+
`checkpoints/p2echo.pth`
|
| 35 |
+
|
| 36 |
+
The model supports four echocardiographic views (2CH, 3CH, 4CH, PSAX) and five cardiac structures (LV, MYO, LA, RV, RA).
|
| 37 |
+
|
| 38 |
+
## Installation
|
| 39 |
+
|
| 40 |
+
```bash
|
| 41 |
+
git clone https://github.com/moeinheidari7829/P2Echo.git
|
| 42 |
+
cd P2Echo
|
| 43 |
+
pip install -r requirements.txt
|
| 44 |
+
```
|
| 45 |
+
|
| 46 |
+
## Evaluation
|
| 47 |
+
|
| 48 |
+
P2Echo is evaluated on an internal multi-view dataset and three held-out external datasets. Detailed quantitative comparisons, ablation studies, and implementation details are available in the paper.
|
| 49 |
+
|
| 50 |
+
## Citation
|
| 51 |
+
|
| 52 |
+
```bibtex
|
| 53 |
+
@article{heidari2026p2echo,
|
| 54 |
+
title={Prompt2Echo (P2Echo): Text-Prompted Multi-View Echocardiography Segmentation with One Model},
|
| 55 |
+
author={Heidari, Moein and Kimbowa, Alvin and Tabatabaei, Mahdi and Bozorgpour, Afshin and Ghorbani Kolahi, Sina and Jiang, River and Chen, Wenjin and Foran, David J. and Grewal, Jasmine and Merhof, Dorit and Hacihaliloglu, Ilker},
|
| 56 |
+
journal={arXiv preprint arXiv:2611.13425},
|
| 57 |
+
year={2026}
|
| 58 |
+
}
|
| 59 |
+
```
|
| 60 |
+
|
| 61 |
+
## Intended Use
|
| 62 |
+
|
| 63 |
+
This model is intended for research in medical image analysis and echocardiographic segmentation. It has not been validated for clinical decision-making.
|