d

RobinNet/mrunetpred_gts.py

@@ -148,9 +148,9 @@ def main():
 
                 if npys3d:
                     if y_pred3d == []:
-                        y_pred3d = np.flipud(y_pred[:,:,:,0].numpy())
+                        y_pred3d = y_pred[:,:,:,0].numpy()
                     else:
-                        y_pred3d =  np.append(np.flipud(y_pred[:,:,:,0].numpy()), y_pred3d, axis=0)
+                        y_pred3d =  np.append((y_pred[:,:,:,0].numpy(), y_pred3d, axis=0)
                     #print(np.shape(y_pred3d))
 
 
@@ -179,7 +179,7 @@ def main():
 
 
             if npys3d:
-                #y_pred3d = np.flip(y_pred3d)
+                y_pred3d = np.flip(y_pred3d)
                 #np.save(img_savepath + "P" + str(TP_num).zfill(2) + "_pred3D", y_pred3d)
                 np.save("data/npy_thresh/" + "P" + str(TP_num).zfill(2) + "_mr_ctunetpred.gipl", y_pred3d)
                 print("ypred shape: ", np.shape(y_pred3d))