VasoVision / README.md
JakJak504's picture
Update README.md
8bfc5d5 verified
---
license: mit
---
# Overview VasoVision
VasoVision is a multi-head classification model designed to classify various aspects related to cardiovascular imaging. <br>
Using a multi-head loss approach, the model can effectively handle multiple classification tasks simultaneously.
# Label Mappings
```yaml
contrast_agent:
"yes": 1
"no": 0
main_structure:
Left Coronary: 0
Right Coronary: 1
Other: 2
Graft: 3
Catheter: 4
Femoral: 5
LV: 6
TAVR: 7
Aorta: 8
Radial: 9
TEE probe: 10
Pigtail: 11
stent_presence:
present: 1
absent: 0
dominance:
right_dominant: 0
left_dominant: 1
```
# Usage
Can be ran using https://github.com/HeartWise-AI/Orion