Commit 39241fa1 authored by sjromuel's avatar sjromuel
Browse files

dj

parent 31846fc0
......@@ -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"):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment