d

RobinNet/threshunetpred_gts.py

@@ -145,7 +145,7 @@ def main():
                 y_pred = Unet(image, weights, filter_multiplier, training=False)
 
                 if npys3d:
-                    y_predflip = np.flip(y_pred.numpy())
+                    y_predflip = y_pred.numpy()
                     if y_pred3d == []:
                         y_pred3d = y_predflip[:,:,:,1]
                     else:
@@ -178,7 +178,6 @@ def main():
 
 
             if npys3d:
-                y_pred3d = np.flip(y_pred3d)
                 filename = "P" + str(TP_num).zfill(2)
                 y_pred3d = cut_mr_ctunetpred(filename, y_pred3d)
                 #np.save(img_savepath + "P" + str(TP_num).zfill(2) + "_pred3D", y_pred3d)