You could return the transformed and the original image tensor, which would increase the batch size.
Alternatively, you could randomly apply the transformation with a probability of 0.5 and double the length of the Dataset in its __len__ method.