diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index a6d9f862a1ecb6462a891f4a6f92c8aca3531384..ce9b188ee7d71323bd0a7323305585a43c16be41 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/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):