snapshot..

modules/deeplearning/icing_fcn.py

@@ -317,11 +317,11 @@ class IcingIntensityFCN:
             data_alt_lr = np.copy(data_alt)
             label_lr = np.copy(label)
 
-            data_r1 = np.rot90(data, k=1)
+            data_r1 = np.rot90(data, k=1, axes=(1, 2))
             data_alt_r1 = np.copy(data_alt)
             label_r1 = np.copy(label)
 
-            data_r2 = np.rot90(data, k=1)
+            data_r2 = np.rot90(data, k=1, axes=(1, 2))
             data_alt_r2 = np.copy(data_alt)
             label_r2 = np.copy(label)