d

RobinNet/allModelsCreatePredNpys.py

@@ -228,11 +228,11 @@ def main():
                             test_loss_hdd2.append(hausdorff_distance_filter2.GetHausdorffDistance())
 
                             if y_pred3d == []:
-                                y_pred3d = pred_np
-                                y_true3d = y_true_np
+                                y_pred3d = np.reshape(pred_np, newshape=(1, 512, 512))
+                                y_true3d = np.reshape(y_true_np, newshape=(1, 512, 512))
                             else:
-                                y_pred3d = np.append(pred_np, y_pred3d, axis=0)
-                                y_true3d = np.append(y_true_np, y_true3d, axis=0)
+                                y_pred3d = np.append(np.reshape(pred_np, newshape=(1, 512, 512)), y_pred3d, axis=0)
+                                y_true3d = np.append(np.reshape(y_true_np, newshape=(1, 512, 512)), y_true3d, axis=0)
                         except:
                             pass
                     npy_savepath = "finalResults/3dnpys/" + model_path

RobinNet/tryOut.py

@@ -9,7 +9,11 @@ import matplotlib.pyplot as plt
 from scipy import ndimage
 from datetime import datetime
 
-pat = "06"
+predseg = np.load("finalResults/3dnpys/ctfgt2/unet_cv_ctfgt2/P01_pred3D.npy")
+
+print(np.shape(predseg))
+
+'''pat = "06"
 
 inputimg = np.load("data/npy_thresh/P"+pat+"_mr_T1.gipl.npy")
 gt = np.load("data/npy_thresh/P"+pat+"_mr_ctunetpred.gipl.npy")
@@ -30,7 +34,7 @@ for i in range(np.shape(inputimg)[0]):
     fig.add_subplot(2, 3, 3)
     plt.imshow(ytrue[i,:,:], cmap=plt.cm.bone)
     plt.title('True')
-    plt.show()
+    plt.show()'''
     
 '''print(np.shape(img))
 img = np.flip(img)