From 637ac0cdf1fa8ad435d98cf6767fd502e0407119 Mon Sep 17 00:00:00 2001 From: felix Date: Fri, 20 Dec 2024 09:35:30 +0100 Subject: [PATCH] Revert rnd shadow p from test --- references/detection/train_pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/references/detection/train_pytorch.py b/references/detection/train_pytorch.py index 4c9d99e26..f4d36b4fe 100644 --- a/references/detection/train_pytorch.py +++ b/references/detection/train_pytorch.py @@ -264,7 +264,7 @@ def main(args): T.RandomApply(T.GaussianBlur(sigma=(0.5, 1.5)), 0.2), ]), Compose([ - T.RandomApply(T.RandomShadow(), 1.0), + T.RandomApply(T.RandomShadow(), 0.3), T.RandomApply(T.GaussianNoise(), 0.1), T.RandomApply(T.GaussianBlur(sigma=(0.5, 1.5)), 0.3), RandomGrayscale(p=0.15),