From c743dfa35096ea17a98c0d951e7a3af9d0f54df9 Mon Sep 17 00:00:00 2001 From: "Victor C." <98432934+CobV@users.noreply.github.com> Date: Wed, 4 Sep 2024 12:05:07 -0400 Subject: [PATCH] =?UTF-8?q?The=20label=200=20is=20used=20to=20represent=20?= =?UTF-8?q?the=20background.=20So=20for=20the=20organs,=20we=20can=20use?= =?UTF-8?q?=20labels=20starting=20from=C2=A01.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- preprocess/data_preprocess.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/preprocess/data_preprocess.py b/preprocess/data_preprocess.py index 1852f78..ded4c14 100644 --- a/preprocess/data_preprocess.py +++ b/preprocess/data_preprocess.py @@ -668,7 +668,8 @@ def batch_process_cases(in_dir, out_dir): oars2 = ['lung_l', 'lung_r','spinalcord', 'esophagus', 'heart', 'ptv'] # for DL labels2 = dict.fromkeys(oars,0) - label_seq = [0, 6, 7, 1, 2, 3, 4, 5, 1] + label_seq = [1, 7, 8, 2, 3, 4, 5, 6, 1] + #label_seq = [0, 6, 7, 1, 2, 3, 4, 5, 1] for w2, key2 in enumerate(oars): labels2[key2] = label_seq[w2] ## labels2 generate full organ mask to caluclate evalution DVHs not for learning