Chang python script from reading png only to read npy images

I have an Augmentation script, it is working with png images, it is doing ( Rotation, Flipping (horizon and vertical), zooming, shearing)
I need you to add ( (Translation, Stretching, Elastic deforestation and contrast)
and after that make it work with npy images