snapshot...

modules/util/augment.py

@@ -3,17 +3,8 @@ from functools import partial
 import tensorflow as tf
 
 
-def augment_image(
-        brightness_delta=0.05,
-        contrast_factor=[0.7, 1.3],
-        saturation=[0.6, 1.6]):
+def augment_image():
     """ Helper function used for augmentation of images in the dataset.
-      Args:
-        brightness_delta: maximum value for randomly assigning brightness of the image.
-        contrast_factor: list / tuple of minimum and maximum value of factor to set random contrast.
-                          None, if not to be used.
-        saturation: list / tuple of minimum and maximum value of factor to set random saturation.
-                    None, if not to be used.
       Returns:
         tf.data.Dataset mappable function for image augmentation
     """