f

RobinNet/mrunetpred_gts.py

@@ -149,9 +149,9 @@ def main():
                 if npys3d:
                     y_predflip = np.flip(y_pred.numpy())
                     if y_pred3d == []:
-                        y_pred3d = y_predflip[:,:,:,0]
+                        y_pred3d = y_predflip[:,:,:,1]
                     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))