If you have numpy arrays, you can convert them to PIL Image format, and then apply data augmentation techniques in torchvision.transforms. The transformation is as follows:

If array of type uint8:

from PIL import Image
im = Image.fromarray(np_arr)


If array has type float:

from PIL import Image
img = Image.fromarray((np_arr*255).astype('uint8'))