diff --git a/lightly/transforms/dino_transform.py b/lightly/transforms/dino_transform.py index 4c59b39e7..42e810401 100644 --- a/lightly/transforms/dino_transform.py +++ b/lightly/transforms/dino_transform.py @@ -1,7 +1,6 @@ from typing import Dict, List, Optional, Tuple, Union import PIL -import torchvision.transforms as T from PIL.Image import Image from torch import Tensor @@ -9,6 +8,7 @@ from lightly.transforms.multi_view_transform import MultiViewTransform from lightly.transforms.rotation import random_rotation_transform from lightly.transforms.solarize import RandomSolarization +from lightly.transforms.torchvision_v2_compatibility import torchvision_transforms as T from lightly.transforms.utils import IMAGENET_NORMALIZE