From e1e2648ebb480fc61ffde17b61eafac400074584 Mon Sep 17 00:00:00 2001 From: Admia1 Date: Sun, 23 Jun 2024 10:36:58 -0700 Subject: [PATCH] Debug --- utils/train_segmentation_utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/utils/train_segmentation_utils.py b/utils/train_segmentation_utils.py index e290c0e..c893044 100644 --- a/utils/train_segmentation_utils.py +++ b/utils/train_segmentation_utils.py @@ -76,8 +76,7 @@ def process_img_train(self, ind: float) -> np.array: :return: volume and labels (output mask) """ - img_vert = self.array_training_img[ind] - + img_vert = self.array_training_img[ind].astype(int) patch_nifti = nib.load(self.training_raw[img_vert[0]]) mask_patch_nifti = nib.load(self.training_derivatives[img_vert[0]])