f

RobinNet/13unet_mrt_threshgt.py

@@ -72,6 +72,8 @@ class mrt_unet(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_T1.gipl.npy")
 
         list.sort(X_img_list)
         list.sort(GT_img_list)

RobinNet/nets/BaseNetwork.py

@@ -122,8 +122,7 @@ 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_T1.gipl.npy")
+
 
         list.sort(X_img_list)
         list.sort(GT_img_list)