snapshot...

modules/deeplearning/icing_cnn.py

@@ -28,6 +28,7 @@ NUM_EPOCHS = 100
 
 TRACK_MOVING_AVERAGE = False
 EARLY_STOP = False
+DO_AUGMENT = True
 
 TRIPLET = False
 CONV3D = False
@@ -311,6 +312,19 @@ class IcingIntensityNN:
             else:
                 self.in_mem_data_cache_test[key] = (data, data_alt, label)
 
+        if is_training and DO_AUGMENT:
+            data_ud = np.flip(data, axis=1)
+            data_alt_ud = np.copy(data_alt)
+            label_ud = np.copy(label)
+
+            data_lr = np.flip(data, axis=2)
+            data_alt_lr = np.copy(data_alt)
+            label_lr = np.copy(label)
+
+            data = np.concatenate([data, data_ud, data_lr])
+            data_alt = np.concatenate([data_alt, data_alt_ud, data_alt_lr])
+            label = np.concatenate([label, label_ud, label_lr])
+
         return data, data_alt, label
 
     def get_parameter_data(self, param, nd_idxs, is_training):