From b83c818faf46db91fad4ba43c7504c29486be037 Mon Sep 17 00:00:00 2001 From: Guotai Wang Date: Wed, 16 May 2018 09:32:46 +0100 Subject: [PATCH] fix bug in loading ground truth --- util/data_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/util/data_loader.py b/util/data_loader.py index 2d08570..5677a28 100644 --- a/util/data_loader.py +++ b/util/data_loader.py @@ -118,7 +118,7 @@ def load_data(self): bbox.append([bbmin, bbmax]) in_size.append(volume_size) if(self.with_ground_truth): - label = self.__load_one_volume(self.patient_names[i], self.label_postfix) + label, _ = self.__load_one_volume(self.patient_names[i], self.label_postfix) label = crop_ND_volume_with_bounding_box(label, bbmin, bbmax) if(self.data_resize): label = resize_3D_volume_to_given_shape(label, self.data_resize, 0)