d

RobinNet/threshunetpred_gts.py

@@ -145,7 +145,7 @@ def main():
                 y_pred = Unet(image, weights, filter_multiplier, training=False)
 
                 if npys3d:
-                    y_predflip = y_pred.numpy()
+                    y_predflip = np.flip(y_pred.numpy())
                     if y_pred3d == []:
                         y_pred3d = y_predflip[:,:,:,1]
                     else: