Skip to content

Commit a2fd1a8

Browse files
committed
Added more intense Augmentation
1 parent aac1eab commit a2fd1a8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utils/data_loader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def augment_batch(self, image, label) -> tuple:
117117
if tf.random.normal([1]) < 0:
118118
image = tf.image.random_brightness(image, 0.2)
119119
if self.NUM_CHANNELS == 3 and tf.random.normal([1]) < 0:
120-
image = tf.image.random_hue(image, 0.1)
120+
image = tf.image.random_hue(image, 0.3)
121121
if self.NUM_CHANNELS == 3 and tf.random.normal([1]) < 0:
122122
image = tf.image.random_saturation(image, 0, 15)
123123

0 commit comments

Comments
 (0)