Commit 6eca8d37 authored by sjromuel's avatar sjromuel
Browse files

f

parent 54c5f5f1
...@@ -149,9 +149,9 @@ def main(): ...@@ -149,9 +149,9 @@ def main():
if npys3d: if npys3d:
y_predflip = np.flip(y_pred.numpy()) y_predflip = np.flip(y_pred.numpy())
if y_pred3d == []: if y_pred3d == []:
y_pred3d = y_predflip[:,:,:,0] y_pred3d = y_predflip[:,:,:,1]
else: else:
y_pred3d = np.append(y_predflip[:,:,:,0], y_pred3d, axis=0) y_pred3d = np.append(y_predflip[:,:,:,1], y_pred3d, axis=0)
#print(np.shape(y_pred3d)) #print(np.shape(y_pred3d))
......
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