d

RobinNet/mrunetpred_gts.py

@@ -21,7 +21,7 @@ def main():
 
     root = tk.Tk()
     root.withdraw()
-    file_path = filedialog.askopenfilename(initialdir="finalResults/complete_segmr/mr_unet_cv_ctthresh")
+    file_path = filedialog.askopenfilename(initialdir="saves/mr_unet_cv_ctthresh")
     file_path = file_path[:-9]
     print(file_path)
 
@@ -147,11 +147,11 @@ def main():
                 y_pred = Unet(image, weights, filter_multiplier, training=False)
 
                 if npys3d:
-                    y_pred3d = np.flip(y_pred3d)
+                    y_predflip = np.flipud(y_pred)
                     if y_pred3d == []:
-                        y_pred3d = y_pred[:,:,:,0].numpy()
+                        y_pred3d = y_predflip[:,:,:,0].numpy()
                     else:
-                        y_pred3d =  np.append(y_pred[:,:,:,0].numpy(), y_pred3d, axis=0)
+                        y_pred3d =  np.append(y_predflip[:,:,:,0].numpy(), y_pred3d, axis=0)
                     #print(np.shape(y_pred3d))