adding random uniform

cheind · Aug 20, 2018 · 192968b · 192968b
1 parent 5217f59
commit 192968b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/style/image.py b/style/image.py
@@ -54,6 +54,9 @@ def new_random_white(shape, mean=None, sigma=1e-2):
     img = np.clip(mean + white_noise(shape, sigma), 0, 1).astype(np.float32)
     return to_image(img)
 
+def new_random_range(shape, low=0, high=0.95):
+    return to_image(np.random.uniform(low, high, size=shape).astype(np.float32))
+
 def save(fname, x):
     to_pil(x).save(fname)