dj

RobinNet/13unet_mrt_threshgt.py

@@ -67,7 +67,7 @@ class mrt_unet(BaseNetwork):
         for elem in full_list:
             if elem.endswith("T1.gipl.npy"):
                 X_img_list.append(elem)
-                if self.gt_type == "ctthresh_gt":
+                if self.gt_type == "ctthresh_gt" or self.gt_type == "_mr_ctunetpred":
                     X_img_list.append(elem)
                     X_img_list.append(elem)
             elif elem.endswith(self.gt_type + ".gipl.npy"):