From 5a48d47656ebe5b340b4372de671da93c0a23ceb Mon Sep 17 00:00:00 2001 From: Sepehr Sameni Date: Sat, 27 Apr 2024 10:31:28 +0200 Subject: [PATCH] Update transforms.py fixing the probability of applying GaussianBlur --- dinov2/data/transforms.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/dinov2/data/transforms.py b/dinov2/data/transforms.py index eb5f252b5..4791e134e 100644 --- a/dinov2/data/transforms.py +++ b/dinov2/data/transforms.py @@ -15,10 +15,8 @@ class GaussianBlur(transforms.RandomApply): """ def __init__(self, *, p: float = 0.5, radius_min: float = 0.1, radius_max: float = 2.0): - # NOTE: torchvision is applying 1 - probability to return the original image - keep_p = 1 - p transform = transforms.GaussianBlur(kernel_size=9, sigma=(radius_min, radius_max)) - super().__init__(transforms=[transform], p=keep_p) + super().__init__(transforms=[transform], p=p) class MaybeToTensor(transforms.ToTensor):