Commit 404926dc authored by sjromuel's avatar sjromuel
Browse files

d

parent f409d5ae
...@@ -145,7 +145,7 @@ def main(): ...@@ -145,7 +145,7 @@ def main():
y_pred = Unet(image, weights, filter_multiplier, training=False) y_pred = Unet(image, weights, filter_multiplier, training=False)
if npys3d: if npys3d:
y_predflip = y_pred.numpy() y_predflip = np.flip(y_pred.numpy())
if y_pred3d == []: if y_pred3d == []:
y_pred3d = y_predflip[:,:,:,1] y_pred3d = y_predflip[:,:,:,1]
else: else:
......
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