58c979f
1
2
3
4
5
6
7
8
9
10
11
import albumentations as A from albumentations.pytorch import ToTensorV2 # Define the transforms (only test) test_transforms = A.Compose([ A.Normalize((0.4914, 0.4822, 0.4465), (0.247, 0.243, 0.261)), ToTensorV2() ])
