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