Commit f409d5ae authored by sjromuel's avatar sjromuel
Browse files

d

parent 137b996a
......@@ -145,7 +145,7 @@ def main():
y_pred = Unet(image, weights, filter_multiplier, training=False)
if npys3d:
y_predflip = np.flip(y_pred.numpy())
y_predflip = y_pred.numpy()
if y_pred3d == []:
y_pred3d = y_predflip[:,:,:,1]
else:
......@@ -178,7 +178,6 @@ def main():
if npys3d:
y_pred3d = np.flip(y_pred3d)
filename = "P" + str(TP_num).zfill(2)
y_pred3d = cut_mr_ctunetpred(filename, y_pred3d)
#np.save(img_savepath + "P" + str(TP_num).zfill(2) + "_pred3D", 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