Commit d4d9085d authored by sjromuel's avatar sjromuel
Browse files

f

parent 4ba8fda9
......@@ -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]])
......
Markdown is supported
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