Commit 39d8c817 authored by sjromuel's avatar sjromuel
Browse files

d

parent 93b56fdb
......@@ -21,7 +21,7 @@ def main():
root = tk.Tk()
root.withdraw()
file_path = filedialog.askopenfilename(initialdir="finalResults/complete_segmr/mr_unet_cv_ctthresh")
file_path = filedialog.askopenfilename(initialdir="saves/mr_unet_cv_ctthresh")
file_path = file_path[:-9]
print(file_path)
......@@ -147,11 +147,11 @@ def main():
y_pred = Unet(image, weights, filter_multiplier, training=False)
if npys3d:
y_pred3d = np.flip(y_pred3d)
y_predflip = np.flipud(y_pred)
if y_pred3d == []:
y_pred3d = y_pred[:,:,:,0].numpy()
y_pred3d = y_predflip[:,:,:,0].numpy()
else:
y_pred3d = np.append(y_pred[:,:,:,0].numpy(), y_pred3d, axis=0)
y_pred3d = np.append(y_predflip[:,:,:,0].numpy(), y_pred3d, axis=0)
#print(np.shape(y_pred3d))
......
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