import pytorchvideo.data
from pytorchvideo.transforms import (
     ApplyTransformToKey,
     Normalize,
     RandomShortSideScale,
     RemoveKey,
     ShortSideScale,
     UniformTemporalSubsample,
 )
from torchvision.transforms import (
     Compose,
     Lambda,
     RandomCrop,
     RandomHorizontalFlip,
     Resize,
 )