Commit 866b63a1 authored by sjromuel's avatar sjromuel
Browse files

dj

parent 91eca17d
......@@ -34,7 +34,7 @@ class mrt_unet(BaseNetwork):
gt_type="segmr",
e=0,
continue_training=False,
modelname=mr_unet_cv_ctseg):
modelname="mr_unet_cv_ctseg"):
super().__init__( test_patients = test_patients,
val_patients = val_patients,
......
......@@ -64,12 +64,10 @@ class mrt_unet(BaseNetwork):
X_img_list = []
GT_img_list = []
# thresh_img_list = []
for elem in full_list:
if elem.endswith("T1.gipl.npy"):
X_img_list.append(elem)
if self.gt_type == "thresh":
if self.gt_type == "ctthresh_gt":
X_img_list.append(elem)
X_img_list.append(elem)
elif elem.endswith(self.gt_type + ".gipl.npy"):
......
......@@ -100,8 +100,8 @@ def main(argv):
#############################
#src_path = "../data/data_normal"
src_path = "../data/threshIMGs"
src_path = "../data/data_normal"
#src_path = "../data/threshIMGs"
#############################
......
......@@ -230,9 +230,6 @@ class BaseNetwork:
patient_npy[1] = np.load(self.img_path + patient_dataname[1])
shape_temp = np.shape(patient_npy[0])
num_slices = shape_temp[0]
print(patient_dataname)
print(np.shape(patient_npy[0]))
print(np.shape(patient_npy[1]))
if self.shrink_data:
new_patient_npy = np.zeros((2, num_slices, self.newSize[0], self.newSize[1]))
......
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