Skip to content

Use torchvision functions / transforms for data augmentation

Use torchvision functions / transforms for data augmentation (https://pytorch.org/vision/stable/transforms.html):

  • remove the apply_preprocessing function
  • select torchvision / imgaug transforms
  • apply transformations using the transform arguments in the Dataset or DataLoader