Yes - not sure how this has been overlooked. For Python/torch I tend to use cv2 with an albumentations augmentation pipeline, which is much faster than PIL/torch transforms.