mirrors
/
dinov2
mirror of https://github.com/facebookresearch/dinov2.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge 5a48d47656 into e1277af2ba

pull/411/merge
Sepehr Sameni 2024-04-27 08:50:40 +00:00 committed by GitHub
parent e1277af2ba 5a48d47656
commit 1054f48e75
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dinov2/data/transforms.py
View File

@ -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):