unify detection augmentations (#1351)

mindee · Oct 20, 2023 · 4c47ac9 · 4c47ac9
1 parent 89745cf
commit 4c47ac9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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(