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(