From d8c3fc93272a18b606a3bbc9884b179bf2dd3b22 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Sun, 27 Sep 2020 11:12:24 -0600
Subject: [PATCH] Fix random noise corruptor

It was functioning as a color shift
---
 codes/data/image_corruptor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/codes/data/image_corruptor.py b/codes/data/image_corruptor.py
index 79d0e47e..ac973989 100644
--- a/codes/data/image_corruptor.py
+++ b/codes/data/image_corruptor.py
@@ -78,7 +78,7 @@ class ImageCorruptor:
         elif 'noise' in aug:
             # Block noise
             noise_intensity = (rand_int % 4 + 2) / 255.0  # Between 1-4
-            img += np.random.randn() * noise_intensity
+            img += np.random.randn(*img.shape) * noise_intensity
         elif 'jpeg' in aug:
             # JPEG compression
             qf = (rand_int % 20 + 5)  # Between 5-25