Skip to contents

The table below shows all preprocessing and augmentation operations that are available in mlr3torch.

Key Label Packages Feature Types
augment_center_crop Center Crop Augmentation torchvision lazy_tensor
augment_color_jitter Color Jitter Augmentation torchvision lazy_tensor
augment_crop Crop Augmentation torchvision lazy_tensor
augment_hflip Horizontal Flip Augmentation torchvision lazy_tensor
augment_random_affine Random Affine Augmentation torchvision lazy_tensor
augment_random_choice Random Choice Augmentation torchvision lazy_tensor
augment_random_crop Random Crop Augmentation torchvision lazy_tensor
augment_random_horizontal_flip Random Horizontal Flip Augmentation torchvision lazy_tensor
augment_random_order Random Order Augmentation torchvision lazy_tensor
augment_random_resized_crop Random Resized Crop Augmentation torchvision lazy_tensor
augment_random_vertical_flip Random Vertical Flip Augmentation torchvision lazy_tensor
augment_resized_crop Resized Crop Augmentation torchvision lazy_tensor
augment_rotate Rotate Augmentation torchvision lazy_tensor
augment_vflip Vertical Flip Augmentation torchvision lazy_tensor
trafo_adjust_brightness Adjust Brightness Transformation torchvision lazy_tensor
trafo_adjust_gamma Adjust Gamma Transformation torchvision lazy_tensor
trafo_adjust_hue Adjust Hue Transformation torchvision lazy_tensor
trafo_adjust_saturation Adjust Saturation Transformation torchvision lazy_tensor
trafo_grayscale Grayscale Transformation torchvision lazy_tensor
trafo_nop No Transformation lazy_tensor
trafo_normalize Normalization Transformation torchvision lazy_tensor
trafo_pad Padding Transformation torchvision lazy_tensor
trafo_reshape Reshaping Transformation lazy_tensor
trafo_resize Resizing Transformation torchvision lazy_tensor
trafo_rgb_to_grayscale RGB to Grayscale Transformation torchvision lazy_tensor