Image Jitter Python Tensorflow