Skip to content
Snippets Groups Projects
Commit 3541ac51 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 8c39ebbe
Branches
No related tags found
No related merge requests found
...@@ -34,6 +34,7 @@ CONV3D = False ...@@ -34,6 +34,7 @@ CONV3D = False
NOISE_TRAINING = False NOISE_TRAINING = False
NOISE_STDDEV = 0.10 NOISE_STDDEV = 0.10
DO_AUGMENT = False
img_width = 16 img_width = 16
...@@ -304,6 +305,15 @@ class IcingIntensityFCN: ...@@ -304,6 +305,15 @@ class IcingIntensityFCN:
else: else:
self.in_mem_data_cache_test[key] = (data, data_alt, label) 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 return data, data_alt, label
def get_parameter_data(self, param, nd_idxs, is_training): def get_parameter_data(self, param, nd_idxs, is_training):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment