From 4c47ac96c5d6524d4b09353791849d79cafd94ba Mon Sep 17 00:00:00 2001 From: Felix Dittrich Date: Fri, 20 Oct 2023 05:28:35 +0200 Subject: [PATCH] unify detection augmentations (#1351) --- references/detection/train_tensorflow.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/references/detection/train_tensorflow.py b/references/detection/train_tensorflow.py index 95437b444f..d87c490699 100644 --- a/references/detection/train_tensorflow.py +++ b/references/detection/train_tensorflow.py @@ -226,9 +226,13 @@ def main(args): # Augmentations T.RandomApply(T.ColorInversion(), 0.1), T.RandomJpegQuality(60), + T.RandomApply(T.GaussianNoise(mean=0.1, std=0.1), 0.1), + T.RandomApply(T.RandomShadow(), 0.4), + T.RandomApply(T.GaussianBlur(kernel_shape=3, std=(0.1, 0.1)), 0.3), T.RandomSaturation(0.3), T.RandomContrast(0.3), T.RandomBrightness(0.3), + T.RandomApply(T.ToGray(), 0.1), ] ), sample_transforms=T.SampleCompose(