From 21dac51a0838518774e775c61a369539f7cdf8ef Mon Sep 17 00:00:00 2001 From: Ravnoor Gill Date: Wed, 27 Sep 2023 10:17:41 -0400 Subject: [PATCH] fix lesion labelling bug (gh-21) --- app/utils/post_processor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app/utils/post_processor.py b/app/utils/post_processor.py index e4df621..df58515 100644 --- a/app/utils/post_processor.py +++ b/app/utils/post_processor.py @@ -56,7 +56,7 @@ def post_processing(input_scan, options, header, save_nifti=True): current_voxels = np.stack(np.where(pred_labels == l), axis=1) labels_scan[ current_voxels[:, 0], current_voxels[:, 1], current_voxels[:, 2] - ] = num_elements_by_lesion[l].astype(np.int) + ] = label_list[l] count = np.count_nonzero(num_elements_by_lesion.astype(dtype=np.int) > l_min)