snapshot...

modules/deeplearning/icing_fcn.py

@@ -34,6 +34,7 @@ CONV3D = False
 
 NOISE_TRAINING = False
 NOISE_STDDEV = 0.10
+DO_AUGMENT = False
 
 img_width = 16
 
@@ -304,6 +305,15 @@ class IcingIntensityFCN:
             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 = np.concatenate([data, data_ud])
+            data_alt = np.concatenate([data_alt, data_alt_ud])
+            label = np.concatenate([label, label_ud])
+
         return data, data_alt, label
 
     def get_parameter_data(self, param, nd_idxs, is_training):