Hi, thanks for your outstanding work and would you give me any suggestions how to resize the image size during the training phase such (224, 244). Besides, your iaa augmentation technique technique don't work for me to resize the image. Further, i tried different augmentation techniques to resize the image size but unfortunately received the errors from dataloaders.

[AliMedVisionTech]

Below Replace with the IAA augmented method
transform = torchvision.transforms.Compose([
transforms.Resize((224, 224)),
transforms.ColorJitter(hue=.05, saturation=.05),
transforms.RandomHorizontalFlip(),
transforms.RandomRotation(20, resample=PIL.Image.BILINEAR),
transforms.ToTensor()
])

x, y = self.transforms(images=self.x[None, index], segmentation_maps=self.y[None, index])
TypeError: call() got an unexpected keyword argument 'images'