def rotate(img, p):
    if (np.random.sample() <= p):
        angle = ((np.random.random_sample() * 40) - 20)
        return img.rotate(angle, Image.BILINEAR)
    else:
        return img
def rotate(img, p):
    if (np.random.sample() <= p):
        angle = ((np.random.random_sample() * 40) - 20)
        return img.rotate(angle, Image.BILINEAR)
    else:
        return img