VasoVision / README.md
JakJak504's picture
Update README.md
8bfc5d5 verified
metadata
license: mit

Overview VasoVision

VasoVision is a multi-head classification model designed to classify various aspects related to cardiovascular imaging.
Using a multi-head loss approach, the model can effectively handle multiple classification tasks simultaneously.

Label Mappings

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