f

RobinNet/nets/BaseNetwork.py

@@ -122,6 +122,8 @@ class BaseNetwork:
                         X_img_list.append(elem)
                 elif elem.endswith(self.gt_type+".gipl.npy"):
                     GT_img_list.append(elem)
+        if self.gt_type =="_mr_ctunetpred":
+            X_img_list.remove("P17_mr_ctunetpred.gipl.npy")
 
         list.sort(X_img_list)
         list.sort(GT_img_list)
@@ -160,6 +162,7 @@ class BaseNetwork:
     def train_test_split(self, number_patients, test_patients=(1, 2), X_img_list=[], GT_img_list=[]):
         train_data = []
         test_data = []
+
         for i in range(len(X_img_list)):
             if X_img_list[i].startswith('P' + str(test_patients[0]).zfill(2)):
                 test_data.append([X_img_list[i], GT_img_list[i]])