Difference between torchvision.transforms

[jizhang02]

Hello,

I would like to know what's the difference between your volumentations and the lib of torchvision.transforms?

import torchvision.transforms as transforms

 im_aug = transforms.Compose([
                #tfs.Resize(120),
                #transforms.RandomCrop(image_size),
                transforms.RandomHorizontalFlip(),
                transforms.RandomVerticalFlip(),
                #transforms.RandomRotation(5),
                #transforms.ColorJitter(brightness=0.5, contrast=0.5, hue=0.5),
                #transforms.RandomPerspective(),
                #transforms.GaussianBlur(kernel_size=(3, 3), sigma=(0.1, 1.5)),
                #transforms.RandomInvert(),
                #transforms.RandomPosterize(bits=2),
                #transforms.RandomAdjustSharpness(sharpness_factor=2),
                #transforms.RandomAutocontrast(),
                #transforms.RandomEqualize()
            ])
            img = im_aug(img)

Thanks,
Jing

[ZFTurbo]

The main difference is that torchvision is for images e.g. shapes like (224, 224, 3). This library is for 3D volumes e.g. shapes like (224, 224, 224, 3).