Skip to content
GitLab
Commit 9bcd9389 authored by sjromuel's avatar sjromuel
Browse files

d

parent 39241fa1
Hide whitespace changes
Inline Side-by-side
RobinNet/nets/BaseNetwork.py
View file @ 9bcd9389
......@@ -108,7 +108,7 @@ class BaseNetwork:
else:
if elem.endswith("ct.gipl.npy"):
X_img_list.append(elem)
if self.gt_type == "thresh":
if self.gt_type == "thresh" or self.gt_type == "_mr_ctunetpred":
X_img_list.append(elem)
X_img_list.append(elem)
elif elem.endswith(self.gt_type+".gipl.npy"):
......@@ -117,7 +117,7 @@ class BaseNetwork:
for elem in full_list:
if elem.endswith("ct.gipl.npy"):
X_img_list.append(elem)
if self.gt_type == "thresh":
if self.gt_type == "thresh" 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