File size: 490 Bytes
5fa1a76
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import pytorchvideo.data
from pytorchvideo.transforms import (
     ApplyTransformToKey,
     Normalize,
     RandomShortSideScale,
     RemoveKey,
     ShortSideScale,
     UniformTemporalSubsample,
 )
from torchvision.transforms import (
     Compose,
     Lambda,
     RandomCrop,
     RandomHorizontalFlip,
     Resize,
 )

For the training dataset transformations, use a combination of uniform temporal subsampling, pixel normalization, random cropping, and random horizontal flipping.