changed labels file

Author: ccosmin97

README.md

@@ -16,7 +16,7 @@
 <div style="text-align: justify; width: 100%;">
 This repository presents an auto-contouring algorithm [??] segmenting the entire heart, including its chambers and coronary arteries. This algorithm was trained on 560 thoracic planning computed tomography (CT) scans from Brigham and Women’s Hospital/Dana-Farber Cancer Institute and internally validated on 70 additional cases. External validation was conducted on 283 patients with lung or breast cancer treated at Cedars-Sinai Medical Center (2005-2020). This auto-contouring algorithm segments the whole heart, the heart chambers (LA, LV, RA, RV) and the heart coronaries (LAD, LCX, LM, PDA, RCA). </div> 
 
-Please refer to [TS_Eclipse.csv](heartSegClinic/data/labels/TS_Eclipse.csv) for additional details on the algorithm segments. 
+Please refer to [modelLabels.csv](heartSegClinic/data/labels/modelLabels.csv) for additional details on the algorithm segments. 
 
 ## Usage
 

heartSegClinic/data/labels/TS_Eclipse.csv

@@ -0,0 +1,17 @@
+Model_label,Name,Color
+heart,Heart,255-128-128
+heart_atrium_left,Atrium_L,255-128-0
+heart_atrium_right,Atrium_R,255-128-0
+heart_ventricle_left,Ventricle_L,255-128-0
+heart_ventricle_right,Ventricle_R,255-128-0
+HEART,Heart,255-128-128
+LA,Atrium_L,255-128-0
+RA,Atrium_R,255-128-0
+LV,Ventricle_L,255-128-0
+RV,Ventricle_R,255-128-0
+LAD,Coronary_LAD,128-0-64
+PDA,Coronary_PDA,128-0-64
+RCA,Coronary_RCA,128-0-64
+LCX,Coronary_LCX,128-0-64
+LM,Coronary_LM,128-0-64
+CORONARIES,Coronaries,128-0-64

heartSegClinic/data/labels/modelLabels.csv

@@ -246,7 +246,7 @@ def str2bool(v):
 def heartSegClinic(inputPath : str, outputPath : str, verbose = False):
 
     modelPath = pkg_resources.files("heartSegClinic").joinpath("data/models")
-    labelsPath = pkg_resources.files("heartSegClinic").joinpath("data/labels/TS_Eclipse.csv")
+    labelsPath = pkg_resources.files("heartSegClinic").joinpath("data/labels/modelLabels.csv")
 
     df_labels   = pd.read_csv(labelsPath)